wrangler 0.0.0-e2a26a0 → 0.0.0-e322475

package/src/__tests__/jest.setup.ts

@@ -1,7 +1,27 @@
 import fetchMock from "jest-fetch-mock";
-import { fetchInternal, fetchKVGetValue } from "../cfetch/internal";
+import {
+  fetchInternal,
+  fetchKVGetValue,
+  getCloudflareAPIBaseURL,
+} from "../cfetch/internal";
 import { confirm, prompt } from "../dialogs";
 import { mockFetchInternal, mockFetchKVGetValue } 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 {
@@ -9,6 +29,7 @@ jest.mock("undici", () => {
     fetch: jest.requireActual("jest-fetch-mock"),
   };
 });
+
 fetchMock.doMock(() => {
   // Any un-mocked fetches should throw
   throw new Error("Unexpected fetch request");
@@ -19,19 +40,68 @@ 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"
+);
 
 jest.mock("../dialogs");
 
 // By default (if not configured by mockConfirm()) calls to `confirm()` should throw.
-(confirm as jest.Mock).mockImplementation(() => {
+(confirm as jest.Mock).mockImplementation((text: string) => {
   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."
+    `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(() => {
+(prompt as jest.Mock).mockImplementation((text: string) => {
   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."
+    `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("../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("../generate-random-state", () => {
+  return {
+    generateRandomState: jest.fn().mockImplementation(() => "MOCK_STATE_PARAM"),
+  };
+});