custom rooms

apps/storage/src/test/integration/api.test.ts

@@ -0,0 +1,134 @@
+import { env, SELF } from 'cloudflare:test'
+import { expect, it } from 'vitest'
+
+import type { Env } from '../../context'
+
+declare module 'cloudflare:test' {
+	interface ProvidedEnv extends Env {}
+}
+
+const ORIGIN = 'https://example.com'
+
+// Mint a token the way the `auth` worker does, using the same dev secret, so the
+// storage worker's validation accepts it.
+const DEV_SECRET = 'dev-insecure-signing-key-change-me'
+
+function b64url(input: ArrayBuffer | string): string {
+	const bytes = typeof input === 'string' ? new TextEncoder().encode(input) : new Uint8Array(input)
+	let binary = ''
+	for (const byte of bytes) binary += String.fromCharCode(byte)
+	return btoa(binary).replace(/\+/g, '-').replace(/\//g, '_').replace(/=+$/, '')
+}
+
+async function bearer(sub = '42'): Promise<Record<string, string>> {
+	const now = Math.floor(Date.now() / 1000)
+	const signingInput = `${b64url(JSON.stringify({ alg: 'HS256', typ: 'JWT' }))}.${b64url(
+		JSON.stringify({ sub, exp: now + 3600 })
+	)}`
+	const key = await crypto.subtle.importKey(
+		'raw',
+		new TextEncoder().encode(DEV_SECRET),
+		{ name: 'HMAC', hash: 'SHA-256' },
+		false,
+		['sign']
+	)
+	const sig = await crypto.subtle.sign('HMAC', key, new TextEncoder().encode(signingInput))
+	return { Authorization: `Bearer ${signingInput}.${b64url(sig)}` }
+}
+
+/** Build the client's multipart upload body: FileType + File. */
+function uploadForm(fileType: string, bytes: Uint8Array): FormData {
+	const form = new FormData()
+	form.set('FileType', fileType)
+	form.set('File', new File([bytes], 'file.bin', { type: 'application/octet-stream' }))
+	return form
+}
+
+it('response with hello world', async () => {
+	const res = await SELF.fetch(ORIGIN)
+	expect(res.status).toBe(200)
+	expect(await res.text()).toMatchInlineSnapshot(`"hello, world!"`)
+})
+
+it('POST /upload 401s without a token', async () => {
+	const res = await SELF.fetch(`${ORIGIN}/upload`, {
+		method: 'POST',
+		body: uploadForm('6', new Uint8Array([1])),
+	})
+	expect(res.status).toBe(401)
+})
+
+it('POST /upload stores a RoomMetadata (FileType 6) file under roommetadata/ and returns its name', async () => {
+	// Mirrors the client's multipart upload: FileType=6, File=<binary>.
+	const bytes = new Uint8Array([0x10, 0x02, 0x1a, 0x00])
+	const res = await SELF.fetch(`${ORIGIN}/upload`, {
+		method: 'POST',
+		headers: await bearer(),
+		body: uploadForm('6', bytes),
+	})
+	expect(res.status).toBe(200)
+	const { filename } = (await res.json()) as { filename: string }
+	expect(filename).toMatch(/^[0-9a-f]{32}$/)
+
+	// The bytes are persisted in the shared CDN bucket under the type subfolder.
+	const stored = await env.CDN_ASSETS.get(`roommetadata/${filename}`)
+	expect(stored).not.toBeNull()
+	expect(new Uint8Array(await stored!.arrayBuffer())).toEqual(bytes)
+})
+
+it('POST /upload folders each FileType under its own subfolder', async () => {
+	const headers = await bearer()
+	const cases: Array<[string, string]> = [
+		['1', 'roomsave'],
+		['3', 'image'],
+		['5', 'invention'],
+	]
+	for (const [fileType, subfolder] of cases) {
+		const res = await SELF.fetch(`${ORIGIN}/upload`, {
+			method: 'POST',
+			headers,
+			body: uploadForm(fileType, new Uint8Array([1, 2, 3])),
+		})
+		expect(res.status).toBe(200)
+		const { filename } = (await res.json()) as { filename: string }
+		expect(await env.CDN_ASSETS.get(`${subfolder}/${filename}`)).not.toBeNull()
+	}
+})
+
+it('POST /upload 400s for a binary with an unknown/missing FileType', async () => {
+	// Unknown type (999) and the Unknown enum value (0) have no destination → 400.
+	for (const fileType of ['999', '0']) {
+		const res = await SELF.fetch(`${ORIGIN}/upload`, {
+			method: 'POST',
+			headers: await bearer(),
+			body: uploadForm(fileType, new Uint8Array([9])),
+		})
+		expect(res.status).toBe(400)
+	}
+})
+
+it('POST /upload echoes an explicit name when no binary is posted', async () => {
+	const form = new FormData()
+	form.set('FileType', '3')
+	form.set('imageName', 'existing-image-key.png')
+	const res = await SELF.fetch(`${ORIGIN}/upload`, {
+		method: 'POST',
+		headers: await bearer(),
+		body: form,
+	})
+	expect(res.status).toBe(200)
+	expect((await res.json()) as { filename: string }).toEqual({
+		filename: 'existing-image-key.png',
+	})
+})
+
+it('POST /upload 400s when there is neither a file nor a name', async () => {
+	const form = new FormData()
+	form.set('FileType', '6')
+	const res = await SELF.fetch(`${ORIGIN}/upload`, {
+		method: 'POST',
+		headers: await bearer(),
+		body: form,
+	})
+	expect(res.status).toBe(400)
+})