69 lines
2.3 KiB
TypeScript
69 lines
2.3 KiB
TypeScript
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.ZHINIAN_RUNTIME_DIR;
|
|
previousSupabaseUrl = process.env.NEXT_PUBLIC_SUPABASE_URL;
|
|
previousSupabaseKey = process.env.SUPABASE_SERVICE_ROLE_KEY;
|
|
process.env.ZHINIAN_RUNTIME_DIR = runtimeDir;
|
|
delete process.env.NEXT_PUBLIC_SUPABASE_URL;
|
|
delete process.env.SUPABASE_SERVICE_ROLE_KEY;
|
|
});
|
|
|
|
afterEach(async () => {
|
|
restoreEnv("ZHINIAN_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;
|
|
}
|