56 lines
1.5 KiB
TypeScript
56 lines
1.5 KiB
TypeScript
import { describe, expect, it } from "vitest";
|
|
|
|
import { enqueueCommand, getQueueSize } from "./command-queue.js";
|
|
|
|
describe("command queue", () => {
|
|
it("runs tasks one at a time in order", async () => {
|
|
let active = 0;
|
|
let maxActive = 0;
|
|
const calls: number[] = [];
|
|
|
|
const makeTask = (id: number) => async () => {
|
|
active += 1;
|
|
maxActive = Math.max(maxActive, active);
|
|
calls.push(id);
|
|
await new Promise((resolve) => setTimeout(resolve, 15));
|
|
active -= 1;
|
|
return id;
|
|
};
|
|
|
|
const results = await Promise.all([
|
|
enqueueCommand(makeTask(1)),
|
|
enqueueCommand(makeTask(2)),
|
|
enqueueCommand(makeTask(3)),
|
|
]);
|
|
|
|
expect(results).toEqual([1, 2, 3]);
|
|
expect(calls).toEqual([1, 2, 3]);
|
|
expect(maxActive).toBe(1);
|
|
expect(getQueueSize()).toBe(0);
|
|
});
|
|
|
|
it("invokes onWait callback when a task waits past the threshold", async () => {
|
|
let waited: number | null = null;
|
|
let queuedAhead: number | null = null;
|
|
|
|
// First task holds the queue long enough to trigger wait notice.
|
|
const first = enqueueCommand(async () => {
|
|
await new Promise((resolve) => setTimeout(resolve, 30));
|
|
});
|
|
|
|
const second = enqueueCommand(async () => {}, {
|
|
warnAfterMs: 5,
|
|
onWait: (ms, ahead) => {
|
|
waited = ms;
|
|
queuedAhead = ahead;
|
|
},
|
|
});
|
|
|
|
await Promise.all([first, second]);
|
|
|
|
expect(waited).not.toBeNull();
|
|
expect(waited as number).toBeGreaterThanOrEqual(5);
|
|
expect(queuedAhead).toBe(0);
|
|
});
|
|
});
|