77 lines
3.6 KiB
TypeScript
77 lines
3.6 KiB
TypeScript
import { afterEach, beforeEach, describe, expect, it, vi } from 'vitest';
|
|
import { newFixture, type Fixture } from '../helpers/fixture.js';
|
|
|
|
vi.mock('../../src/lib/server/pipeline/sweep.js', () => ({ sweep: vi.fn() }));
|
|
vi.mock('../../src/lib/server/pipeline/beta-runner.js', () => ({ betaTick: vi.fn() }));
|
|
vi.mock('../../src/lib/server/maintenance/index.js', () => ({ runMaintenance: vi.fn() }));
|
|
vi.mock('../../src/lib/server/provisioning/worker.js', () => ({ processProvisioning: vi.fn(async () => {}) }));
|
|
|
|
function deferred() {
|
|
let resolve!: () => void;
|
|
const promise = new Promise<void>(done => { resolve = done; });
|
|
return { promise, resolve };
|
|
}
|
|
let fixture: Fixture;
|
|
beforeEach(async () => { vi.resetModules(); vi.clearAllMocks(); fixture = await newFixture([]); });
|
|
afterEach(async () => { await fixture.cleanup(); });
|
|
|
|
async function setup() {
|
|
const { sweep } = await import('../../src/lib/server/pipeline/sweep.js');
|
|
const { betaTick } = await import('../../src/lib/server/pipeline/beta-runner.js');
|
|
const { runMaintenance } = await import('../../src/lib/server/maintenance/index.js');
|
|
const background = deferred();
|
|
vi.mocked(sweep).mockImplementation(() => background.promise as never);
|
|
vi.mocked(betaTick).mockImplementation(() => background.promise as never);
|
|
vi.mocked(runMaintenance).mockImplementation(() => background.promise);
|
|
const { processProvisioning } = await import('../../src/lib/server/provisioning/worker.js');
|
|
const worker = await import('../../src/lib/server/pipeline/worker.js');
|
|
const { workerActivity } = await import('../../src/lib/server/pipeline/activity.js');
|
|
return { ...worker, workerActivity, background, process: vi.mocked(processProvisioning), sweep, betaTick, runMaintenance };
|
|
}
|
|
|
|
describe('independent provisioning scheduling', () => {
|
|
it('admits and checks new work while campaign, beta and maintenance observers are still waiting', async () => {
|
|
const s = await setup();
|
|
try {
|
|
await s.kick();
|
|
expect(s.process).toHaveBeenCalledTimes(1);
|
|
expect(s.workerActivity().map(work => work.key)).toEqual(expect.arrayContaining(['campaign', 'beta', 'maintenance']));
|
|
await s.kick();
|
|
expect(s.process).toHaveBeenCalledTimes(2);
|
|
expect(s.sweep).toHaveBeenCalledTimes(1);
|
|
expect(s.betaTick).toHaveBeenCalledTimes(1);
|
|
expect(s.runMaintenance).toHaveBeenCalledTimes(1);
|
|
} finally {
|
|
s.background.resolve();
|
|
await vi.waitFor(() => expect(s.workerActivity()).toEqual([]));
|
|
}
|
|
});
|
|
|
|
it('coalesces overlapping kicks but makes another pass for requests arriving during execution', async () => {
|
|
const s = await setup();
|
|
const busy = deferred();
|
|
s.process.mockImplementationOnce(() => busy.promise);
|
|
const first = s.kick();
|
|
const second = s.kick();
|
|
const receipt = s.kickProvisioning();
|
|
try {
|
|
expect(s.process).toHaveBeenCalledTimes(1);
|
|
busy.resolve();
|
|
await Promise.all([first, second, receipt]);
|
|
expect(s.process).toHaveBeenCalledTimes(2);
|
|
} finally {
|
|
busy.resolve(); s.background.resolve();
|
|
await vi.waitFor(() => expect(s.workerActivity()).toEqual([]));
|
|
}
|
|
});
|
|
|
|
it('wakes provisioning when a background observer finishes submitting its requests', async () => {
|
|
const s = await setup();
|
|
await s.kick();
|
|
s.process.mockClear();
|
|
s.background.resolve();
|
|
await vi.waitFor(() => expect(s.process).toHaveBeenCalled());
|
|
await vi.waitFor(() => expect(s.workerActivity()).toEqual([]));
|
|
});
|
|
});
|