From 490a4c5d9bb0879b56ddd56307f394e7ff9b1201 Mon Sep 17 00:00:00 2001
From: zengLingbiao <zeng.lingbiao@xydigit.com>
Date: Wed, 1 Jul 2026 08:53:56 +0800
Subject: [PATCH] test(gateway): add unit tests for node wake state tracking
 and testing seam (#98205)

Add 17 unit tests for nodes-wake-state.ts covering:
- Exported wait/poll constants
- nodeWakeById Map operations (insert, overwrite, multi-entry, inFlight)
- nodeWakeNudgeById Map operations (independent tracking)
- clearNodeWakeState function (removal, no-op, scoped deletion)
- testing seam (getNodeWakeByIdSize, hasNodeWakeEntry, resetWakeState)

Co-authored-by: Claude Opus 4.8 <noreply@anthropic.com>
---
 .../server-methods/nodes-wake-state.test.ts   | 136 ++++++++++++++++++
 1 file changed, 136 insertions(+)
 create mode 100644 src/gateway/server-methods/nodes-wake-state.test.ts

diff --git a/src/gateway/server-methods/nodes-wake-state.test.ts b/src/gateway/server-methods/nodes-wake-state.test.ts
new file mode 100644
index 00000000000..b181da2d83d
--- /dev/null
+++ b/src/gateway/server-methods/nodes-wake-state.test.ts
@@ -0,0 +1,136 @@
+// Tests for node wake state tracking and testing seam.
+import { afterEach, describe, expect, it } from "vitest";
+import {
+  NODE_WAKE_RECONNECT_WAIT_MS,
+  NODE_WAKE_RECONNECT_RETRY_WAIT_MS,
+  NODE_WAKE_RECONNECT_POLL_MS,
+  clearNodeWakeState,
+  nodeWakeById,
+  nodeWakeNudgeById,
+  testing,
+} from "./nodes-wake-state.js";
+
+afterEach(() => {
+  testing.resetWakeState();
+});
+
+describe("constants", () => {
+  it("exports expected wait/poll constants", () => {
+    expect(NODE_WAKE_RECONNECT_WAIT_MS).toBe(3_000);
+    expect(NODE_WAKE_RECONNECT_RETRY_WAIT_MS).toBe(12_000);
+    expect(NODE_WAKE_RECONNECT_POLL_MS).toBe(150);
+  });
+});
+
+describe("nodeWakeById", () => {
+  it("starts empty", () => {
+    expect(nodeWakeById.size).toBe(0);
+  });
+
+  it("stores NodeWakeState entries", () => {
+    const now = Date.now();
+    nodeWakeById.set("node-1", { lastWakeAtMs: now });
+    expect(nodeWakeById.size).toBe(1);
+    expect(nodeWakeById.get("node-1")?.lastWakeAtMs).toBe(now);
+  });
+
+  it("stores multiple entries", () => {
+    nodeWakeById.set("a", { lastWakeAtMs: 100 });
+    nodeWakeById.set("b", { lastWakeAtMs: 200 });
+    nodeWakeById.set("c", { lastWakeAtMs: 300 });
+    expect(nodeWakeById.size).toBe(3);
+  });
+
+  it("overwrites existing entry when key is reused", () => {
+    nodeWakeById.set("node-1", { lastWakeAtMs: 100 });
+    nodeWakeById.set("node-1", { lastWakeAtMs: 200 });
+    expect(nodeWakeById.size).toBe(1);
+  });
+
+  it("supports inFlight promise property", () => {
+    const promise = Promise.resolve({
+      available: true,
+      throttled: false,
+      path: "sent" as const,
+      durationMs: 50,
+    });
+    nodeWakeById.set("node-1", { lastWakeAtMs: Date.now(), inFlight: promise });
+    expect(nodeWakeById.get("node-1")?.inFlight).toBe(promise);
+  });
+});
+
+describe("nodeWakeNudgeById", () => {
+  it("starts empty", () => {
+    expect(nodeWakeNudgeById.size).toBe(0);
+  });
+
+  it("stores nudge timestamps", () => {
+    nodeWakeNudgeById.set("node-1", 1000);
+    expect(nodeWakeNudgeById.size).toBe(1);
+    expect(nodeWakeNudgeById.get("node-1")).toBe(1000);
+  });
+
+  it("independently tracked from nodeWakeById", () => {
+    nodeWakeById.set("node-1", { lastWakeAtMs: 500 });
+    nodeWakeNudgeById.set("node-1", 1000);
+    expect(nodeWakeById.size).toBe(1);
+    expect(nodeWakeNudgeById.size).toBe(1);
+  });
+});
+
+describe("clearNodeWakeState", () => {
+  it("removes the wake entry and nudge for the given node", () => {
+    nodeWakeById.set("node-1", { lastWakeAtMs: 100 });
+    nodeWakeNudgeById.set("node-1", 200);
+    clearNodeWakeState("node-1");
+    expect(nodeWakeById.has("node-1")).toBe(false);
+    expect(nodeWakeNudgeById.has("node-1")).toBe(false);
+  });
+
+  it("is a no-op when the node id does not exist", () => {
+    expect(() => clearNodeWakeState("ghost")).not.toThrow();
+    expect(nodeWakeById.size).toBe(0);
+    expect(nodeWakeNudgeById.size).toBe(0);
+  });
+
+  it("only removes the specified node, leaving others intact", () => {
+    nodeWakeById.set("a", { lastWakeAtMs: 1 });
+    nodeWakeById.set("b", { lastWakeAtMs: 2 });
+    nodeWakeNudgeById.set("a", 10);
+    nodeWakeNudgeById.set("b", 20);
+    clearNodeWakeState("a");
+    expect(nodeWakeById.has("a")).toBe(false);
+    expect(nodeWakeById.has("b")).toBe(true);
+    expect(nodeWakeNudgeById.has("a")).toBe(false);
+    expect(nodeWakeNudgeById.has("b")).toBe(true);
+  });
+});
+
+describe("testing seam", () => {
+  it("getNodeWakeByIdSize returns 0 when nothing is stored", () => {
+    expect(testing.getNodeWakeByIdSize()).toBe(0);
+  });
+
+  it("getNodeWakeByIdSize reflects inserted entries", () => {
+    nodeWakeById.set("x", { lastWakeAtMs: 1 });
+    nodeWakeById.set("y", { lastWakeAtMs: 2 });
+    expect(testing.getNodeWakeByIdSize()).toBe(2);
+  });
+
+  it("hasNodeWakeEntry returns true for stored ids", () => {
+    nodeWakeById.set("present", { lastWakeAtMs: 1 });
+    expect(testing.hasNodeWakeEntry("present")).toBe(true);
+  });
+
+  it("hasNodeWakeEntry returns false for unknown ids", () => {
+    expect(testing.hasNodeWakeEntry("missing")).toBe(false);
+  });
+
+  it("resetWakeState clears both maps", () => {
+    nodeWakeById.set("a", { lastWakeAtMs: 1 });
+    nodeWakeNudgeById.set("a", 100);
+    testing.resetWakeState();
+    expect(nodeWakeById.size).toBe(0);
+    expect(nodeWakeNudgeById.size).toBe(0);
+  });
+});