LLM/moltbot
1
0
Fork 0
mirror of https://github.com/moltbot/moltbot.git synced 2026-05-20 05:07:34 +00:00
Code Issues Packages Projects Releases Wiki Activity

test: guard discord gateway mock calls

Browse Source
This commit is contained in:
Peter Steinberger
2026-05-12 02:01:54 +01:00
parent 95b6c10fc7
commit e649cb92c0
1 changed files with 19 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

21
extensions/discord/src/internal/gateway.test.ts
View File

@@ -29,6 +29,23 @@ function sentGatewayOpcodes(send: ReturnType<typeof attachOpenSocket>) {
});
}
function firstDispatchedData(dispatchGatewayEvent: ReturnType<typeof vi.fn>): unknown {
const [call] = dispatchGatewayEvent.mock.calls;
if (!call) {
throw new Error("Expected dispatched gateway event call");
}
return call[1];
}
function firstSentGatewayPayload(send: ReturnType<typeof attachOpenSocket>): unknown {
const [call] = send.mock.calls;
if (!call) {
throw new Error("Expected gateway socket send call");
}
const [rawPayload] = call;
return JSON.parse(String(rawPayload));
}
function presenceUpdate(
status: PresenceUpdateStatus.Online | PresenceUpdateStatus.Idle = PresenceUpdateStatus.Online,
): GatewaySendPayload {
@@ -201,7 +218,7 @@ describe("GatewayPlugin", () => {
});
expect(dispatchGatewayEvent).toHaveBeenCalledTimes(1);
const dispatched = dispatchGatewayEvent.mock.calls[0]?.[1] as {
const dispatched = firstDispatchedData(dispatchGatewayEvent) as {
author?: { id: string };
message?: { author?: { id: string } | null; content?: string };
};
@@ -606,7 +623,7 @@ describe("GatewayPlugin", () => {
valid.requestGuildMembers({ guild_id: "guild1", query: "", limit: 0, presences: true });
expect(send).toHaveBeenCalledTimes(1);
expect(JSON.parse(send.mock.calls[0]?.[0] as string)).toEqual({
expect(firstSentGatewayPayload(send)).toEqual({
op: GatewayOpcodes.RequestGuildMembers,
d: { guild_id: "guild1", query: "", limit: 0, presences: true },
});