test: align contract verification worker tests with cloudflare runtime APIs

Author: o-az

apps/contract-verification/test/e2e/verification.test.ts

@@ -2,7 +2,7 @@ import * as z from 'zod/mini'
 import { eq } from 'drizzle-orm'
 import { drizzle } from 'drizzle-orm/d1'
 import { describe, expect, it } from 'vitest'
-import { env, exports } from 'cloudflare:workers'
+import { env, SELF, runInDurableObject } from 'cloudflare:test'
 
 import * as DB from '#database/schema.ts'
 import { runVerificationJob } from '#route.verify.ts'
@@ -33,15 +33,13 @@ const getFirst = <T>(items: T[], label: string) => {
 
 describe('full verification flow', () => {
 	async function createVerificationJob(): Promise<string> {
-		const verifyResponse = await exports.default.fetch(
-			new Request(
-				`https://test.local/v2/verify/${counterFixture.chainId}/${counterFixture.address}`,
-				{
-					method: 'POST',
-					headers: { 'Content-Type': 'application/json' },
-					body: JSON.stringify(verifyRequestData),
-				},
-			),
+		const verifyResponse = await SELF.fetch(
+			`https://test.local/v2/verify/${counterFixture.chainId}/${counterFixture.address}`,
+			{
+				method: 'POST',
+				headers: { 'Content-Type': 'application/json' },
+				body: JSON.stringify(verifyRequestData),
+			},
 		)
 
 		if (verifyResponse.status !== 202) {
@@ -81,8 +79,8 @@ describe('full verification flow', () => {
 			},
 		)
 
-		const statusResponse = await exports.default.fetch(
-			new Request(`https://test.local/v2/verify/${verificationId}`),
+		const statusResponse = await SELF.fetch(
+			`https://test.local/v2/verify/${verificationId}`,
 		)
 		expect(statusResponse.status).toBe(200)
 
@@ -99,8 +97,8 @@ describe('full verification flow', () => {
 	it('verifies a simple contract and persists to database', async () => {
 		const verificationId = await runSuccessfulVerification()
 
-		const statusResponse = await exports.default.fetch(
-			new Request(`https://test.local/v2/verify/${verificationId}`),
+		const statusResponse = await SELF.fetch(
+			`https://test.local/v2/verify/${verificationId}`,
 		)
 		expect(statusResponse.status).toBe(200)
 		const status = z.parse(
@@ -109,10 +107,8 @@ describe('full verification flow', () => {
 		)
 		expect(status.error).toBeUndefined()
 
-		const lookupResponse = await exports.default.fetch(
-			new Request(
-				`https://test.local/v2/contract/${counterFixture.chainId}/${counterFixture.address}?fields=sources,signatures`,
-			),
+		const lookupResponse = await SELF.fetch(
+			`https://test.local/v2/contract/${counterFixture.chainId}/${counterFixture.address}?fields=sources,signatures`,
 		)
 
 		expect(lookupResponse.status).toBe(200)
@@ -199,15 +195,13 @@ describe('full verification flow', () => {
 	it('returns 409 for already verified contract', async () => {
 		await runSuccessfulVerification()
 
-		const secondResponse = await exports.default.fetch(
-			new Request(
-				`https://test.local/v2/verify/${counterFixture.chainId}/${counterFixture.address}`,
-				{
-					method: 'POST',
-					headers: { 'Content-Type': 'application/json' },
-					body: JSON.stringify(verifyRequestData),
-				},
-			),
+		const secondResponse = await SELF.fetch(
+			`https://test.local/v2/verify/${counterFixture.chainId}/${counterFixture.address}`,
+			{
+				method: 'POST',
+				headers: { 'Content-Type': 'application/json' },
+				body: JSON.stringify(verifyRequestData),
+			},
 		)
 
 		expect(secondResponse.status).toBe(409)
@@ -216,4 +210,65 @@ describe('full verification flow', () => {
 			'already_verified',
 		)
 	})
+
+	it('stores the job in the DO and completes via the DO alarm handler', async () => {
+		const verificationId = await createVerificationJob()
+		const stub = env.VERIFICATION_JOB_RUNNER.get(
+			env.VERIFICATION_JOB_RUNNER.idFromName(verificationId),
+		)
+
+		const storedState = await runInDurableObject(
+			stub,
+			async (_instance, state) => {
+				return {
+					job: await state.storage.get('job'),
+					alarm: await state.storage.getAlarm(),
+				}
+			},
+		)
+		expect(storedState.job).toBeDefined()
+
+		await runInDurableObject(stub, async (instance, state) => {
+			const job = await state.storage.get<{
+				jobId: string
+				chainId: number
+				address: string
+				stdJsonInput: unknown
+				compilerVersion: string
+				contractIdentifier: string
+			}>('job')
+			if (!job) {
+				return
+			}
+
+			await state.storage.put('job', {
+				...job,
+				chainId: 999_999,
+			})
+			await instance.alarm()
+		})
+
+		const finalState = await runInDurableObject(
+			stub,
+			async (_instance, state) => {
+				return {
+					job: await state.storage.get('job'),
+					alarm: await state.storage.getAlarm(),
+				}
+			},
+		)
+		expect(finalState.job).toBeUndefined()
+		expect(finalState.alarm).toBeNull()
+
+		const statusResponse = await SELF.fetch(
+			`https://test.local/v2/verify/${verificationId}`,
+		)
+		expect(statusResponse.status).toBe(200)
+		const status = z.parse(
+			VerificationStatusSchema,
+			await statusResponse.json(),
+		)
+		expect(status.isJobCompleted).toBe(true)
+		expect(status.error?.customCode).toBe('internal_error')
+	})
 })

apps/contract-verification/test/integration/rate-limit-whitelist.test.ts

@@ -1,4 +1,4 @@
-import { env } from 'cloudflare:workers'
+import { env } from 'cloudflare:test'
 import { describe, expect, it, vi } from 'vitest'
 
 import { app } from '#index.tsx'

apps/contract-verification/test/integration/route.lookup-verified.test.ts

@@ -1,6 +1,6 @@
 import { Hash, Hex } from 'ox'
 import { eq } from 'drizzle-orm'
-import { env } from 'cloudflare:workers'
+import { env } from 'cloudflare:test'
 import { drizzle } from 'drizzle-orm/d1'
 import { describe, expect, it } from 'vitest'
 

apps/contract-verification/test/integration/route.lookup.test.ts

@@ -1,7 +1,7 @@
 import { Hex } from 'ox'
 import * as z from 'zod/mini'
 import { eq } from 'drizzle-orm'
-import { env } from 'cloudflare:workers'
+import { env } from 'cloudflare:test'
 import { drizzle } from 'drizzle-orm/d1'
 import { describe, it, expect } from 'vitest'
 

apps/contract-verification/test/integration/route.verify-legacy.test.ts

@@ -1,5 +1,5 @@
 import * as CBOR from 'cbor-x'
-import { env } from 'cloudflare:workers'
+import { env } from 'cloudflare:test'
 import { drizzle } from 'drizzle-orm/d1'
 import { Hono } from 'hono'
 import { Hash, Hex } from 'ox'

apps/contract-verification/test/integration/route.verify-status.test.ts

@@ -1,4 +1,4 @@
-import { env } from 'cloudflare:workers'
+import { SELF, env } from 'cloudflare:test'
 import { eq } from 'drizzle-orm'
 import { drizzle } from 'drizzle-orm/d1'
 import { Hex, Hash } from 'ox'
@@ -12,14 +12,13 @@ import {
 	verifiedContractsTable,
 	verificationJobsTable,
 } from '#database/schema.ts'
-import { app } from '#index.tsx'
 
 function statusUrl(id: string): string {
-	return `/v2/verify/${id}`
+	return `https://test.local/v2/verify/${id}`
 }
 
 async function getStatus(id: string): Promise<Response> {
-	return app.request(statusUrl(id), { method: 'GET' }, env)
+	return SELF.fetch(statusUrl(id), { method: 'GET' })
 }
 
 const db = () => drizzle(env.CONTRACTS_DB)