wrangler 0.0.0-e6733a3 → 0.0.0-e9307365

package/src/__tests__/jest.setup.ts

@@ -1,22 +1,150 @@
-import { mockFetchInternal } from "./mock-cfetch";
+import fetchMock from "jest-fetch-mock";
+import {
+	fetchInternal,
+	fetchKVGetValue,
+	fetchR2Objects,
+	getCloudflareAPIBaseURL,
+} from "../cfetch/internal";
 import { confirm, prompt } from "../dialogs";
-import { fetchInternal } from "../cfetch/internal";
+import {
+	mockFetchInternal,
+	mockFetchKVGetValue,
+	mockFetchR2Objects,
+} from "./helpers/mock-cfetch";
+import { MockWebSocket } from "./helpers/mock-web-socket";
+
+// Mock out getPort since we don't actually care about what ports are open in unit tests.
+jest.mock("get-port", () => {
+	return {
+		__esModule: true,
+		default: jest.fn().mockImplementation(async (options) => options.port),
+	};
+});
+
+jest.mock("ws", () => {
+	return {
+		__esModule: true,
+		default: MockWebSocket,
+	};
+});
+
+jest.mock("undici", () => {
+	return {
+		...jest.requireActual("undici"),
+		fetch: jest.requireActual("jest-fetch-mock"),
+	};
+});
+
+fetchMock.doMock(() => {
+	// Any un-mocked fetches should throw
+	throw new Error("Unexpected fetch request");
+});
+
+jest.mock("../package-manager");
 
 jest.mock("../cfetch/internal");
 (fetchInternal as jest.Mock).mockImplementation(mockFetchInternal);
+(fetchKVGetValue as jest.Mock).mockImplementation(mockFetchKVGetValue);
+(getCloudflareAPIBaseURL as jest.Mock).mockReturnValue(
+	"https://api.cloudflare.com/client/v4"
+);
+(fetchR2Objects as jest.Mock).mockImplementation(mockFetchR2Objects);
 
 jest.mock("../dialogs");
 
 // By default (if not configured by mockConfirm()) calls to `confirm()` should throw.
-(confirm as jest.Mock).mockImplementation(() => {
-  throw new Error(
-    "Unexpected call to `confirm()`. You should use `mockConfirm()` to mock calls to `confirm()` with expectations. Search the codebase for `mockConfirm` to learn more."
-  );
+(confirm as jest.Mock).mockImplementation((text: string) => {
+	throw new Error(
+		`Unexpected call to \`confirm("${text}")\`.\nYou should use \`mockConfirm()\` to mock calls to \`confirm()\` with expectations. Search the codebase for \`mockConfirm\` to learn more.`
+	);
 });
 
 // By default (if not configured by mockPrompt()) calls to `prompt()` should throw.
-(prompt as jest.Mock).mockImplementation(() => {
-  throw new Error(
-    "Unexpected call to `prompt()`. You should use `mockPrompt()` to mock calls to `prompt()` with expectations. Search the codebase for `mockPrompt` to learn more."
-  );
+(prompt as jest.Mock).mockImplementation((text: string) => {
+	throw new Error(
+		`Unexpected call to \`prompt(${text}, ...)\`.\nYou should use \`mockPrompt()\` to mock calls to \`prompt()\` with expectations. Search the codebase for \`mockPrompt\` to learn more.`
+	);
+});
+
+jest.mock("../dev/dev", () => {
+	const { useApp } = jest.requireActual("ink");
+	const { useEffect } = jest.requireActual("react");
+	return jest.fn().mockImplementation(() => {
+		const { exit } = useApp();
+		useEffect(() => {
+			exit();
+		});
+		return null;
+	});
+});
+
+// Make sure that we don't accidentally try to open a browser window when running tests.
+// We will actually provide a mock implementation for `openInBrowser()` within relevant tests.
+jest.mock("../open-in-browser");
+
+// Mock the functions involved in getAuthURL so we don't take snapshots of the constantly changing URL.
+jest.mock("../user/generate-auth-url", () => {
+	return {
+		generateRandomState: jest.fn().mockImplementation(() => "MOCK_STATE_PARAM"),
+		generateAuthUrl: jest
+			.fn()
+			.mockImplementation(({ authUrl, clientId, callbackUrl, scopes }) => {
+				return (
+					authUrl +
+					`?response_type=code&` +
+					`client_id=${encodeURIComponent(clientId)}&` +
+					`redirect_uri=${encodeURIComponent(callbackUrl)}&` +
+					// we add offline_access manually for every request
+					`scope=${encodeURIComponent(
+						[...scopes, "offline_access"].join(" ")
+					)}&` +
+					`state=MOCK_STATE_PARAM&` +
+					`code_challenge=${encodeURIComponent("MOCK_CODE_CHALLENGE")}&` +
+					`code_challenge_method=S256`
+				);
+			}),
+	};
+});
+
+jest.mock("../user/generate-random-state", () => {
+	return {
+		generateRandomState: jest.fn().mockImplementation(() => "MOCK_STATE_PARAM"),
+	};
+});
+
+jest.mock("xdg-app-paths", () => {
+	return {
+		__esModule: true,
+		default: jest.fn().mockImplementation(() => {
+			return {
+				config() {
+					return jest.requireActual("node:path").resolve("test-xdg-config");
+				},
+			};
+		}),
+	};
+});
+
+jest.mock("create-cloudflare");
+
+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;
 });