Initial 智念AIGC platform

tests/data-store-concurrency.test.ts

@@ -0,0 +1,68 @@
+import { mkdtemp, rm } from "node:fs/promises";
+import { tmpdir } from "node:os";
+import { join } from "node:path";
+import { afterEach, beforeEach, describe, expect, it } from "vitest";
+import {
+  createGenerationJob,
+  listGenerationJobs,
+  updateGenerationJob
+} from "@/lib/server/data-store";
+import { DEFAULT_OWNER_ID } from "@/lib/server/runtime";
+
+let runtimeDir = "";
+let previousRuntimeDir: string | undefined;
+let previousSupabaseUrl: string | undefined;
+let previousSupabaseKey: string | undefined;
+
+describe("local data store concurrency", () => {
+  beforeEach(async () => {
+    runtimeDir = await mkdtemp(join(tmpdir(), "zhinian-store-"));
+    previousRuntimeDir = process.env.NIANXXPLAY_RUNTIME_DIR;
+    previousSupabaseUrl = process.env.NEXT_PUBLIC_SUPABASE_URL;
+    previousSupabaseKey = process.env.SUPABASE_SERVICE_ROLE_KEY;
+    process.env.NIANXXPLAY_RUNTIME_DIR = runtimeDir;
+    delete process.env.NEXT_PUBLIC_SUPABASE_URL;
+    delete process.env.SUPABASE_SERVICE_ROLE_KEY;
+  });
+
+  afterEach(async () => {
+    restoreEnv("NIANXXPLAY_RUNTIME_DIR", previousRuntimeDir);
+    restoreEnv("NEXT_PUBLIC_SUPABASE_URL", previousSupabaseUrl);
+    restoreEnv("SUPABASE_SERVICE_ROLE_KEY", previousSupabaseKey);
+    await rm(runtimeDir, { force: true, recursive: true });
+  });
+
+  it("serializes concurrent job writes without losing records", async () => {
+    const created = await Promise.all(Array.from({ length: 8 }, (_, index) => createGenerationJob({
+      ownerId: DEFAULT_OWNER_ID,
+      capability: "image.generate",
+      provider: "mock",
+      reqKey: "jimeng_seedream46_cvtob",
+      status: "queued",
+      prompt: `job ${index}`,
+      inputAssetIds: [],
+      inputUrls: [],
+      outputAssetIds: [],
+      requestPayload: { index }
+    })));
+
+    await Promise.all(created.map((job, index) => updateGenerationJob(job.id, {
+      status: "succeeded",
+      outputAssetIds: [`asset-${index}`]
+    })));
+
+    const stored = await listGenerationJobs(DEFAULT_OWNER_ID, 20);
+    const storedById = new Map(stored.map((job) => [job.id, job]));
+    for (const job of created) {
+      expect(storedById.get(job.id)?.status).toBe("succeeded");
+    }
+  });
+});
+
+function restoreEnv(name: string, value: string | undefined) {
+  if (value === undefined) {
+    delete process.env[name];
+    return;
+  }
+  process.env[name] = value;
+}