32 lines
685 B
TypeScript
32 lines
685 B
TypeScript
import { applyD1Migrations, type D1Migration, env } from 'cloudflare:test'
|
|
import { beforeAll, beforeEach } from 'vitest'
|
|
|
|
declare module 'cloudflare:test' {
|
|
interface ProvidedEnv {
|
|
DB: D1Database
|
|
ERROR_CONTEXTS: R2Bucket
|
|
TEST_MIGRATIONS: D1Migration[]
|
|
}
|
|
}
|
|
|
|
const tables = [
|
|
'error_context_reservations',
|
|
'error_context_samples',
|
|
'error_context_budget',
|
|
'error_reports',
|
|
'error_daily',
|
|
'error_groups',
|
|
'daily_active',
|
|
'daily_totals',
|
|
'accepted_batches',
|
|
'installations',
|
|
]
|
|
|
|
beforeAll(async () => {
|
|
await applyD1Migrations(env.DB, env.TEST_MIGRATIONS)
|
|
})
|
|
|
|
beforeEach(async () => {
|
|
await env.DB.batch(tables.map((table) => env.DB.prepare(`DELETE FROM ${table}`)))
|
|
})
|