wrangler 0.0.0-e1c2f5b9 → 0.0.0-e1d50407

package/src/__tests__/middleware.test.ts

@@ -1,750 +0,0 @@
-import * as fs from "node:fs";
-import { unstable_dev } from "../api";
-import { runInTempDir } from "./helpers/run-in-tmp";
-
-jest.unmock("undici");
-
-describe("workers change behaviour with middleware with wrangler dev", () => {
-	runInTempDir();
-
-	process.env.EXPERIMENTAL_MIDDLEWARE = "true";
-
-	describe("module workers", () => {
-		it("should register a middleware and intercept", async () => {
-			const scriptContent = `
-			const middleware = async (request, env, _ctx, middlewareCtx) => {
-				const response = await middlewareCtx.next(request, env);
-				const text = await response.text();
-				return new Response(text + ' world');
-			}
-
-			export default {
-				middleware: [middleware],
-				fetch(request, env, ctx) {
-					return new Response('Hello');
-				}
-			};
-			`;
-			fs.writeFileSync("index.js", scriptContent);
-
-			const worker = await unstable_dev("index.js", {
-				experimental: {
-					disableExperimentalWarning: true,
-					disableDevRegistry: true,
-				},
-			});
-
-			const resp = await worker.fetch();
-			let text;
-			if (resp) text = await resp.text();
-			expect(text).toMatchInlineSnapshot(`"Hello world"`);
-			await worker.stop();
-		});
-
-		it("should be able to access scheduled workers from middleware", async () => {
-			const scriptContent = `
-			const middleware = async (request, env, _ctx, middlewareCtx) => {
-				await middlewareCtx.dispatch("scheduled", { cron: "* * * * *" });
-				return new Response("OK");
-			}
-
-			export default {
-				middleware: [middleware],
-				scheduled(controller, env, ctx) {
-					console.log("Scheduled worker called");
-				}
-			}
-			`;
-
-			fs.writeFileSync("index.js", scriptContent);
-
-			const worker = await unstable_dev("index.js", {
-				experimental: {
-					disableExperimentalWarning: true,
-					disableDevRegistry: true,
-				},
-			});
-
-			const resp = await worker.fetch();
-			let text;
-			if (resp) text = await resp.text();
-			expect(text).toMatchInlineSnapshot(`"OK"`);
-			await worker.stop();
-		});
-
-		it("should trigger an error in a scheduled work from middleware", async () => {
-			const scriptContent = `
-			const middleware = async (request, env, _ctx, middlewareCtx) => {
-				try {
-					await middlewareCtx.dispatch("scheduled", { cron: "* * * * *" });
-				} catch (e) {
-					return new Response(e.message);
-				}
-			}
-
-			export default {
-				middleware: [middleware],
-				scheduled(controller, env, ctx) {
-					throw new Error("Error in scheduled worker");
-				}
-			}
-			`;
-
-			fs.writeFileSync("index.js", scriptContent);
-
-			const worker = await unstable_dev("index.js", {
-				experimental: {
-					disableExperimentalWarning: true,
-					disableDevRegistry: true,
-				},
-			});
-
-			const resp = await worker.fetch();
-			let text;
-			if (resp) text = await resp.text();
-			expect(text).toMatchInlineSnapshot(`"Error in scheduled worker"`);
-			await worker.stop();
-		});
-	});
-
-	describe("service workers", () => {
-		it("should register a middleware and intercept using addMiddleware", async () => {
-			const scriptContent = `
-			const middleware = async (request, env, _ctx, middlewareCtx) => {
-				const response = await middlewareCtx.next(request, env);
-				const text = await response.text();
-				return new Response(text + ' world');
-			}
-
-			addMiddleware(middleware);
-
-			addEventListener("fetch", (event) => {
-				event.respondWith(new Response('Hello'));
-			});
-			`;
-			fs.writeFileSync("index.js", scriptContent);
-
-			const worker = await unstable_dev("index.js", {
-				experimental: {
-					disableExperimentalWarning: true,
-					disableDevRegistry: true,
-				},
-			});
-
-			const resp = await worker.fetch();
-			let text;
-			if (resp) text = await resp.text();
-			expect(text).toMatchInlineSnapshot(`"Hello world"`);
-			await worker.stop();
-		});
-
-		it("should register a middleware and intercept using addMiddlewareInternal", async () => {
-			const scriptContent = `
-			const middleware = async (request, env, _ctx, middlewareCtx) => {
-				const response = await middlewareCtx.next(request, env);
-				const text = await response.text();
-				return new Response(text + ' world');
-			}
-
-			addMiddlewareInternal(middleware);
-
-			addEventListener("fetch", (event) => {
-				event.respondWith(new Response('Hello'));
-			});
-			`;
-			fs.writeFileSync("index.js", scriptContent);
-
-			const worker = await unstable_dev("index.js", {
-				experimental: {
-					disableExperimentalWarning: true,
-					disableDevRegistry: true,
-				},
-			});
-
-			const resp = await worker.fetch();
-			let text;
-			if (resp) text = await resp.text();
-			expect(text).toMatchInlineSnapshot(`"Hello world"`);
-			await worker.stop();
-		});
-
-		it("should be able to access scheduled workers from middleware", async () => {
-			const scriptContent = `
-			const middleware = async (request, env, _ctx, middlewareCtx) => {
-				await middlewareCtx.dispatch("scheduled", { cron: "* * * * *" });
-				return new Response("OK");
-			}
-
-			addMiddleware(middleware);
-
-			addEventListener("scheduled", (event) => {
-				console.log("Scheduled worker called");
-			});
-			`;
-
-			fs.writeFileSync("index.js", scriptContent);
-
-			const worker = await unstable_dev("index.js", {
-				experimental: {
-					disableExperimentalWarning: true,
-					disableDevRegistry: true,
-				},
-			});
-
-			const resp = await worker.fetch();
-			let text;
-			if (resp) text = await resp.text();
-			expect(text).toMatchInlineSnapshot(`"OK"`);
-			await worker.stop();
-		});
-
-		it("should trigger an error in a scheduled work from middleware", async () => {
-			const scriptContent = `
-				const middleware = async (request, env, _ctx, middlewareCtx) => {
-					try {
-						await middlewareCtx.dispatch("scheduled", { cron: "* * * * *" });
-					} catch (e) {
-						return new Response(e.message);
-					}
-				}
-
-				addMiddleware(middleware);
-
-				addEventListener("scheduled", (event) => {
-					throw new Error("Error in scheduled worker");
-				});
-				`;
-
-			fs.writeFileSync("index.js", scriptContent);
-
-			const worker = await unstable_dev("index.js", {
-				experimental: {
-					disableExperimentalWarning: true,
-					disableDevRegistry: true,
-				},
-			});
-
-			const resp = await worker.fetch();
-			let text;
-			if (resp) text = await resp.text();
-			expect(text).toMatchInlineSnapshot(`"Error in scheduled worker"`);
-			await worker.stop();
-		});
-	});
-});
-
-describe("unchanged functionality when wrapping with middleware", () => {
-	runInTempDir();
-
-	process.env.EXPERIMENTAL_MIDDLEWARE = "true";
-
-	describe("module workers", () => {
-		it("should return Hello World with no middleware export", async () => {
-			const scriptContent = `
-			export default {
-				fetch(request, env, ctx) {
-					return new Response("Hello world");
-				}
-			};
-			`;
-			fs.writeFileSync("index.js", scriptContent);
-
-			const worker = await unstable_dev("index.js", {
-				experimental: {
-					disableExperimentalWarning: true,
-					disableDevRegistry: true,
-				},
-			});
-
-			const resp = await worker.fetch();
-			if (resp) {
-				const text = await resp.text();
-				expect(text).toMatchInlineSnapshot(`"Hello world"`);
-			}
-			await worker.stop();
-		});
-
-		it("should return hello world with empty middleware array", async () => {
-			const scriptContent = `
-			export default {
-				middleware: [],
-				fetch() {
-					return new Response("Hello world");
-				}
-			}
-			`;
-			fs.writeFileSync("index.js", scriptContent);
-
-			const worker = await unstable_dev("index.js", {
-				experimental: {
-					disableExperimentalWarning: true,
-					disableDevRegistry: true,
-				},
-			});
-
-			const resp = await worker.fetch();
-			let text;
-			if (resp) text = await resp.text();
-			expect(text).toMatchInlineSnapshot(`"Hello world"`);
-			await worker.stop();
-		});
-
-		it("should return hello world passing through middleware", async () => {
-			const scriptContent = `
-			const middleware = async (request, env, _ctx, middlewareCtx) => {
-				return middlewareCtx.next(request, env);
-			}
-
-			export default {
-				middleware: [middleware],
-				fetch(request, env, ctx) {
-					return new Response("Hello world");
-				}
-			}
-			`;
-			fs.writeFileSync("index.js", scriptContent);
-
-			const worker = await unstable_dev("index.js", {
-				experimental: {
-					disableExperimentalWarning: true,
-					disableDevRegistry: true,
-				},
-			});
-
-			const resp = await worker.fetch();
-			if (resp) {
-				const text = await resp.text();
-				expect(text).toMatchInlineSnapshot(`"Hello world"`);
-			}
-			await worker.stop();
-		});
-
-		it("should return hello world with multiple middleware in array", async () => {
-			const scriptContent = `
-			const middleware = async (request, env, _ctx, middlewareCtx) => {
-				return middlewareCtx.next(request, env);
-			}
-			const middleware2 = async (request, env, _ctx, middlewareCtx) => {
-				return middlewareCtx.next(request, env);
-			}
-
-
-			export default {
-				middleware: [middleware, middleware2],
-				fetch() {
-					return new Response("Hello world");
-				}
-			}
-			`;
-			fs.writeFileSync("index.js", scriptContent);
-
-			const worker = await unstable_dev("index.js", {
-				experimental: {
-					disableExperimentalWarning: true,
-					disableDevRegistry: true,
-				},
-			});
-
-			const resp = await worker.fetch();
-			let text;
-			if (resp) text = await resp.text();
-			expect(text).toMatchInlineSnapshot(`"Hello world"`);
-			await worker.stop();
-		});
-
-		it("should leave response headers unchanged with middleware", async () => {
-			const scriptContent = `
-			const middleware = async (request, env, _ctx, middlewareCtx) => {
-				return middlewareCtx.next(request, env);
-			}
-
-			export default {
-				middleware: [middleware],
-				fetch() {
-					return new Response("Hello world", { status: 500, headers: { "x-test": "test" } });
-				}
-			}
-			`;
-			fs.writeFileSync("index.js", scriptContent);
-
-			const worker = await unstable_dev("index.js", {
-				experimental: {
-					disableExperimentalWarning: true,
-					disableDevRegistry: true,
-				},
-			});
-
-			const resp = await worker.fetch();
-			const status = resp?.status;
-			let text;
-			if (resp) text = await resp.text();
-			const testHeader = resp?.headers.get("x-test");
-			expect(status).toEqual(500);
-			expect(text).toMatchInlineSnapshot(`"Hello world"`);
-			expect(testHeader).toEqual("test");
-			await worker.stop();
-		});
-
-		it("waitUntil should not block responses", async () => {
-			const scriptContent = `
-			const middleware = async (request, env, _ctx, middlewareCtx) => {
-				return middlewareCtx.next(request, env);
-			}
-
-			export default {
-				middleware: [middleware],
-				async fetch(request, env, ctx) {
-					let count = 0;
-					ctx.waitUntil(new Promise(resolve => {
-						setTimeout(() => {
-							count += 1;
-							console.log("waitUntil", count);
-							resolve()
-						}, 1000);
-					}));
-					return new Response("Hello world" + String(count));
-				}
-			}
-			`;
-			fs.writeFileSync("index.js", scriptContent);
-
-			const worker = await unstable_dev("index.js", {
-				experimental: {
-					disableExperimentalWarning: true,
-					disableDevRegistry: true,
-				},
-			});
-
-			const resp = await worker.fetch();
-			let text;
-			if (resp) text = await resp.text();
-			expect(text).toMatchInlineSnapshot(`"Hello world0"`);
-			await worker.stop();
-		});
-	});
-
-	describe("service workers", () => {
-		it("should return Hello World with no middleware export", async () => {
-			const scriptContent = `
-			addEventListener("fetch", (event) => {
-				event.respondWith(new Response("Hello world"));
-			});
-			`;
-			fs.writeFileSync("index.js", scriptContent);
-
-			const worker = await unstable_dev("index.js", {
-				experimental: {
-					disableExperimentalWarning: true,
-					disableDevRegistry: true,
-				},
-			});
-
-			const resp = await worker.fetch();
-			if (resp) {
-				const text = await resp.text();
-				expect(text).toMatchInlineSnapshot(`"Hello world"`);
-			}
-			await worker.stop();
-		});
-
-		it("should return hello world with empty middleware array", async () => {
-			const scriptContent = `
-			addMiddleware([]);
-
-			addEventListener("fetch", (event) => {
-				event.respondWith(new Response("Hello world"));
-			});
-			`;
-			fs.writeFileSync("index.js", scriptContent);
-
-			const worker = await unstable_dev("index.js", {
-				experimental: {
-					disableExperimentalWarning: true,
-					disableDevRegistry: true,
-				},
-			});
-
-			const resp = await worker.fetch();
-			let text;
-			if (resp) text = await resp.text();
-			expect(text).toMatchInlineSnapshot(`"Hello world"`);
-			await worker.stop();
-		});
-
-		it("should return hello world passing through middleware", async () => {
-			const scriptContent = `
-			const middleware = async (request, env, _ctx, middlewareCtx) => {
-				return middlewareCtx.next(request, env);
-			}
-
-			addMiddleware(middleware);
-
-			addEventListener("fetch", (event) => {
-				event.respondWith(new Response("Hello world"));
-			});
-			`;
-			fs.writeFileSync("index.js", scriptContent);
-
-			const worker = await unstable_dev("index.js", {
-				experimental: {
-					disableExperimentalWarning: true,
-					disableDevRegistry: true,
-				},
-			});
-
-			const resp = await worker.fetch();
-			if (resp) {
-				const text = await resp.text();
-				expect(text).toMatchInlineSnapshot(`"Hello world"`);
-			}
-			await worker.stop();
-		});
-
-		it("should return hello world with addMiddleware function called multiple times", async () => {
-			const scriptContent = `
-			const middleware = async (request, env, _ctx, middlewareCtx) => {
-				return middlewareCtx.next(request, env);
-			}
-			const middleware2 = async (request, env, _ctx, middlewareCtx) => {
-				return middlewareCtx.next(request, env);
-			}
-
-			addMiddleware(middleware);
-			addMiddleware(middleware2);
-
-			addEventListener("fetch", (event) => {
-				event.respondWith(new Response("Hello world"));
-			});
-			`;
-			fs.writeFileSync("index.js", scriptContent);
-
-			const worker = await unstable_dev("index.js", {
-				experimental: {
-					disableExperimentalWarning: true,
-					disableDevRegistry: true,
-				},
-			});
-
-			const resp = await worker.fetch();
-			let text;
-			if (resp) text = await resp.text();
-			expect(text).toMatchInlineSnapshot(`"Hello world"`);
-			await worker.stop();
-		});
-
-		it("should return hello world with addMiddleware function called with array of middleware", async () => {
-			const scriptContent = `
-			const middleware = async (request, env, _ctx, middlewareCtx) => {
-				return middlewareCtx.next(request, env);
-			}
-			const middleware2 = async (request, env, _ctx, middlewareCtx) => {
-				return middlewareCtx.next(request, env);
-			}
-
-			addMiddleware(middleware, middleware2);
-
-			addEventListener("fetch", (event) => {
-				event.respondWith(new Response("Hello world"));
-			});
-			`;
-			fs.writeFileSync("index.js", scriptContent);
-
-			const worker = await unstable_dev("index.js", {
-				experimental: {
-					disableExperimentalWarning: true,
-					disableDevRegistry: true,
-				},
-			});
-
-			const resp = await worker.fetch();
-			let text;
-			if (resp) text = await resp.text();
-			expect(text).toMatchInlineSnapshot(`"Hello world"`);
-			await worker.stop();
-		});
-
-		it("should return hello world with addMiddlewareInternal function called multiple times", async () => {
-			const scriptContent = `
-			const middleware = async (request, env, _ctx, middlewareCtx) => {
-				return middlewareCtx.next(request, env);
-			}
-			const middleware2 = async (request, env, _ctx, middlewareCtx) => {
-				return middlewareCtx.next(request, env);
-			}
-
-			addMiddlewareInternal(middleware);
-			addMiddlewareInternal(middleware2);
-
-			addEventListener("fetch", (event) => {
-				event.respondWith(new Response("Hello world"));
-			});
-			`;
-			fs.writeFileSync("index.js", scriptContent);
-
-			const worker = await unstable_dev("index.js", {
-				experimental: {
-					disableExperimentalWarning: true,
-					disableDevRegistry: true,
-				},
-			});
-
-			const resp = await worker.fetch();
-			let text;
-			if (resp) text = await resp.text();
-			expect(text).toMatchInlineSnapshot(`"Hello world"`);
-			await worker.stop();
-		});
-
-		it("should return hello world with addMiddlewareInternal function called with array of middleware", async () => {
-			const scriptContent = `
-			const middleware = async (request, env, _ctx, middlewareCtx) => {
-				return middlewareCtx.next(request, env);
-			}
-			const middleware2 = async (request, env, _ctx, middlewareCtx) => {
-				return middlewareCtx.next(request, env);
-			}
-
-			addMiddlewareInternal(middleware, middleware2);
-
-			addEventListener("fetch", (event) => {
-				event.respondWith(new Response("Hello world"));
-			});
-			`;
-			fs.writeFileSync("index.js", scriptContent);
-
-			const worker = await unstable_dev("index.js", {
-				experimental: {
-					disableExperimentalWarning: true,
-					disableDevRegistry: true,
-				},
-			});
-
-			const resp = await worker.fetch();
-			let text;
-			if (resp) text = await resp.text();
-			expect(text).toMatchInlineSnapshot(`"Hello world"`);
-			await worker.stop();
-		});
-
-		it("should return hello world with both addMiddleware and addMiddlewareInternal called", async () => {
-			const scriptContent = `
-			const middleware = async (request, env, _ctx, middlewareCtx) => {
-				return middlewareCtx.next(request, env);
-			}
-			const middleware2 = async (request, env, _ctx, middlewareCtx) => {
-				return middlewareCtx.next(request, env);
-			}
-
-			addMiddleware(middleware);
-			addMiddlewareInternal(middleware2);
-
-			addEventListener("fetch", (event) => {
-				event.respondWith(new Response("Hello world"));
-			});
-			`;
-			fs.writeFileSync("index.js", scriptContent);
-
-			const worker = await unstable_dev("index.js", {
-				experimental: {
-					disableExperimentalWarning: true,
-					disableDevRegistry: true,
-				},
-			});
-
-			const resp = await worker.fetch();
-			let text;
-			if (resp) text = await resp.text();
-			expect(text).toMatchInlineSnapshot(`"Hello world"`);
-			await worker.stop();
-		});
-
-		it("should leave response headers unchanged with middleware", async () => {
-			const scriptContent = `
-			const middleware = async (request, env, _ctx, middlewareCtx) => {
-				return middlewareCtx.next(request, env);
-			}
-
-			addEventListener("fetch", (event) => {
-				event.respondWith(new Response("Hello world", { status: 500, headers: { "x-test": "test" } }));
-			});
-			`;
-			fs.writeFileSync("index.js", scriptContent);
-
-			const worker = await unstable_dev("index.js", {
-				experimental: {
-					disableExperimentalWarning: true,
-					disableDevRegistry: true,
-				},
-			});
-
-			const resp = await worker.fetch();
-			const status = resp?.status;
-			let text;
-			if (resp) text = await resp.text();
-			const testHeader = resp?.headers.get("x-test");
-			expect(status).toEqual(500);
-			expect(text).toMatchInlineSnapshot(`"Hello world"`);
-			expect(testHeader).toEqual("test");
-			await worker.stop();
-		});
-
-		it("should allow multiple addEventListeners for fetch", async () => {
-			const scriptContent = `
-			let count = 0;
-
-			addEventListener("fetch", (event) => {
-				count += 1;
-			});
-
-			addEventListener("fetch", (event) => {
-				event.respondWith(new Response("Hello world" + String(count)));
-			});
-			`;
-			fs.writeFileSync("index.js", scriptContent);
-
-			const worker = await unstable_dev("index.js", {
-				experimental: {
-					disableExperimentalWarning: true,
-					disableDevRegistry: true,
-				},
-			});
-
-			const resp = await worker.fetch();
-			let text;
-			if (resp) text = await resp.text();
-			expect(text).toMatchInlineSnapshot(`"Hello world1"`);
-			await worker.stop();
-		});
-
-		it("waitUntil should not block responses", async () => {
-			const scriptContent = `
-			addEventListener("fetch", (event) => {
-
-				let count = 0;
-				event.waitUntil(new Promise((resolve) => {
-					setTimeout(() => {
-						count +=1;
-						console.log('waitUntil', count);
-						resolve();
-					}, 1000);
-				}));
-				event.respondWith(new Response("Hello world" + String(count)));
-			});
-			`;
-			fs.writeFileSync("index.js", scriptContent);
-
-			const worker = await unstable_dev("index.js", {
-				experimental: {
-					disableExperimentalWarning: true,
-					disableDevRegistry: true,
-				},
-			});
-
-			const resp = await worker.fetch();
-			let text;
-			if (resp) text = await resp.text();
-			expect(text).toMatchInlineSnapshot(`"Hello world0"`);
-			await worker.stop();
-		});
-	});
-});