@wingman-ai/gateway 0.2.0 → 0.2.2

package/dist/tests/gateway.test.js

@@ -1,14 +1,51 @@
 import { afterAll, beforeAll, describe, expect, it, vi } from "vitest";
 import { GatewayClient, GatewayServer } from "../gateway/index.js";
+function _define_property(obj, key, value) {
+    if (key in obj) Object.defineProperty(obj, key, {
+        value: value,
+        enumerable: true,
+        configurable: true,
+        writable: true
+    });
+    else obj[key] = value;
+    return obj;
+}
 const isBun = void 0 !== globalThis.Bun;
 const describeIfBun = isBun ? describe : describe.skip;
 vi.mock("@/cli/core/agentInvoker.js", ()=>({
         AgentInvoker: class {
-            async invokeAgent() {
+            async invokeAgent(_agentId, _content, _sessionId, _attachments, options) {
+                const signal = options?.signal;
+                await new Promise((resolve)=>{
+                    const timer = setTimeout(resolve, 75);
+                    if (signal) {
+                        const onAbort = ()=>{
+                            clearTimeout(timer);
+                            resolve();
+                        };
+                        if (signal.aborted) return void onAbort();
+                        signal.addEventListener("abort", onAbort, {
+                            once: true
+                        });
+                    }
+                });
+                if (signal?.aborted) {
+                    this.outputManager?.emitAgentError?.("Request cancelled");
+                    return {
+                        cancelled: true
+                    };
+                }
+                this.outputManager?.emitAgentComplete?.({
+                    streaming: true
+                });
                 return {
                     streaming: true
                 };
             }
+            constructor(options){
+                _define_property(this, "outputManager", void 0);
+                this.outputManager = options?.outputManager;
+            }
         }
     }));
 describeIfBun("Gateway", ()=>{
@@ -294,6 +331,34 @@ describeIfBun("Gateway", ()=>{
         desktopClient.close();
         requester.close();
     });
+    it("should cancel an in-flight agent request", async ()=>{
+        const requester = await connectClient("session-cancel-requester");
+        const requestId = "req-cancel-test";
+        requester.send(JSON.stringify({
+            type: "req:agent",
+            id: requestId,
+            payload: {
+                agentId: "main",
+                sessionKey: "session-cancel-test",
+                content: "cancel me"
+            },
+            timestamp: Date.now()
+        }));
+        requester.send(JSON.stringify({
+            type: "req:agent:cancel",
+            id: "cancel-req-cancel-test",
+            payload: {
+                requestId
+            },
+            timestamp: Date.now()
+        }));
+        const ack = await waitForMessage(requester, (msg)=>"ack" === msg.type && msg.payload?.action === "req:agent:cancel" && msg.payload?.requestId === requestId);
+        expect([
+            "cancelled",
+            "not_found"
+        ]).toContain(ack.payload?.status);
+        requester.close();
+    });
     it("should clear session messages via API", async ()=>{
         const createRes = await fetch(`http://localhost:${port}/api/sessions`, {
             method: "POST",