Add more tests which fail right now. sigh

Author: davidbarratt

workers/handler/test/purge-tag.spec.ts

@@ -2,10 +2,6 @@ import { randomBytes } from "node:crypto";
 import { env, SELF } from "cloudflare:test";
 import { it, expect, vi } from "vitest";
 
-// For now, you'll need to do something like this to get a correctly-typed
-// `Request` to pass to `worker.fetch()`.
-// const IncomingRequest = Request<unknown, IncomingRequestCfProperties>;
-
 interface CacheTagPurge {
   tag: string;
   zone?: string;

workers/handler/test/purge-url.spec.ts

@@ -0,0 +1,63 @@
+import { randomBytes } from "node:crypto";
+import { SELF } from "cloudflare:test";
+import { it, expect, vi, afterEach } from "vitest";
+import type Cloudflare from "cloudflare";
+
+const zoneList = vi.fn<
+  Parameters<typeof Cloudflare.prototype.zones.list>,
+  Promise<{ result: { id: string }[] }>
+>();
+
+const cachePurge = vi.fn<
+  Parameters<typeof Cloudflare.prototype.cache.purge>,
+  Promise<void>
+>();
+
+interface CacheUrlPurge {
+  url: string;
+  zone: string;
+}
+
+vi.mock("cloudflare", () => ({
+  default: vi.fn().mockImplementation(
+    () =>
+      console.log("HELLO") || {
+        zones: {
+          list: zoneList,
+        },
+        cache: {
+          purge: cachePurge,
+        },
+      },
+  ),
+}));
+
+afterEach(() => {
+  vi.clearAllMocks();
+});
+
+it("purges a URL", async () => {
+  const url = "https://example.com";
+  const zone = "example.com";
+  const id = "test";
+
+  zoneList.mockResolvedValueOnce({
+    result: [{ id }],
+  });
+  cachePurge.mockResolvedValueOnce();
+
+  const messages: ServiceBindingQueueMessage<CacheUrlPurge>[] = [
+    {
+      id: randomBytes(16).toString("hex"),
+      timestamp: new Date(1000),
+      attempts: 1,
+      body: { url, zone },
+    },
+  ];
+
+  const { outcome } = await SELF.queue("cache-purge-url", messages);
+  expect(outcome).toBe("ok");
+
+  expect(zoneList).toBeCalledWith({ name: zone });
+  expect(cachePurge).toBeCalledWith({ zone_id: id, files: [url] });
+});

workers/handler/vitest.config.mts

@@ -1,34 +1,34 @@
 import path from "node:path";
 import {
-	defineWorkersProject,
-	readD1Migrations,
+  defineWorkersProject,
+  readD1Migrations,
 } from "@cloudflare/vitest-pool-workers/config";
 
 export default defineWorkersProject(async () => {
-	// Read all migrations in the `migrations` directory
-	const migrationsPath = path.join(__dirname, "migrations");
-	const migrations = await readD1Migrations(migrationsPath);
+  // Read all migrations in the `migrations` directory
+  const migrationsPath = path.join(__dirname, "migrations");
+  const migrations = await readD1Migrations(migrationsPath);
 
-	return {
-		test: {
-			setupFiles: ["./test/apply-migrations.ts"],
-			poolOptions: {
-				workers: {
-					singleWorker: true,
-					wrangler: {
-						configPath: "./wrangler.toml",
-						// environment: "production",
-					},
-					miniflare: {
-						// Required to use `SELF.queue()`. This is an experimental
-						// compatibility flag, and cannot be enabled in production.
-						compatibilityFlags: ["service_binding_extra_handlers"],
-						// Add a test-only binding for migrations, so we can apply them in a
-						// setup file
-						bindings: { TEST_MIGRATIONS: migrations },
-					},
-				},
-			},
-		},
-	};
+  return {
+    test: {
+      setupFiles: ["./test/apply-migrations.ts"],
+      poolOptions: {
+        workers: {
+          singleWorker: true,
+          wrangler: {
+            configPath: "./wrangler.toml",
+            environment: "test",
+          },
+          miniflare: {
+            // Required to use `SELF.queue()`. This is an experimental
+            // compatibility flag, and cannot be enabled in production.
+            compatibilityFlags: ["service_binding_extra_handlers"],
+            // Add a test-only binding for migrations, so we can apply them in a
+            // setup file
+            bindings: { TEST_MIGRATIONS: migrations },
+          },
+        },
+      },
+    },
+  };
 });

workers/handler/wrangler.toml

@@ -115,3 +115,17 @@ max_batch_size = 30
 # [[vectorize]]
 # binding = "MY_INDEX"
 # index_name = "my-index"
+
+[env.test]
+vars = { API_TOKEN = "test" }
+d1_databases = [
+  { binding = "DB", database_name = "cache-tag", database_id = "49a13934-d832-4521-bcdf-7127b8f8f16b" }
+]
+queues.producers = [
+  { binding = "CACHE_PURGE_URL", queue = "cache-purge-url" },
+]
+queues.consumers = [
+  { queue = "cache-capture" },
+  { queue = "cache-purge-tag", max_batch_size = 95  },
+  { queue = "cache-purge-url", max_batch_size = 30 },
+]