249 lines
16 KiB
TypeScript
249 lines
16 KiB
TypeScript
import { afterEach, beforeEach, describe, expect, it, vi } from 'vitest';
|
|
import { existsSync } from 'node:fs';
|
|
import { readFile, rm } from 'node:fs/promises';
|
|
import { join } from 'node:path';
|
|
import { newFixture, type Fixture } from '../helpers/fixture.js';
|
|
|
|
vi.mock('../../src/lib/server/integrations/komodo.js', () => ({
|
|
komodoCreateDeployment: vi.fn(async () => ({ ok: true })),
|
|
komodoDeploy: vi.fn(async () => ({ ok: true })),
|
|
komodoDestroyDeployment: vi.fn(async () => ({ ok: true })),
|
|
komodoDeleteDeployment: vi.fn(async () => ({ ok: true })),
|
|
komodoInspectHealthy: vi.fn(),
|
|
komodoEnsureTag: vi.fn(async (name: string) => ({ ok: true, id: name })),
|
|
komodoUpdateTagsOnResource: vi.fn(async () => ({ ok: true }))
|
|
}));
|
|
vi.mock('../../src/lib/server/pipeline/worker.js', () => ({ kick: vi.fn(async () => {}) }));
|
|
|
|
let fixture: Fixture;
|
|
const serverName = 't5_zombie_cosmodrome_1';
|
|
const input = { gameMode: 'zombie' as const, mapSlug: 'cosmodrome', plutoKey: 'test-key' };
|
|
|
|
beforeEach(async () => {
|
|
vi.resetModules();
|
|
vi.clearAllMocks();
|
|
fixture = await newFixture([]);
|
|
vi.stubEnv('KOMODO_ADDRESS', 'http://komodo.test');
|
|
vi.stubEnv('KOMODO_API_KEY', 'test');
|
|
vi.stubEnv('KOMODO_API_SECRET', 'test');
|
|
vi.stubEnv('KOMODO_TARGET_SERVER', 'test-host');
|
|
vi.stubEnv('RCON_PASSWORD', 'test-rcon');
|
|
});
|
|
|
|
afterEach(async () => {
|
|
await fixture.cleanup();
|
|
vi.unstubAllEnvs();
|
|
});
|
|
|
|
async function promotedConfig() {
|
|
const { createScript, getLatestVersion } = await import('../../src/lib/server/fleet/scripts.js');
|
|
const { db } = await import('../../src/lib/server/db/index.js');
|
|
for (const artifact of [
|
|
{ name: 'baseline', kind: 'baseline' as const, scopeType: 'game' as const, scopeValue: 't5',
|
|
content: Buffer.from('set sv_voice "1"\nset sv_maprotation "map zombie_theater"\n') },
|
|
{ name: 'map settings', kind: 'config' as const, scopeType: 'map' as const, scopeValue: 'zombie_cosmodrome',
|
|
content: Buffer.from('set sv_maprotation "map zombie_moon"\nset g_inactivity "0"\n') },
|
|
{ name: 'other map', kind: 'config' as const, scopeType: 'map' as const, scopeValue: 'zombie_theater',
|
|
content: Buffer.from('set sv_voice "0"\n') }
|
|
]) {
|
|
const { script } = createScript(artifact);
|
|
db().prepare('UPDATE scripts SET current_version_id = ? WHERE id = ?')
|
|
.run(getLatestVersion(script.id)!.id, script.id);
|
|
}
|
|
}
|
|
|
|
describe('server creation map selection', () => {
|
|
it('records in-flight creation and retains both operations after destruction', async () => {
|
|
const { createServer, destroyServer } = await import('../../src/lib/server/fleet/create.js');
|
|
const { listDeployments } = await import('../../src/lib/server/fleet/deployments.js');
|
|
const { komodoDeploy, komodoInspectHealthy, komodoDestroyDeployment } = await import('../../src/lib/server/integrations/komodo.js');
|
|
vi.mocked(komodoDeploy).mockImplementationOnce(async () => {
|
|
expect(listDeployments()).toContainEqual(expect.objectContaining({ server: serverName, action: 'create', state: 'running', stage: 'Starting container' }));
|
|
return { ok: true };
|
|
});
|
|
expect((await createServer(input)).ok).toBe(true);
|
|
vi.mocked(komodoInspectHealthy).mockResolvedValueOnce({ ok: false, reason: 'gone' });
|
|
vi.mocked(komodoDestroyDeployment).mockImplementationOnce(async () => {
|
|
expect(listDeployments()).toContainEqual(expect.objectContaining({ action: 'destroy', state: 'running' }));
|
|
return { ok: true };
|
|
});
|
|
expect((await destroyServer(serverName)).ok).toBe(true);
|
|
expect(listDeployments()).toEqual(expect.arrayContaining([
|
|
expect.objectContaining({ server: serverName, action: 'create', state: 'done', server_exists: 0 }),
|
|
expect.objectContaining({ server: serverName, action: 'destroy', state: 'done', server_exists: 0 })
|
|
]));
|
|
});
|
|
|
|
it('records failed and unexpectedly interrupted creation attempts', async () => {
|
|
const { createServer } = await import('../../src/lib/server/fleet/create.js');
|
|
const { listDeployments, beginDeployment, interruptDeployments } = await import('../../src/lib/server/fleet/deployments.js');
|
|
const { komodoDeploy } = await import('../../src/lib/server/integrations/komodo.js');
|
|
vi.mocked(komodoDeploy).mockResolvedValueOnce({ ok: false, kind: 'http-error', message: 'test failure' });
|
|
expect((await createServer(input)).ok).toBe(false);
|
|
expect(listDeployments()).toContainEqual(expect.objectContaining({ state: 'failed', error: 'Deploy: test failure' }));
|
|
vi.mocked(komodoDeploy).mockRejectedValueOnce(new Error('connection lost'));
|
|
await expect(createServer(input)).rejects.toThrow('connection lost');
|
|
expect(listDeployments()).toContainEqual(expect.objectContaining({ state: 'failed', error: 'connection lost' }));
|
|
const id = beginDeployment('interrupted-server', 'destroy');
|
|
interruptDeployments();
|
|
expect(listDeployments()).toContainEqual(expect.objectContaining({ id, state: 'interrupted', finished_at: expect.any(String) }));
|
|
});
|
|
|
|
it('installs only promoted scope-matching scripts and plugins before first boot', async () => {
|
|
const { createScript } = await import('../../src/lib/server/fleet/scripts.js');
|
|
const { db } = await import('../../src/lib/server/db/index.js');
|
|
const { destOfKind } = await import('../../src/lib/server/config.js');
|
|
const { getServerScript } = await import('../../src/lib/server/fleet/inventory.js');
|
|
const { createServer } = await import('../../src/lib/server/fleet/create.js');
|
|
const { komodoDeploy } = await import('../../src/lib/server/integrations/komodo.js');
|
|
const { scanServerDrift } = await import('../../src/lib/server/fleet/drift.js');
|
|
const artifacts = [
|
|
{ kind: 'script' as const, target: 'zm_spawn_fix.gsc', scopeType: 'game' as const, scopeValue: 't5', promoted: true, expected: true },
|
|
{ kind: 'plugin' as const, target: 'plugin.dll', scopeType: 'map' as const, scopeValue: 'zombie_cosmodrome', promoted: true, expected: true },
|
|
{ kind: 'script' as const, target: 'other.gsc', scopeType: 'map' as const, scopeValue: 'zombie_theater', promoted: true, expected: false },
|
|
{ kind: 'script' as const, target: 'draft.gsc', scopeType: 'game' as const, scopeValue: 't5', promoted: false, expected: false }
|
|
].map((a) => {
|
|
const { script, version } = createScript({ name: a.target, kind: a.kind, targetRelPath: a.target,
|
|
scopeType: a.scopeType, scopeValue: a.scopeValue, content: Buffer.from(a.target) });
|
|
if (a.promoted) db().prepare('UPDATE scripts SET current_version_id = ? WHERE id = ?').run(version.id, script.id);
|
|
return { ...a, version };
|
|
});
|
|
vi.mocked(komodoDeploy).mockImplementationOnce(async () => {
|
|
for (const a of artifacts) {
|
|
const path = join(destOfKind(serverName, a.kind), a.target);
|
|
expect(existsSync(path)).toBe(a.expected);
|
|
if (a.expected) {
|
|
expect(await readFile(path)).toEqual(a.version.content);
|
|
expect(getServerScript(serverName, a.target)).toMatchObject({ script_version_id: a.version.id, source: 'create' });
|
|
}
|
|
}
|
|
return { ok: true };
|
|
});
|
|
expect(await createServer(input)).toMatchObject({ ok: true });
|
|
expect(await scanServerDrift(serverName)).toEqual([]);
|
|
});
|
|
|
|
it('keeps Plutainer defaults without a baseline and overrides the map before rotation', async () => {
|
|
const { createServer } = await import('../../src/lib/server/fleet/create.js');
|
|
const { komodoCreateDeployment, komodoDeploy } = await import('../../src/lib/server/integrations/komodo.js');
|
|
const { getServer } = await import('../../src/lib/server/fleet/servers.js');
|
|
const result = await createServer(input);
|
|
expect(result).toMatchObject({ ok: true, name: serverName });
|
|
expect(existsSync(join(fixture.t5, serverName, 'configs', 'dedicated_sp.cfg'))).toBe(false);
|
|
const spec = vi.mocked(komodoCreateDeployment).mock.calls[0][0];
|
|
const env = Object.fromEntries(spec.environment.map(({ variable, value }) => [variable, value]));
|
|
expect(env.PLUTAINER_MAP_ROTATE).toBe('false');
|
|
expect(env.PLUTAINER_EXTRA_ARGS).toBe(`+exec electric_cherry_map.cfg +set tws_server_id ${serverName} +map_rotate`);
|
|
expect(await readFile(join(fixture.t5, serverName, 'configs', 'electric_cherry_map.cfg'), 'utf8'))
|
|
.toContain('set sv_maprotation "map zombie_cosmodrome"');
|
|
expect(env.PLUTAINER_CONFIG_FILE).toBe('dedicated_sp.cfg');
|
|
expect(env.PLUTAINER_SKIP_SEED).toBeUndefined();
|
|
expect(komodoDeploy).toHaveBeenCalledWith(serverName);
|
|
expect(getServer(serverName)?.map).toBe('zombie_cosmodrome');
|
|
const { scanServerDrift, runDriftDeployment } = await import('../../src/lib/server/fleet/drift.js');
|
|
expect(await scanServerDrift(serverName)).toEqual([]);
|
|
const mapPath = join(fixture.t5, serverName, 'configs', 'electric_cherry_map.cfg');
|
|
await rm(mapPath);
|
|
expect((await runDriftDeployment({ servers: [serverName] })).totalRestored).toBe(1);
|
|
expect(await readFile(mapPath, 'utf8')).toContain('set sv_maprotation "map zombie_cosmodrome"');
|
|
expect(await scanServerDrift(serverName)).toEqual([]);
|
|
});
|
|
|
|
it('writes promoted settings and selected map before first boot and tracks the file', async () => {
|
|
await promotedConfig();
|
|
const { createServer } = await import('../../src/lib/server/fleet/create.js');
|
|
const { komodoDeploy } = await import('../../src/lib/server/integrations/komodo.js');
|
|
const { getServerScript } = await import('../../src/lib/server/fleet/inventory.js');
|
|
const { scanServerDrift } = await import('../../src/lib/server/fleet/drift.js');
|
|
vi.mocked(komodoDeploy).mockImplementationOnce(async () => {
|
|
const config = await readFile(join(fixture.t5, serverName, 'configs', 'dedicated_sp.cfg'), 'utf8');
|
|
expect(config).toContain('set sv_maprotation "map zombie_cosmodrome"');
|
|
expect(config).toContain('set sv_voice "1"');
|
|
expect(config).toContain('set g_inactivity "0"');
|
|
expect(config).not.toContain('map zombie_moon');
|
|
expect(getServerScript(serverName, 'dedicated_sp.cfg')?.source).toBe('create');
|
|
return { ok: true };
|
|
});
|
|
expect(await createServer(input)).toMatchObject({ ok: true });
|
|
expect(await scanServerDrift(serverName)).toEqual([]);
|
|
});
|
|
|
|
it('removes the generated config and inventory if deployment fails', async () => {
|
|
await promotedConfig();
|
|
const { createScript } = await import('../../src/lib/server/fleet/scripts.js');
|
|
const { db } = await import('../../src/lib/server/db/index.js');
|
|
const { script, version } = createScript({ name: 'spawn fix', kind: 'script', scopeType: 'game', scopeValue: 't5',
|
|
targetRelPath: 'zm_spawn_fix.gsc', content: Buffer.from('main(){}') });
|
|
db().prepare('UPDATE scripts SET current_version_id = ? WHERE id = ?').run(version.id, script.id);
|
|
const { createServer } = await import('../../src/lib/server/fleet/create.js');
|
|
const { komodoDeploy } = await import('../../src/lib/server/integrations/komodo.js');
|
|
const { getServer } = await import('../../src/lib/server/fleet/servers.js');
|
|
const { getServerScript } = await import('../../src/lib/server/fleet/inventory.js');
|
|
vi.mocked(komodoDeploy).mockResolvedValueOnce({ ok: false, kind: 'http-error', message: 'test failure' });
|
|
expect(await createServer(input)).toMatchObject({ ok: false, kind: 'komodo' });
|
|
expect(getServer(serverName)).toBeUndefined();
|
|
expect(getServerScript(serverName, 'dedicated_sp.cfg')).toBeUndefined();
|
|
expect(getServerScript(serverName, 'zm_spawn_fix.gsc')).toBeUndefined();
|
|
expect(existsSync(join(fixture.t5, serverName))).toBe(false);
|
|
});
|
|
|
|
it('rejects unknown maps before creating a deployment', async () => {
|
|
const { createServer } = await import('../../src/lib/server/fleet/create.js');
|
|
const { komodoCreateDeployment } = await import('../../src/lib/server/integrations/komodo.js');
|
|
expect(await createServer({ ...input, mapSlug: 'unknown' })).toMatchObject({ ok: false, kind: 'unknown-map' });
|
|
expect(komodoCreateDeployment).not.toHaveBeenCalled();
|
|
});
|
|
});
|
|
|
|
|
|
describe('persistent server keys', () => {
|
|
it('saves a key, retains it through destruction, and reuses it for the exact lobby', async () => {
|
|
const { createServer, destroyServer } = await import('../../src/lib/server/fleet/create.js');
|
|
const { getServerKey, listServerKeys } = await import('../../src/lib/server/fleet/server-keys.js');
|
|
const { komodoCreateDeployment, komodoInspectHealthy } = await import('../../src/lib/server/integrations/komodo.js');
|
|
vi.mocked(komodoInspectHealthy).mockResolvedValue({ ok: false, kind: 'not-found', message: 'gone' } as never);
|
|
const name = 't5_zombie_cosmodrome_2';
|
|
expect(await createServer(input)).toMatchObject({ ok: true, name: serverName });
|
|
expect(await createServer(input)).toMatchObject({ ok: true, name });
|
|
expect(getServerKey(name)).toBe('test-key');
|
|
expect(await destroyServer(name)).toMatchObject({ ok: true });
|
|
expect(listServerKeys()).toContainEqual(expect.objectContaining({ server_id: name, deployed: 0 }));
|
|
expect(JSON.stringify(listServerKeys())).not.toContain('test-key');
|
|
expect(await createServer({ gameMode: 'zombie', mapSlug: 'cosmodrome', expectedName: name })).toMatchObject({ ok: true, name });
|
|
const spec = vi.mocked(komodoCreateDeployment).mock.calls.at(-1)![0];
|
|
expect(spec.environment).toContainEqual({ variable: 'PLUTO_SERVER_KEY', value: 'test-key' });
|
|
expect(await createServer({ gameMode: 'zombie', mapSlug: 'cosmodrome' })).toMatchObject({ ok: false, kind: 'bad-request' });
|
|
});
|
|
|
|
it('updates and removes keys independently of deployments and survives reopening the database', async () => {
|
|
const { saveServerKey, getServerKey, removeServerKey } = await import('../../src/lib/server/fleet/server-keys.js');
|
|
const { closeDb } = await import('../../src/lib/server/db/index.js');
|
|
saveServerKey(serverName, ' first-key ');
|
|
closeDb();
|
|
expect(getServerKey(serverName)).toBe('first-key');
|
|
saveServerKey(serverName, 'replacement');
|
|
expect(getServerKey(serverName)).toBe('replacement');
|
|
expect(removeServerKey(serverName)).toBe(true);
|
|
expect(getServerKey(serverName)).toBeUndefined();
|
|
expect(() => saveServerKey('t5_zombie_unknown_1', 'key')).toThrow();
|
|
expect(() => saveServerKey(serverName, ' ')).toThrow();
|
|
expect(() => saveServerKey(serverName, 'two words')).toThrow();
|
|
});
|
|
|
|
it('fills the lowest gap and rejects stale previews before saving a key or deploying', async () => {
|
|
const { createServer, destroyServer } = await import('../../src/lib/server/fleet/create.js');
|
|
const { getServerKey } = await import('../../src/lib/server/fleet/server-keys.js');
|
|
const { komodoCreateDeployment, komodoInspectHealthy } = await import('../../src/lib/server/integrations/komodo.js');
|
|
vi.mocked(komodoInspectHealthy).mockResolvedValue({ ok: false } as never);
|
|
expect(await createServer(input)).toMatchObject({ ok: true, name: serverName });
|
|
expect(await createServer(input)).toMatchObject({ ok: true, name: 't5_zombie_cosmodrome_2' });
|
|
await destroyServer(serverName);
|
|
expect(await createServer({ ...input, plutoKey: 'wrong-key', expectedName: 't5_zombie_cosmodrome_3' }))
|
|
.toMatchObject({ ok: false, kind: 'no-slot' });
|
|
expect(getServerKey(serverName)).toBe('test-key');
|
|
expect(komodoCreateDeployment).toHaveBeenCalledTimes(2);
|
|
expect(await createServer({ gameMode: 'zombie', mapSlug: 'cosmodrome', expectedName: serverName }))
|
|
.toMatchObject({ ok: true, name: serverName });
|
|
});
|
|
});
|