chore: add d1 and kv to e2e cleanup job (#7681)

* add d1 and kv to e2e cleanup job

* add d1 and kv to cleanup

* tests

Author: emily-shen

tools/e2e/__tests__/common.test.ts

@@ -1,10 +1,14 @@
 import { getGlobalDispatcher, MockAgent, setGlobalDispatcher } from "undici";
 import { afterEach, beforeEach, describe, it } from "vitest";
 import {
+	deleteDatabase,
+	deleteKVNamespace,
 	deleteProject,
 	deleteWorker,
+	listTmpDatabases,
 	listTmpE2EProjects,
 	listTmpE2EWorkers,
+	listTmpKVNamespaces,
 } from "../common";
 
 const originalAccountID = process.env.CLOUDFLARE_ACCOUNT_ID;
@@ -108,6 +112,154 @@ describe("deleteProject()", () => {
 	});
 });
 
+describe("listTmpKVNamespaces()", () => {
+	it("makes a REST request and returns a filtered list of kv namespaces", async ({
+		expect,
+	}) => {
+		agent
+			.get("https://api.cloudflare.com")
+			.intercept({
+				path: `/client/v4/accounts/${MOCK_CLOUDFLARE_ACCOUNT_ID}/storage/kv/namespaces`,
+				method: "GET",
+				query: {
+					per_page: 10,
+					page: 1,
+					direction: "asc",
+					order: "title",
+				},
+			})
+			.reply(
+				200,
+				JSON.stringify({
+					result: [
+						{ id: "kv-tmp-e2e", title: "kv-1" },
+						{ id: "kv-2", title: "kv-2" },
+						{ id: "tmp_e2e", title: "kv-3" },
+						{ id: "kv-4", title: "kv-4" },
+						{ id: "kv-5", title: "kv-5" },
+						{ id: "kv-6", title: "kv-6" },
+						{ id: "tmp_e2e_kv", title: "kv-7" },
+						{ id: "kv-8", title: "kv-8" },
+						{ id: "kv-9", title: "kv-9" },
+						{ id: "kv-10", title: "kv-10" },
+					],
+				})
+			);
+		agent
+			.get("https://api.cloudflare.com")
+			.intercept({
+				path: `/client/v4/accounts/${MOCK_CLOUDFLARE_ACCOUNT_ID}/storage/kv/namespaces`,
+				method: "GET",
+				query: {
+					per_page: 10,
+					page: 2,
+					direction: "asc",
+					order: "title",
+				},
+			})
+			.reply(
+				200,
+				JSON.stringify({
+					result: [{ id: "kv-tmp-e2e-11", title: "kv-11" }],
+				})
+			);
+
+		const result = await listTmpKVNamespaces();
+
+		expect(result.map((p) => p.id)).toMatchInlineSnapshot(`[]`);
+	});
+});
+
+describe("deleteKVNamespace()", () => {
+	it("makes a REST request to delete the given project", async () => {
+		const MOCK_KV = "tmp_e2e_kv";
+		agent
+			.get("https://api.cloudflare.com")
+			.intercept({
+				path: `/client/v4/accounts/${MOCK_CLOUDFLARE_ACCOUNT_ID}/storage/kv/namespaces/${MOCK_KV}`,
+				method: "DELETE",
+			})
+			.reply(200, JSON.stringify({ result: [] }));
+		await deleteKVNamespace(MOCK_KV);
+	});
+});
+
+describe("listTmpDatabases()", () => {
+	it("makes a REST request and returns a filtered list of d1 databases", async ({
+		expect,
+	}) => {
+		agent
+			.get("https://api.cloudflare.com")
+			.intercept({
+				path: `/client/v4/accounts/${MOCK_CLOUDFLARE_ACCOUNT_ID}/d1/database`,
+				method: "GET",
+				query: {
+					per_page: 10,
+					page: 1,
+				},
+			})
+			.reply(
+				200,
+				JSON.stringify({
+					result: [
+						{ uuid: "1", name: "db-1", created_at: nowStr },
+						{ uuid: "2", name: "db-2", created_at: oldTimeStr },
+						{ uuid: "3", name: "tmp-e2e-db-1", created_at: nowStr },
+						{ uuid: "4", name: "tmp-e2e-db-2", created_at: oldTimeStr },
+						{ uuid: "5", name: "db-3", created_at: nowStr },
+						{ uuid: "6", name: "db-4", created_at: oldTimeStr },
+						{ uuid: "7", name: "tmp-e2e-db-3", created_at: nowStr },
+						{ uuid: "8", name: "tmp-e2e-db-4", created_at: oldTimeStr },
+						{ uuid: "9", name: "db-5", created_at: nowStr },
+						{ uuid: "10", name: "db-6", created_at: oldTimeStr },
+					],
+				})
+			);
+		agent
+			.get("https://api.cloudflare.com")
+			.intercept({
+				path: `/client/v4/accounts/${MOCK_CLOUDFLARE_ACCOUNT_ID}/d1/database`,
+				method: "GET",
+				query: {
+					per_page: 10,
+					page: 2,
+				},
+			})
+			.reply(
+				200,
+				JSON.stringify({
+					result: [
+						{ uuid: "11", name: "db-11", created_at: nowStr },
+						{ uuid: "12", name: "db-12", created_at: oldTimeStr },
+					],
+				})
+			);
+
+		const result = await listTmpDatabases();
+
+		expect(result.map((p) => p.name)).toMatchInlineSnapshot(`
+			[
+			  "tmp-e2e-db-2",
+			  "tmp-e2e-db-4",
+			]
+		`);
+	});
+});
+
+describe("deleteDatabase()", () => {
+	it("makes a REST request to delete the given project", async () => {
+		const MOCK_DB = "tmp-e2e-db";
+		agent
+			.get("https://api.cloudflare.com")
+			.intercept({
+				path: `/client/v4/accounts/${MOCK_CLOUDFLARE_ACCOUNT_ID}/d1/database/${MOCK_DB}`,
+				method: "DELETE",
+			})
+			.reply(200, JSON.stringify({ result: [] }));
+		await deleteDatabase(MOCK_DB);
+	});
+});
+
 describe("listTmpE2EWorkers()", () => {
 	it("makes a REST request and returns a filtered list of workers", async ({
 		expect,