test: add upload_cleanup_jobs table to test databases

Author: workHMZ

worker/test/admin-auth.test.ts

@@ -53,6 +53,13 @@ describe('administrator cookie session', () => {
         created_at TEXT NOT NULL,
         orphaned_at TEXT
       )`),
+      env.DB.prepare(`CREATE TABLE IF NOT EXISTS upload_cleanup_jobs (
+        id TEXT PRIMARY KEY,
+        upload_id TEXT NOT NULL,
+        r2_key TEXT NOT NULL,
+        size_bytes INTEGER NOT NULL,
+        claimed_at TEXT NOT NULL
+      )`),
       env.DB.prepare(`CREATE TRIGGER IF NOT EXISTS test_file_blobs_orphan_after_share_soft_delete
         AFTER UPDATE OF deleted_at ON shares
         WHEN OLD.type = 'file'

worker/test/instant-upload.test.ts

@@ -73,6 +73,13 @@ describe('verified instant upload', () => {
         object_uploaded_at TEXT, status TEXT NOT NULL,
         created_at TEXT NOT NULL, orphaned_at TEXT
       )`),
+      env.DB.prepare(`CREATE TABLE IF NOT EXISTS upload_cleanup_jobs (
+        id TEXT PRIMARY KEY,
+        upload_id TEXT NOT NULL,
+        r2_key TEXT NOT NULL,
+        size_bytes INTEGER NOT NULL,
+        claimed_at TEXT NOT NULL
+      )`),
       env.DB.prepare(`CREATE UNIQUE INDEX IF NOT EXISTS test_file_blobs_active_fingerprint
         ON file_blobs(fingerprint_algorithm, content_fingerprint, size_bytes)
         WHERE status = 'active' AND fingerprint_algorithm IS NOT NULL
@@ -126,6 +133,7 @@ describe('verified instant upload', () => {
   beforeEach(async () => {
     await env.DB.prepare('DELETE FROM shares').run()
     await env.DB.prepare('DELETE FROM upload_sessions').run()
+    await env.DB.prepare('DELETE FROM upload_cleanup_jobs').run()
     await env.DB.prepare('DELETE FROM file_blobs').run()
     await env.DB.prepare('DELETE FROM audit_logs').run()
     await env.DB.prepare('DELETE FROM abuse_counters').run()

worker/test/scheduled.test.ts

@@ -46,6 +46,13 @@ describe('scheduled cleanup reporting', () => {
         created_at TEXT NOT NULL,
         orphaned_at TEXT
       )`),
+      env.DB.prepare(`CREATE TABLE IF NOT EXISTS upload_cleanup_jobs (
+        id TEXT PRIMARY KEY,
+        upload_id TEXT NOT NULL,
+        r2_key TEXT NOT NULL,
+        size_bytes INTEGER NOT NULL,
+        claimed_at TEXT NOT NULL
+      )`),
       env.DB.prepare(`CREATE TRIGGER IF NOT EXISTS test_scheduled_orphan_after_share_soft_delete
         AFTER UPDATE OF deleted_at ON shares
         WHEN OLD.type = 'file'
@@ -70,6 +77,7 @@ describe('scheduled cleanup reporting', () => {
     await env.DB.batch([
       env.DB.prepare('DELETE FROM shares'),
       env.DB.prepare('DELETE FROM upload_sessions'),
+      env.DB.prepare('DELETE FROM upload_cleanup_jobs'),
       env.DB.prepare('DELETE FROM file_blobs'),
       env.DB.prepare('DELETE FROM abuse_counters'),
       env.DB.prepare('DELETE FROM audit_logs'),

worker/test/share-download.test.ts

@@ -72,6 +72,13 @@ describe('share download accounting', () => {
         created_at TEXT NOT NULL,
         orphaned_at TEXT
       )`),
+      env.DB.prepare(`CREATE TABLE IF NOT EXISTS upload_cleanup_jobs (
+        id TEXT PRIMARY KEY,
+        upload_id TEXT NOT NULL,
+        r2_key TEXT NOT NULL,
+        size_bytes INTEGER NOT NULL,
+        claimed_at TEXT NOT NULL
+      )`),
       env.DB.prepare(`CREATE TABLE IF NOT EXISTS storage_usage (
         id INTEGER PRIMARY KEY CHECK (id = 1),
         active_bytes INTEGER NOT NULL DEFAULT 0
@@ -172,6 +179,10 @@ describe('share download accounting', () => {
 
   beforeEach(async () => {
     await env.DB.batch([
+      env.DB.prepare('DELETE FROM shares'),
+      env.DB.prepare('DELETE FROM upload_sessions'),
+      env.DB.prepare('DELETE FROM upload_cleanup_jobs'),
+      env.DB.prepare('DELETE FROM file_blobs'),
       env.DB.prepare('DELETE FROM abuse_counters'),
       env.DB.prepare('DELETE FROM settings'),
     ])

worker/test/storage-accounting.test.ts

@@ -40,6 +40,13 @@ describe('storage accounting with D1 triggers', () => {
         created_at TEXT NOT NULL,
         orphaned_at TEXT
       )`),
+      env.DB.prepare(`CREATE TABLE IF NOT EXISTS upload_cleanup_jobs (
+        id TEXT PRIMARY KEY,
+        upload_id TEXT NOT NULL,
+        r2_key TEXT NOT NULL,
+        size_bytes INTEGER NOT NULL,
+        claimed_at TEXT NOT NULL
+      )`),
       env.DB.prepare(`CREATE UNIQUE INDEX IF NOT EXISTS test_active_blob_fingerprint
         ON file_blobs(fingerprint_algorithm, content_fingerprint, size_bytes)
         WHERE status = 'active'
@@ -128,6 +135,7 @@ describe('storage accounting with D1 triggers', () => {
   beforeEach(async () => {
     await env.DB.prepare('DELETE FROM shares').run()
     await env.DB.prepare('DELETE FROM upload_sessions').run()
+    await env.DB.prepare('DELETE FROM upload_cleanup_jobs').run()
     await env.DB.prepare('DELETE FROM file_blobs').run()
     await env.DB.prepare('UPDATE storage_usage SET active_bytes = 0 WHERE id = 1').run()
   })