wrangler 0.0.0-e64812e → 0.0.0-e8e684c7

package/src/__tests__/jest.setup.ts

@@ -1,38 +1,45 @@
 import fetchMock from "jest-fetch-mock";
 import {
-  fetchInternal,
-  fetchKVGetValue,
-  getCloudflareAPIBaseURL,
+	fetchDashboardScript,
+	fetchInternal,
+	fetchKVGetValue,
+	fetchR2Objects,
+	getCloudflareAPIBaseURL,
 } from "../cfetch/internal";
 import { confirm, prompt } from "../dialogs";
-import { mockFetchInternal, mockFetchKVGetValue } from "./helpers/mock-cfetch";
+import {
+	mockFetchDashScript,
+	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),
-  };
+	return {
+		__esModule: true,
+		default: jest.fn().mockImplementation(async (options) => options.port),
+	};
 });
 
 jest.mock("ws", () => {
-  return {
-    __esModule: true,
-    default: MockWebSocket,
-  };
+	return {
+		__esModule: true,
+		default: MockWebSocket,
+	};
 });
 
 jest.mock("undici", () => {
-  return {
-    ...jest.requireActual("undici"),
-    fetch: jest.requireActual("jest-fetch-mock"),
-  };
+	return {
+		...jest.requireActual("undici"),
+		fetch: jest.requireActual("jest-fetch-mock"),
+	};
 });
 
 fetchMock.doMock(() => {
-  // Any un-mocked fetches should throw
-  throw new Error("Unexpected fetch request");
+	// Any un-mocked fetches should throw
+	throw new Error("Unexpected fetch request");
 });
 
 jest.mock("../package-manager");
@@ -41,37 +48,110 @@ 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"
+	"https://api.cloudflare.com/client/v4"
 );
+(fetchR2Objects as jest.Mock).mockImplementation(mockFetchR2Objects);
+(fetchDashboardScript as jest.Mock).mockImplementation(mockFetchDashScript);
 
 jest.mock("../dialogs");
 
 // By default (if not configured by mockConfirm()) calls to `confirm()` should throw.
 (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.`
-  );
+	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((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.`
-  );
+	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;
-  });
+	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("../is-ci", () => {
+	return { CI: { isCI: jest.fn().mockImplementation(() => false) } };
+});
+
+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;
+});