File size: 3,919 Bytes
fc93158 | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 104 105 106 107 108 109 110 111 112 113 114 115 116 117 118 119 120 121 122 123 124 125 126 | import { randomUUID } from "node:crypto";
import { afterAll, describe, expect, it } from "vitest";
import { GatewayClient } from "../src/gateway/client.js";
import { connectGatewayClient } from "../src/gateway/test-helpers.e2e.js";
import { GATEWAY_CLIENT_MODES, GATEWAY_CLIENT_NAMES } from "../src/utils/message-channel.js";
import {
type ChatEventPayload,
type GatewayInstance,
connectNode,
extractFirstTextBlock,
postJson,
spawnGatewayInstance,
stopGatewayInstance,
waitForChatFinalEvent,
waitForNodeStatus,
} from "./helpers/gateway-e2e-harness.js";
const E2E_TIMEOUT_MS = 120_000;
describe("gateway multi-instance e2e", () => {
const instances: GatewayInstance[] = [];
const nodeClients: GatewayClient[] = [];
const chatClients: GatewayClient[] = [];
afterAll(async () => {
for (const client of nodeClients) {
client.stop();
}
for (const client of chatClients) {
client.stop();
}
for (const inst of instances) {
await stopGatewayInstance(inst);
}
});
it(
"spins up two gateways and exercises WS + HTTP + node pairing",
{ timeout: E2E_TIMEOUT_MS },
async () => {
const [gwA, gwB] = await Promise.all([spawnGatewayInstance("a"), spawnGatewayInstance("b")]);
instances.push(gwA, gwB);
const [hookResA, hookResB] = await Promise.all([
postJson(
`http://127.0.0.1:${gwA.port}/hooks/wake`,
{
text: "wake a",
mode: "now",
},
{ "x-openclaw-token": gwA.hookToken },
),
postJson(
`http://127.0.0.1:${gwB.port}/hooks/wake`,
{
text: "wake b",
mode: "now",
},
{ "x-openclaw-token": gwB.hookToken },
),
]);
expect(hookResA.status).toBe(200);
expect((hookResA.json as { ok?: boolean } | undefined)?.ok).toBe(true);
expect(hookResB.status).toBe(200);
expect((hookResB.json as { ok?: boolean } | undefined)?.ok).toBe(true);
const [nodeA, nodeB] = await Promise.all([
connectNode(gwA, "node-a"),
connectNode(gwB, "node-b"),
]);
nodeClients.push(nodeA.client, nodeB.client);
await Promise.all([
waitForNodeStatus(gwA, nodeA.nodeId),
waitForNodeStatus(gwB, nodeB.nodeId),
]);
},
);
it(
"delivers final chat event for telegram-shaped session keys",
{ timeout: E2E_TIMEOUT_MS },
async () => {
const gw = await spawnGatewayInstance("chat-telegram-fixture");
instances.push(gw);
const chatEvents: ChatEventPayload[] = [];
const chatClient = await connectGatewayClient({
url: `ws://127.0.0.1:${gw.port}`,
token: gw.gatewayToken,
clientName: GATEWAY_CLIENT_NAMES.CLI,
clientDisplayName: "chat-e2e-cli",
clientVersion: "1.0.0",
platform: "test",
mode: GATEWAY_CLIENT_MODES.CLI,
onEvent: (evt) => {
if (evt.event === "chat" && evt.payload && typeof evt.payload === "object") {
chatEvents.push(evt.payload as ChatEventPayload);
}
},
});
chatClients.push(chatClient);
const sessionKey = "agent:main:telegram:direct:123456";
const idempotencyKey = `idem-${randomUUID()}`;
const sendRes = await chatClient.request<{ runId?: string; status?: string }>("chat.send", {
sessionKey,
message: "/context list",
idempotencyKey,
});
expect(sendRes.status).toBe("started");
const runId = sendRes.runId;
expect(typeof runId).toBe("string");
const finalEvent = await waitForChatFinalEvent({
events: chatEvents,
runId: String(runId),
sessionKey,
});
const finalText = extractFirstTextBlock(finalEvent.message);
expect(typeof finalText).toBe("string");
expect(finalText?.length).toBeGreaterThan(0);
},
);
});
|