test: combine web reconnect progression assertions

This commit is contained in:
Peter Steinberger
2026-02-23 21:57:30 +00:00
parent ecd278b67b
commit 8b192beaaf

View File

@@ -91,40 +91,59 @@ describe("web auto-reply", () => {
expect(() => formatEnvelopeTimestamp(d, " America/Los_Angeles ")).not.toThrow();
});
it("reconnects after a connection close", async () => {
const closeResolvers: Array<() => void> = [];
const sleep = vi.fn(async () => {});
const listenerFactory = vi.fn(async () => {
let _resolve!: () => void;
const onClose = new Promise<void>((res) => {
_resolve = res;
closeResolvers.push(res);
});
return { close: vi.fn(), onClose };
});
const { runtime, controller, run } = startMonitorWebChannel({
monitorWebChannelFn: monitorWebChannel as never,
listenerFactory,
sleep,
});
await Promise.resolve();
expect(listenerFactory).toHaveBeenCalledTimes(1);
closeResolvers[0]?.();
await vi.waitFor(
() => {
expect(listenerFactory).toHaveBeenCalledTimes(2);
it("handles reconnect progress and max-attempt stop behavior", async () => {
for (const scenario of [
{
reconnect: { initialMs: 10, maxMs: 10, maxAttempts: 3, factor: 1.1 },
expectedCallsAfterFirstClose: 2,
closeTwiceAndFinish: false,
expectedError: "Retry 1",
},
{ timeout: 500, interval: 5 },
);
expect(listenerFactory).toHaveBeenCalledTimes(2);
expect(runtime.error).toHaveBeenCalledWith(expect.stringContaining("Retry 1"));
{
reconnect: { initialMs: 5, maxMs: 5, maxAttempts: 2, factor: 1.1 },
expectedCallsAfterFirstClose: 2,
closeTwiceAndFinish: true,
expectedError: "max attempts reached",
},
]) {
const closeResolvers: Array<() => void> = [];
const sleep = vi.fn(async () => {});
const listenerFactory = vi.fn(async () => {
const onClose = new Promise<void>((res) => {
closeResolvers.push(res);
});
return { close: vi.fn(), onClose };
});
const { runtime, controller, run } = startMonitorWebChannel({
monitorWebChannelFn: monitorWebChannel as never,
listenerFactory,
sleep,
reconnect: scenario.reconnect,
});
controller.abort();
closeResolvers[1]?.();
await Promise.resolve();
await run;
await Promise.resolve();
expect(listenerFactory).toHaveBeenCalledTimes(1);
closeResolvers.shift()?.();
await vi.waitFor(
() => {
expect(listenerFactory).toHaveBeenCalledTimes(scenario.expectedCallsAfterFirstClose);
},
{ timeout: 500, interval: 5 },
);
if (scenario.closeTwiceAndFinish) {
closeResolvers.shift()?.();
await run;
} else {
controller.abort();
closeResolvers.shift()?.();
await Promise.resolve();
await run;
}
expect(runtime.error).toHaveBeenCalledWith(expect.stringContaining(scenario.expectedError));
}
});
it("forces reconnect when watchdog closes without onClose", async () => {
vi.useFakeTimers();
@@ -205,51 +224,6 @@ describe("web auto-reply", () => {
}
}, 15_000);
it("stops after hitting max reconnect attempts", { timeout: 60_000 }, async () => {
const closeResolvers: Array<() => void> = [];
const sleep = vi.fn(async () => {});
const listenerFactory = vi.fn(async () => {
const onClose = new Promise<void>((res) => closeResolvers.push(res));
return { close: vi.fn(), onClose };
});
const runtime = {
log: vi.fn(),
error: vi.fn(),
exit: vi.fn(),
};
const run = monitorWebChannel(
false,
listenerFactory as never,
true,
async () => ({ text: "ok" }),
runtime as never,
undefined,
{
heartbeatSeconds: 1,
reconnect: { initialMs: 5, maxMs: 5, maxAttempts: 2, factor: 1.1 },
sleep,
},
);
await Promise.resolve();
expect(listenerFactory).toHaveBeenCalledTimes(1);
closeResolvers.shift()?.();
await vi.waitFor(
() => {
expect(listenerFactory).toHaveBeenCalledTimes(2);
},
{ timeout: 500, interval: 5 },
);
expect(listenerFactory).toHaveBeenCalledTimes(2);
closeResolvers.shift()?.();
await run;
expect(runtime.error).toHaveBeenCalledWith(expect.stringContaining("max attempts reached"));
});
it("processes inbound messages without batching and preserves timestamps", async () => {
await withEnvAsync({ TZ: "Europe/Vienna" }, async () => {
const originalMax = process.getMaxListeners();