Testing with Durable Objects

import { unstable_dev } from "wrangler"
import type { UnstableDevWorker } from "wrangler"
import { describe, expect, it, beforeAll, afterAll } from "vitest"

describe("Worker", () => {
	let worker: UnstableDevWorker

	beforeAll(async () => {
		worker = await unstable_dev("src/index.ts", {
			experimental: { disableExperimentalWarning: true },
		});
	});

	afterAll(async () => {
		await worker.stop()
	})

	it("should deny request for short paths", async () => {
		const cases = {
			failures: ["/", "/foo", "/foo/", "/%2F"],
		}
		for (const path of cases.failures) {
			const resp = await worker.fetch(`http://example.com${path}`)
			if (resp) {
				const text = await resp.text()
				expect(text).toMatchInlineSnapshot('"path must be at least 5 characters"')
			}
		}
	})

	describe("durable object", () => {
		it("Should send text from a POST to a matching GET", async () => {
			const path = "/stuff1"
			const url = `http://example.com${path}`

			// The get request should wait for the post request to complete
			const getResponsePromise = worker.fetch(url)

			// The post request to the same path should receive a response that the text was consumed
			const postResponse = await worker.fetch(url, { method: "POST", body: "Hello World 12345" })
			expect(postResponse.status).toBe(200)
			const postText = await postResponse.text()
			expect(postText).toBe("The text was consumed!")

			// The get request should now receive the text
			const getResponse = await getResponsePromise
			expect(getResponse.status).toBe(200)
			const text = await getResponse.text()
			expect(text).toBe("Hello World 12345")
		})

		it("Shouldn't send text from a POST to a different GET", async () => {
			const path1 = "/stuff1"
			const path2 = "/stuff2"
			const url = (p: string) => `http://example.com${p}`

			// The get request should wait for the post request to complete
			const getResponsePromise1 = worker.fetch(url(path1))
			const getResponsePromise2 = worker.fetch(url(path2))

			// The post request to the same path should receive a response that the text was consumed
			const postResponse1 = await worker.fetch(url(path1), { method: "POST", body: "Hello World 12345" })
			expect(postResponse1.status).toBe(200)
			const postText1 = await postResponse1.text()
			expect(postText1).toBe("The text was consumed!")

			const postResponse2 = await worker.fetch(url(path2), { method: "POST", body: "Hello World 789" })
			expect(postResponse2.status).toBe(200)
			const postText2 = await postResponse2.text()
			expect(postText2).toBe("The text was consumed!")

			// The get request should now receive the text
			const getResponse1 = await getResponsePromise1
			expect(getResponse1.status).toBe(200)
			const text1 = await getResponse1.text()
			expect(text1).toBe("Hello World 12345")

			const getResponse2 = await getResponsePromise2
			expect(getResponse2.status).toBe(200)
			const text2 = await getResponse2.text()
			expect(text2).toBe("Hello World 789")
		})

		it("Should not send the same POST twice", async () => {
			const path = "/stuff1"
			const url = (p: string) => `http://example.com${p}`

			// The get request should wait for the post request to complete
			const getResponsePromise1 = worker.fetch(url(path))

			// The post request to the same path should receive a response that the text was consumed
			const postResponse1 = await worker.fetch(url(path), { method: "POST", body: "Hello World 12345" })
			expect(postResponse1.status).toBe(200)
			const postText1 = await postResponse1.text()
			expect(postText1).toBe("The text was consumed!")

			// The get request should now receive the text
			const getResponse1 = await getResponsePromise1
			expect(getResponse1.status).toBe(200)
			const text1 = await getResponse1.text()
			expect(text1).toBe("Hello World 12345")

			// The next get request should wait for the next post request to complete
			const getResponsePromise2 = worker.fetch(url(path))

			// Send a new POST with different text
			const postResponse2 = await worker.fetch(url(path), { method: "POST", body: "Hello World 789" })
			expect(postResponse2.status).toBe(200)
			const postText2 = await postResponse2.text()
			expect(postText2).toBe("The text was consumed!")
			
			// The get request should receive the new text, not the old text
			const getResponse2 = await getResponsePromise2
			expect(getResponse2.status).toBe(200)
			const text2 = await getResponse2.text()
			expect(text2).toBe("Hello World 789")
		})
	})
})