Fix native provisioning scheduling and progress reporting
This commit is contained in:
@@ -0,0 +1,76 @@
|
||||
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([]));
|
||||
});
|
||||
});
|
||||
Reference in New Issue
Block a user