wrangler 0.0.0-eab9542 → 0.0.0-eaf71b86c

package/src/__tests__/secret.test.ts

@@ -1,486 +0,0 @@
-import * as fs from "node:fs";
-import * as TOML from "@iarna/toml";
-import { mockAccountId, mockApiToken } from "./helpers/mock-account-id";
-import { setMockResponse, unsetAllMocks } from "./helpers/mock-cfetch";
-import { mockConsoleMethods } from "./helpers/mock-console";
-import {
-	mockConfirm,
-	mockPrompt,
-	clearConfirmMocks,
-	clearPromptMocks,
-} from "./helpers/mock-dialogs";
-import {
-	mockGetMemberships,
-	mockGetMembershipsFail,
-} from "./helpers/mock-oauth-flow";
-import { useMockStdin } from "./helpers/mock-stdin";
-import { runInTempDir } from "./helpers/run-in-tmp";
-import { runWrangler } from "./helpers/run-wrangler";
-
-describe("wrangler secret", () => {
-	const std = mockConsoleMethods();
-
-	runInTempDir();
-	mockAccountId();
-	mockApiToken();
-
-	afterEach(() => {
-		unsetAllMocks();
-		clearConfirmMocks();
-		clearPromptMocks();
-	});
-
-	describe("put", () => {
-		function mockPutRequest(
-			input: { name: string; text: string },
-			env?: string,
-			legacyEnv = false
-		) {
-			const servicesOrScripts = env && !legacyEnv ? "services" : "scripts";
-			const environment = env && !legacyEnv ? "/environments/:envName" : "";
-			setMockResponse(
-				`/accounts/:accountId/workers/${servicesOrScripts}/:scriptName${environment}/secrets`,
-				"PUT",
-				([_url, accountId, scriptName, envName], { body }) => {
-					expect(accountId).toEqual("some-account-id");
-					expect(scriptName).toEqual(
-						legacyEnv && env ? `script-name-${env}` : "script-name"
-					);
-					if (!legacyEnv) {
-						expect(envName).toEqual(env);
-					}
-					const { name, text, type } = JSON.parse(body as string);
-					expect(type).toEqual("secret_text");
-					expect(name).toEqual(input.name);
-					expect(text).toEqual(input.text);
-
-					return { name, type };
-				}
-			);
-		}
-
-		describe("interactive", () => {
-			useMockStdin({ isTTY: true });
-
-			it("should trim stdin secret value", async () => {
-				mockPrompt({
-					text: "Enter a secret value:",
-					type: "password",
-					result: `hunter2
-          `,
-				});
-
-				mockPutRequest({ name: `secret-name`, text: `hunter2` });
-				await runWrangler("secret put secret-name --name script-name");
-				expect(std.out).toMatchInlineSnapshot(`
-          "🌀 Creating the secret for script script-name
-          ✨ Success! Uploaded secret secret-name"
-        `);
-			});
-
-			it("should create a secret", async () => {
-				mockPrompt({
-					text: "Enter a secret value:",
-					type: "password",
-					result: "the-secret",
-				});
-
-				mockPutRequest({ name: "the-key", text: "the-secret" });
-				await runWrangler("secret put the-key --name script-name");
-
-				expect(std.out).toMatchInlineSnapshot(`
-                  "🌀 Creating the secret for script script-name
-                  ✨ Success! Uploaded secret the-key"
-              `);
-				expect(std.err).toMatchInlineSnapshot(`""`);
-			});
-
-			it("should create a secret: legacy envs", async () => {
-				mockPrompt({
-					text: "Enter a secret value:",
-					type: "password",
-					result: "the-secret",
-				});
-
-				mockPutRequest(
-					{ name: "the-key", text: "the-secret" },
-					"some-env",
-					true
-				);
-				await runWrangler(
-					"secret put the-key --name script-name --env some-env --legacy-env"
-				);
-
-				expect(std.out).toMatchInlineSnapshot(`
-                  "🌀 Creating the secret for script script-name-some-env
-                  ✨ Success! Uploaded secret the-key"
-              `);
-				expect(std.err).toMatchInlineSnapshot(`""`);
-			});
-
-			it("should create a secret: service envs", async () => {
-				mockPrompt({
-					text: "Enter a secret value:",
-					type: "password",
-					result: "the-secret",
-				});
-
-				mockPutRequest(
-					{ name: "the-key", text: "the-secret" },
-					"some-env",
-					false
-				);
-				await runWrangler(
-					"secret put the-key --name script-name --env some-env --legacy-env false"
-				);
-
-				expect(std.out).toMatchInlineSnapshot(`
-                  "🌀 Creating the secret for script script-name (some-env)
-                  ✨ Success! Uploaded secret the-key"
-              `);
-				expect(std.err).toMatchInlineSnapshot(`""`);
-			});
-
-			it("should error without a script name", async () => {
-				let error: Error | undefined;
-				try {
-					await runWrangler("secret put the-key");
-				} catch (e) {
-					error = e as Error;
-				}
-				expect(std.out).toMatchInlineSnapshot(`
-          "
-          If you think this is a bug then please create an issue at https://github.com/cloudflare/wrangler2/issues/new/choose"
-        `);
-				expect(std.err).toMatchInlineSnapshot(`
-          "X [ERROR] Missing script name
-
-          "
-        `);
-				expect(error).toMatchInlineSnapshot(`[Error: Missing script name]`);
-			});
-		});
-
-		describe("non-interactive", () => {
-			const mockStdIn = useMockStdin({ isTTY: false });
-
-			it("should trim stdin secret value, from piped input", async () => {
-				mockPutRequest({ name: "the-key", text: "the-secret" });
-				// Pipe the secret in as three chunks to test that we reconstitute it correctly.
-				mockStdIn.send(
-					`the`,
-					`-`,
-					`secret
-          ` // whitespace & newline being removed
-				);
-				await runWrangler("secret put the-key --name script-name");
-
-				expect(std.out).toMatchInlineSnapshot(`
-          "🌀 Creating the secret for script script-name
-          ✨ Success! Uploaded secret the-key"
-        `);
-				expect(std.warn).toMatchInlineSnapshot(`""`);
-				expect(std.err).toMatchInlineSnapshot(`""`);
-			});
-
-			it("should create a secret, from piped input", async () => {
-				mockPutRequest({ name: "the-key", text: "the-secret" });
-				// Pipe the secret in as three chunks to test that we reconstitute it correctly.
-				mockStdIn.send("the", "-", "secret");
-				await runWrangler("secret put the-key --name script-name");
-
-				expect(std.out).toMatchInlineSnapshot(`
-          "🌀 Creating the secret for script script-name
-          ✨ Success! Uploaded secret the-key"
-        `);
-				expect(std.warn).toMatchInlineSnapshot(`""`);
-				expect(std.err).toMatchInlineSnapshot(`""`);
-			});
-
-			it("should error if the piped input fails", async () => {
-				mockPutRequest({ name: "the-key", text: "the-secret" });
-				mockStdIn.throwError(new Error("Error in stdin stream"));
-				await expect(
-					runWrangler("secret put the-key --name script-name")
-				).rejects.toThrowErrorMatchingInlineSnapshot(`"Error in stdin stream"`);
-
-				expect(std.out).toMatchInlineSnapshot(`
-          "
-          If you think this is a bug then please create an issue at https://github.com/cloudflare/wrangler2/issues/new/choose"
-        `);
-				expect(std.warn).toMatchInlineSnapshot(`""`);
-			});
-
-			describe("with accountId", () => {
-				mockAccountId({ accountId: null });
-
-				it("should error if request for memberships fails", async () => {
-					mockGetMembershipsFail();
-					await expect(
-						runWrangler("secret put the-key --name script-name")
-					).rejects.toThrowErrorMatchingInlineSnapshot(
-						`"A request to the Cloudflare API (/memberships) failed."`
-					);
-				});
-
-				it("should error if a user has no account", async () => {
-					mockGetMemberships([]);
-					await expect(runWrangler("secret put the-key --name script-name"))
-						.rejects.toThrowErrorMatchingInlineSnapshot(`
-                  "Failed to automatically retrieve account IDs for the logged in user.
-                  In a non-interactive environment, it is mandatory to specify an account ID, either by assigning its value to CLOUDFLARE_ACCOUNT_ID, or as \`account_id\` in your \`wrangler.toml\` file."
-                `);
-				});
-
-				it("should use the account from wrangler.toml", async () => {
-					fs.writeFileSync(
-						"wrangler.toml",
-						TOML.stringify({
-							account_id: "some-account-id",
-						}),
-						"utf-8"
-					);
-					mockStdIn.send("the-secret");
-					mockPutRequest({ name: "the-key", text: "the-secret" });
-					await runWrangler("secret put the-key --name script-name");
-					expect(std.out).toMatchInlineSnapshot(`
-            "🌀 Creating the secret for script script-name
-            ✨ Success! Uploaded secret the-key"
-          `);
-					expect(std.warn).toMatchInlineSnapshot(`""`);
-					expect(std.err).toMatchInlineSnapshot(`""`);
-				});
-
-				it("should error if a user has multiple accounts, and has not specified an account in wrangler.toml", async () => {
-					mockGetMemberships([
-						{
-							id: "1",
-							account: { id: "account-id-1", name: "account-name-1" },
-						},
-						{
-							id: "2",
-							account: { id: "account-id-2", name: "account-name-2" },
-						},
-						{
-							id: "3",
-							account: { id: "account-id-3", name: "account-name-3" },
-						},
-					]);
-
-					await expect(runWrangler("secret put the-key --name script-name"))
-						.rejects.toThrowErrorMatchingInlineSnapshot(`
-                  "More than one account available but unable to select one in non-interactive mode.
-                  Please set the appropriate \`account_id\` in your \`wrangler.toml\` file.
-                  Available accounts are (\\"<name>\\" - \\"<id>\\"):
-                    \\"account-name-1\\" - \\"account-id-1\\")
-                    \\"account-name-2\\" - \\"account-id-2\\")
-                    \\"account-name-3\\" - \\"account-id-3\\")"
-                `);
-				});
-			});
-		});
-	});
-
-	describe("delete", () => {
-		function mockDeleteRequest(
-			input: {
-				scriptName: string;
-				secretName: string;
-			},
-			env?: string,
-			legacyEnv = false
-		) {
-			const servicesOrScripts = env && !legacyEnv ? "services" : "scripts";
-			const environment = env && !legacyEnv ? "/environments/:envName" : "";
-			setMockResponse(
-				`/accounts/:accountId/workers/${servicesOrScripts}/:scriptName${environment}/secrets/:secretName`,
-				"DELETE",
-				([_url, accountId, scriptName, envOrSecretName, secretName]) => {
-					expect(accountId).toEqual("some-account-id");
-					expect(scriptName).toEqual(
-						legacyEnv && env ? `script-name-${env}` : "script-name"
-					);
-					if (!legacyEnv) {
-						if (env) {
-							expect(envOrSecretName).toEqual(env);
-							expect(secretName).toEqual(input.secretName);
-						} else {
-							expect(envOrSecretName).toEqual(input.secretName);
-						}
-					} else {
-						expect(envOrSecretName).toEqual(input.secretName);
-					}
-					return null;
-				}
-			);
-		}
-		it("should delete a secret", async () => {
-			mockDeleteRequest({ scriptName: "script-name", secretName: "the-key" });
-			mockConfirm({
-				text: "Are you sure you want to permanently delete the variable the-key on the script script-name?",
-				result: true,
-			});
-			await runWrangler("secret delete the-key --name script-name");
-			expect(std.out).toMatchInlineSnapshot(`
-        "🌀 Deleting the secret the-key on script script-name
-        ✨ Success! Deleted secret the-key"
-      `);
-			expect(std.err).toMatchInlineSnapshot(`""`);
-		});
-
-		it("should delete a secret: legacy envs", async () => {
-			mockDeleteRequest(
-				{ scriptName: "script-name", secretName: "the-key" },
-				"some-env",
-				true
-			);
-			mockConfirm({
-				text: "Are you sure you want to permanently delete the variable the-key on the script script-name-some-env?",
-				result: true,
-			});
-			await runWrangler(
-				"secret delete the-key --name script-name --env some-env --legacy-env"
-			);
-			expect(std.out).toMatchInlineSnapshot(`
-        "🌀 Deleting the secret the-key on script script-name-some-env
-        ✨ Success! Deleted secret the-key"
-      `);
-			expect(std.err).toMatchInlineSnapshot(`""`);
-		});
-
-		it("should delete a secret: service envs", async () => {
-			mockDeleteRequest(
-				{ scriptName: "script-name", secretName: "the-key" },
-				"some-env"
-			);
-			mockConfirm({
-				text: "Are you sure you want to permanently delete the variable the-key on the script script-name (some-env)?",
-				result: true,
-			});
-			await runWrangler(
-				"secret delete the-key --name script-name --env some-env --legacy-env false"
-			);
-			expect(std.out).toMatchInlineSnapshot(`
-        "🌀 Deleting the secret the-key on script script-name (some-env)
-        ✨ Success! Deleted secret the-key"
-      `);
-			expect(std.err).toMatchInlineSnapshot(`""`);
-		});
-
-		it("should error without a script name", async () => {
-			let error: Error | undefined;
-
-			try {
-				await runWrangler("secret delete the-key");
-			} catch (e) {
-				error = e as Error;
-			}
-			expect(std.out).toMatchInlineSnapshot(`
-        "
-        If you think this is a bug then please create an issue at https://github.com/cloudflare/wrangler2/issues/new/choose"
-      `);
-			expect(std.err).toMatchInlineSnapshot(`
-        "X [ERROR] Missing script name
-
-        "
-      `);
-			expect(error).toMatchInlineSnapshot(`[Error: Missing script name]`);
-		});
-	});
-
-	describe("list", () => {
-		function mockListRequest(
-			input: { scriptName: string },
-			env?: string,
-			legacyEnv = false
-		) {
-			const servicesOrScripts = env && !legacyEnv ? "services" : "scripts";
-			const environment = env && !legacyEnv ? "/environments/:envName" : "";
-			setMockResponse(
-				`/accounts/:accountId/workers/${servicesOrScripts}/:scriptName${environment}/secrets`,
-				"GET",
-				([_url, accountId, scriptName, envName]) => {
-					expect(accountId).toEqual("some-account-id");
-					expect(scriptName).toEqual(
-						legacyEnv && env ? `script-name-${env}` : "script-name"
-					);
-					if (!legacyEnv) {
-						expect(envName).toEqual(env);
-					}
-
-					return [
-						{
-							name: "the-secret-name",
-							type: "secret_text",
-						},
-					];
-				}
-			);
-		}
-
-		it("should list secrets", async () => {
-			mockListRequest({ scriptName: "script-name" });
-			await runWrangler("secret list --name script-name");
-			expect(std.out).toMatchInlineSnapshot(`
-        "[
-          {
-            \\"name\\": \\"the-secret-name\\",
-            \\"type\\": \\"secret_text\\"
-          }
-        ]"
-      `);
-			expect(std.err).toMatchInlineSnapshot(`""`);
-		});
-
-		it("should list secrets: legacy envs", async () => {
-			mockListRequest({ scriptName: "script-name" }, "some-env", true);
-			await runWrangler(
-				"secret list --name script-name --env some-env --legacy-env"
-			);
-			expect(std.out).toMatchInlineSnapshot(`
-        "[
-          {
-            \\"name\\": \\"the-secret-name\\",
-            \\"type\\": \\"secret_text\\"
-          }
-        ]"
-      `);
-			expect(std.err).toMatchInlineSnapshot(`""`);
-		});
-
-		it("should list secrets: service envs", async () => {
-			mockListRequest({ scriptName: "script-name" }, "some-env");
-			await runWrangler(
-				"secret list --name script-name --env some-env --legacy-env false"
-			);
-			expect(std.out).toMatchInlineSnapshot(`
-        "[
-          {
-            \\"name\\": \\"the-secret-name\\",
-            \\"type\\": \\"secret_text\\"
-          }
-        ]"
-      `);
-			expect(std.err).toMatchInlineSnapshot(`""`);
-		});
-
-		it("should error without a script name", async () => {
-			let error: Error | undefined;
-			try {
-				await runWrangler("secret list");
-			} catch (e) {
-				error = e as Error;
-			}
-			expect(std.out).toMatchInlineSnapshot(`
-        "
-        If you think this is a bug then please create an issue at https://github.com/cloudflare/wrangler2/issues/new/choose"
-      `);
-			expect(std.err).toMatchInlineSnapshot(`
-        "X [ERROR] Missing script name
-
-        "
-      `);
-			expect(error).toMatchInlineSnapshot(`[Error: Missing script name]`);
-		});
-	});
-});