134 lines
5.0 KiB
TypeScript
134 lines
5.0 KiB
TypeScript
import { afterEach, beforeEach, describe, expect, it, vi } from 'vitest';
|
||
import { newFixture, type Fixture } from '../helpers/fixture.js';
|
||
|
||
let fixture: Fixture;
|
||
beforeEach(async () => {
|
||
vi.resetModules();
|
||
fixture = await newFixture([]);
|
||
});
|
||
afterEach(async () => {
|
||
await fixture.cleanup();
|
||
vi.restoreAllMocks();
|
||
});
|
||
|
||
async function seedBeta(multiple: boolean) {
|
||
const { db, closeDb } = await import('../../src/lib/server/db/index.js');
|
||
const { createScript, getLatestVersion } =
|
||
await import('../../src/lib/server/fleet/scripts.js');
|
||
const games = multiple ? ['t5', 't6'] : ['t5'];
|
||
const scripts = games.map((game) => {
|
||
const { script } = createScript({
|
||
name: `${game} test`,
|
||
targetRelPath: `${game}.gsc`,
|
||
scopeType: 'game',
|
||
scopeValue: game,
|
||
content: Buffer.from(`${game} version`)
|
||
});
|
||
return { ...script, versionId: getLatestVersion(script.id)!.id };
|
||
});
|
||
const betaId = Number(
|
||
db()
|
||
.prepare(
|
||
`INSERT INTO betas (script_id, script_version_id, trigger_kind, state)
|
||
VALUES (?, ?, 'bundle', 'running')`
|
||
)
|
||
.run(scripts[0].id, scripts[0].versionId).lastInsertRowid
|
||
);
|
||
for (const script of scripts)
|
||
db()
|
||
.prepare(
|
||
'INSERT INTO beta_scripts (beta_id, script_id, script_version_id) VALUES (?, ?, ?)'
|
||
)
|
||
.run(betaId, script.id, script.versionId);
|
||
for (const [name, game] of [
|
||
['first', 't5'],
|
||
['second', multiple ? 't6' : 't5']
|
||
]) {
|
||
db().prepare('INSERT INTO servers (name, game) VALUES (?, ?)').run(name, game);
|
||
db()
|
||
.prepare(
|
||
`INSERT INTO beta_servers (beta_id, server, had_original, prior_version_id, stage)
|
||
VALUES (?, ?, 1, ?, 'awaiting-empty')`
|
||
)
|
||
.run(betaId, name, scripts[0].versionId);
|
||
}
|
||
const assign = (server: string, scriptId: number) =>
|
||
db()
|
||
.prepare(
|
||
`INSERT INTO beta_server_scripts
|
||
(beta_id, server, script_id, had_original) VALUES (?, ?, ?, 0)`
|
||
)
|
||
.run(betaId, server, scriptId);
|
||
const memberships = () =>
|
||
db()
|
||
.prepare(
|
||
`SELECT server, script_id, had_original, prior_version_id
|
||
FROM beta_server_scripts WHERE beta_id = ? ORDER BY server, script_id`
|
||
)
|
||
.all(betaId);
|
||
return { betaId, scripts, assign, memberships, db, closeDb };
|
||
}
|
||
|
||
describe('per-server beta membership', () => {
|
||
it('shows only each game’s held files in fleet and server snapshots', async () => {
|
||
const x = await seedBeta(true);
|
||
x.assign('first', x.scripts[0].id);
|
||
x.assign('second', x.scripts[1].id);
|
||
const { fleetSnapshot } = await import('../../src/lib/server/fleet/snapshot.js');
|
||
const expected = [
|
||
{ name: 'first', held: [{ target: 't5.gsc', betaId: x.betaId }] },
|
||
{ name: 'second', held: [{ target: 't6.gsc', betaId: x.betaId }] }
|
||
];
|
||
expect(fleetSnapshot().servers.map(({ name, held }) => ({ name, held }))).toEqual(expected);
|
||
expect(fleetSnapshot('second').servers.map(({ name, held }) => ({ name, held }))).toEqual([
|
||
expected[1]
|
||
]);
|
||
});
|
||
|
||
it('preserves game-specific membership when the database reopens', async () => {
|
||
const x = await seedBeta(true);
|
||
x.assign('first', x.scripts[0].id);
|
||
x.assign('second', x.scripts[1].id);
|
||
const expected = x.memberships();
|
||
x.closeDb();
|
||
expect(x.memberships()).toEqual(expected);
|
||
x.closeDb();
|
||
expect(x.memberships()).toEqual(expected);
|
||
});
|
||
|
||
it('does not infer missing bundle membership from the primary script', async () => {
|
||
const x = await seedBeta(true);
|
||
x.assign('first', x.scripts[0].id);
|
||
const expected = x.memberships();
|
||
x.closeDb();
|
||
expect(x.memberships()).toEqual(expected);
|
||
});
|
||
|
||
it('still backfills legacy single-script servers while retaining existing snapshots', async () => {
|
||
const x = await seedBeta(false);
|
||
x.assign('first', x.scripts[0].id);
|
||
x.db().prepare('DELETE FROM beta_scripts WHERE beta_id = ?').run(x.betaId);
|
||
x.closeDb();
|
||
expect(x.memberships()).toEqual([
|
||
{
|
||
server: 'first',
|
||
script_id: x.scripts[0].id,
|
||
had_original: 0,
|
||
prior_version_id: null
|
||
},
|
||
{
|
||
server: 'second',
|
||
script_id: x.scripts[0].id,
|
||
had_original: 1,
|
||
prior_version_id: x.scripts[0].versionId
|
||
}
|
||
]);
|
||
expect(
|
||
x
|
||
.db()
|
||
.prepare('SELECT script_id, script_version_id FROM beta_scripts WHERE beta_id = ?')
|
||
.all(x.betaId)
|
||
).toEqual([{ script_id: x.scripts[0].id, script_version_id: x.scripts[0].versionId }]);
|
||
});
|
||
});
|