wrangler 2.0.12 → 2.0.16

package/src/__tests__/jest.setup.ts

@@ -1,8 +1,8 @@
 import fetchMock from "jest-fetch-mock";
 import {
-  fetchInternal,
-  fetchKVGetValue,
-  getCloudflareAPIBaseURL,
+	fetchInternal,
+	fetchKVGetValue,
+	getCloudflareAPIBaseURL,
 } from "../cfetch/internal";
 import { confirm, prompt } from "../dialogs";
 import { mockFetchInternal, mockFetchKVGetValue } from "./helpers/mock-cfetch";
@@ -10,29 +10,29 @@ 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 +41,67 @@ 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"
 );
 
 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("../user/generate-random-state", () => {
+	return {
+		generateRandomState: jest.fn().mockImplementation(() => "MOCK_STATE_PARAM"),
+	};
+});