SnowLuma v2.4.2 source snapshot

packages/protocol/tests/highway/upload-nodes.test.ts

@@ -0,0 +1,56 @@
+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]);
+    });
+});