Files
SnowLuma/packages/protocol/tests/highway/upload-nodes.test.ts
T
TanTan fa9c73363d
Dev Build / cleanup (push) Waiting to run
Dev Build / typecheck (push) Waiting to run
Publish MCP to npm / Build, test, and publish MCP (push) Waiting to run
Publish SDK to npm / Build, test, and publish SDK (push) Waiting to run
Sync action catalog to docs / sync (push) Waiting to run
Publish UI to npm / Build and publish UI (push) Waiting to run
Build and Release / preflight (push) Waiting to run
SnowLuma v2.4.2 source snapshot
2026-11-13 09:02:00 +08:00

57 lines
2.2 KiB
TypeScript

import { afterEach, describe, expect, it, vi } from 'vitest';
import { UploadNodeHistory } from '../../src/highway/upload-nodes';
const a = { host: '192.0.2.1', port: 14000 };
const b = { host: '192.0.2.1', port: 80 };
const c = { host: '192.0.2.2', port: 80 };
afterEach(() => vi.useRealTimers());
describe('upload node observations', () => {
it('orders current candidates by confirmed success, unknown, then recent failure', () => {
const history = new UploadNodeHistory();
history.record(1004, a, false);
history.record(1004, b, true);
const original = [a, c, b, a];
expect(history.order(1004, original)).toEqual([b, c, a]);
expect(original).toEqual([a, c, b, a]);
expect(history.order(1003, [a, c, b])).toEqual([a, c, b]);
expect(history.order(1004, [c])).toEqual([c]);
});
it('expires failures and successes and still allows every failed candidate', () => {
vi.useFakeTimers();
const history = new UploadNodeHistory();
history.record(1004, a, false);
history.record(1004, b, false);
expect(history.order(1004, [a, b])).toEqual([a, b]);
expect(history.order(1004, [a, c])).toEqual([c, a]);
vi.advanceTimersByTime(30_000);
expect(history.order(1004, [a, c])).toEqual([a, c]);
history.record(1004, b, true);
expect(history.order(1004, [a, b])).toEqual([b, a]);
vi.advanceTimersByTime(5 * 60_000);
expect(history.order(1004, [a, b])).toEqual([a, b]);
});
it('lets a new failure replace success and a subsequent success recover', () => {
const history = new UploadNodeHistory();
history.record(1004, b, true);
history.record(1004, b, false);
expect(history.order(1004, [b, a])).toEqual([a, b]);
history.record(1004, b, true);
expect(history.order(1004, [a, b])).toEqual([b, a]);
});
it('evicts old observations when many distinct endpoints arrive', () => {
const history = new UploadNodeHistory();
history.record(1004, a, false);
for (let port = 10000; port < 10200; port++) {
history.record(1004, { host: '192.0.2.3', port }, false);
}
expect(history.order(1004, [a, c])).toEqual([a, c]);
});
});