wrangler 0.0.0-e178d6fb → 0.0.0-e1c2f5b9

package/src/__tests__/jest.setup.ts

@@ -1,12 +1,35 @@
 import fetchMock from "jest-fetch-mock";
 import {
+	fetchDashboardScript,
 	fetchInternal,
 	fetchKVGetValue,
-	getCloudflareAPIBaseURL,
+	fetchR2Objects,
+	performApiFetch,
 } from "../cfetch/internal";
 import { confirm, prompt } from "../dialogs";
-import { mockFetchInternal, mockFetchKVGetValue } from "./helpers/mock-cfetch";
+import {
+	mockFetchInternal,
+	mockFetchR2Objects,
+	mockPerformApiFetch,
+} from "./helpers/mock-cfetch";
 import { MockWebSocket } from "./helpers/mock-web-socket";
+import { msw } from "./helpers/msw";
+
+/**
+ * The relative path between the bundled code and the Wrangler package.
+ * This is used as a reliable way to compute paths relative to the Wrangler package
+ * in the source files, rather than relying upon `__dirname` which can change depending
+ * on whether the source files have been bundled and the location of the outdir.
+ *
+ * This is exposed in the source via the `getBasePath()` function, which should be used
+ * in place of `__dirname` and similar Node.js constants.
+ */
+(
+	global as unknown as { __RELATIVE_PACKAGE_PATH__: string }
+).__RELATIVE_PACKAGE_PATH__ = "..";
+
+// Set `LC_ALL` to fix the language as English for the messages thrown by Yargs.
+process.env.LC_ALL = "en";
 
 // Mock out getPort since we don't actually care about what ports are open in unit tests.
 jest.mock("get-port", () => {
@@ -16,6 +39,17 @@ jest.mock("get-port", () => {
 	};
 });
 
+jest.mock("child_process", () => {
+	return {
+		__esModule: true,
+		...jest.requireActual("child_process"),
+		spawnSync: jest.fn().mockImplementation(async (binary, ...args) => {
+			if (binary === "cloudflared") return { error: true };
+			return jest.requireActual("child_process").spawnSync(binary, ...args);
+		}),
+	};
+});
+
 jest.mock("ws", () => {
 	return {
 		__esModule: true,
@@ -37,12 +71,35 @@ fetchMock.doMock(() => {
 
 jest.mock("../package-manager");
 
+// requests not mocked with `jest-fetch-mock` fall through
+// to `mock-service-worker`
+fetchMock.dontMock();
+beforeAll(() => {
+	msw.listen({
+		onUnhandledRequest: (request) => {
+			throw new Error(
+				`No mock found for ${request.method} ${request.url.href}
+				`
+			);
+		},
+	});
+});
+afterEach(() => {
+	msw.restoreHandlers();
+	msw.resetHandlers();
+});
+afterAll(() => msw.close());
+
 jest.mock("../cfetch/internal");
+(fetchDashboardScript as jest.Mock).mockImplementation(
+	jest.requireActual("../cfetch/internal")["fetchDashboardScript"]
+);
 (fetchInternal as jest.Mock).mockImplementation(mockFetchInternal);
-(fetchKVGetValue as jest.Mock).mockImplementation(mockFetchKVGetValue);
-(getCloudflareAPIBaseURL as jest.Mock).mockReturnValue(
-	"https://api.cloudflare.com/client/v4"
+(fetchKVGetValue as jest.Mock).mockImplementation(
+	jest.requireActual("../cfetch/internal").fetchKVGetValue
 );
+(fetchR2Objects as jest.Mock).mockImplementation(mockFetchR2Objects);
+(performApiFetch as jest.Mock).mockImplementation(mockPerformApiFetch);
 
 jest.mock("../dialogs");
 
@@ -100,6 +157,10 @@ jest.mock("../user/generate-auth-url", () => {
 	};
 });
 
+jest.mock("../is-ci", () => {
+	return { CI: { isCI: jest.fn().mockImplementation(() => false) } };
+});
+
 jest.mock("../user/generate-random-state", () => {
 	return {
 		generateRandomState: jest.fn().mockImplementation(() => "MOCK_STATE_PARAM"),
@@ -118,3 +179,25 @@ jest.mock("xdg-app-paths", () => {
 		}),
 	};
 });
+
+jest.mock("../metrics/metrics-config", () => {
+	const realModule = jest.requireActual("../metrics/metrics-config");
+	const fakeModule = {
+		...realModule,
+		// Although we mock out the getMetricsConfig() function in most tests,
+		// we need a way to reinstate it for the metrics specific tests.
+		// This is what `useOriginal` is for.
+		useOriginal: false,
+		getMetricsConfig: (...args: unknown[]) =>
+			fakeModule.useOriginal
+				? realModule.getMetricsConfig(...args)
+				: async () => {
+						return {
+							enabled: false,
+							deviceId: "mock-device",
+							userId: undefined,
+						};
+				  },
+	};
+	return fakeModule;
+});