fix reserved turn queue concurrency

2026-07-26 18:32:52 +00:00
parent ab6ed3553a
commit 3cd1ad3b7f
18 changed files with 613 additions and 106 deletions
@@ -0,0 +1,63 @@
+import { afterAll, beforeAll, describe, expect, it } from 'vitest';
+
+import { createGamePostgresConnector } from '@sammo-ts/infra';
+import type { DatabaseClient } from '../src/context.js';
+import {
+    ReservedTurnRevisionConflictError,
+    getGeneralTurnSnapshot,
+    setGeneralTurn,
+} from '../src/turns/reservedTurns.js';
+
+const databaseUrl = process.env.RESERVED_TURN_DATABASE_URL;
+const describeIntegration = databaseUrl ? describe : describe.skip;
+const GENERAL_ID = 2_147_400_001;
+
+describeIntegration('reserved turn queue revision integration', () => {
+    const connector = databaseUrl ? createGamePostgresConnector({ url: databaseUrl }) : null;
+
+    beforeAll(async () => {
+        if (!connector) {
+            return;
+        }
+        await connector.connect();
+        await connector.prisma.generalTurn.deleteMany({ where: { generalId: GENERAL_ID } });
+        await connector.prisma.generalTurnRevision.deleteMany({ where: { generalId: GENERAL_ID } });
+    });
+
+    afterAll(async () => {
+        if (!connector) {
+            return;
+        }
+        await connector.prisma.generalTurn.deleteMany({ where: { generalId: GENERAL_ID } });
+        await connector.prisma.generalTurnRevision.deleteMany({ where: { generalId: GENERAL_ID } });
+        await connector.disconnect();
+    });
+
+    it('allows exactly one writer for the same expected revision', async () => {
+        if (!connector) {
+            throw new Error('integration connector is unavailable');
+        }
+
+        const write = (action: string) =>
+            connector.prisma.$transaction((transaction) =>
+                setGeneralTurn(transaction as unknown as DatabaseClient, GENERAL_ID, 0, action, {}, 0)
+            );
+
+        const results = await Promise.allSettled([write('che_훈련'), write('che_사기진작')]);
+        const fulfilled = results.filter(
+            (result): result is PromiseFulfilledResult<Awaited<ReturnType<typeof write>>> =>
+                result.status === 'fulfilled'
+        );
+        const rejected = results.filter((result): result is PromiseRejectedResult => result.status === 'rejected');
+
+        expect(fulfilled).toHaveLength(1);
+        expect(fulfilled[0]?.value.revision).toBe(1);
+        expect(rejected).toHaveLength(1);
+        expect(rejected[0]?.reason).toBeInstanceOf(ReservedTurnRevisionConflictError);
+
+        const snapshot = await getGeneralTurnSnapshot(connector.prisma as unknown as DatabaseClient, GENERAL_ID);
+        expect(snapshot.revision).toBe(1);
+        expect(['che_훈련', 'che_사기진작']).toContain(snapshot.turns[0]?.action);
+        expect(await connector.prisma.generalTurn.count({ where: { generalId: GENERAL_ID } })).toBe(30);
+    });
+});