@spikard/node 0.11.0 → 0.12.0

package/dist/index.mjs

@@ -903,6 +903,36 @@ import fs from "fs/promises";
 import { createRequire as createRequire3 } from "module";
 import path from "path";
 import { gunzipSync, gzipSync } from "zlib";
+var GRAPHQL_WS_TIMEOUT_MS = 2e3;
+var GRAPHQL_WS_MAX_CONTROL_MESSAGES = 32;
+var withTimeout = async (promise, timeoutMs, context) => {
+  let timer;
+  try {
+    return await Promise.race([
+      promise,
+      new Promise((_resolve, reject) => {
+        timer = setTimeout(() => reject(new Error(`Timed out waiting for ${context}`)), timeoutMs);
+      })
+    ]);
+  } finally {
+    if (timer) {
+      clearTimeout(timer);
+    }
+  }
+};
+var decodeGraphqlWsMessage = (value) => {
+  if (typeof value === "string") {
+    const parsed = JSON.parse(value);
+    if (parsed && typeof parsed === "object" && !Array.isArray(parsed)) {
+      return parsed;
+    }
+    throw new Error("Expected GraphQL WebSocket JSON object message");
+  }
+  if (value && typeof value === "object" && !Array.isArray(value)) {
+    return value;
+  }
+  throw new Error("Expected GraphQL WebSocket message object");
+};
 var MockWebSocketConnection = class {
   handler;
   queue = [];
@@ -1616,6 +1646,107 @@ var TestClient = class {
       bodyText: response.text()
     };
   }
+  /**
+   * Send a GraphQL subscription over WebSocket and return the first event payload.
+   */
+  async graphqlSubscription(query, variables, operationName, path2 = "/graphql") {
+    const operationId = "spikard-subscription-1";
+    const subscriptionPayload = { query };
+    if (variables !== null && variables !== void 0) {
+      subscriptionPayload.variables = variables;
+    }
+    if (operationName !== null && operationName !== void 0) {
+      subscriptionPayload.operationName = operationName;
+    }
+    const ws = await this.websocketConnect(path2);
+    try {
+      await ws.sendJson({ type: "connection_init" });
+      let acknowledged = false;
+      for (let i = 0; i < GRAPHQL_WS_MAX_CONTROL_MESSAGES; i++) {
+        const message = decodeGraphqlWsMessage(
+          await withTimeout(ws.receiveJson(), GRAPHQL_WS_TIMEOUT_MS, "GraphQL connection_ack")
+        );
+        const messageType = typeof message.type === "string" ? message.type : "";
+        if (messageType === "connection_ack") {
+          acknowledged = true;
+          break;
+        }
+        if (messageType === "ping") {
+          const pong = { type: "pong" };
+          if ("payload" in message) {
+            pong.payload = message.payload;
+          }
+          await ws.sendJson(pong);
+          continue;
+        }
+        if (messageType === "connection_error" || messageType === "error") {
+          throw new Error(`GraphQL subscription rejected during init: ${JSON.stringify(message)}`);
+        }
+      }
+      if (!acknowledged) {
+        throw new Error("No GraphQL connection_ack received");
+      }
+      await ws.sendJson({
+        id: operationId,
+        type: "subscribe",
+        payload: subscriptionPayload
+      });
+      let event = null;
+      const errors = [];
+      let completeReceived = false;
+      for (let i = 0; i < GRAPHQL_WS_MAX_CONTROL_MESSAGES; i++) {
+        const message = decodeGraphqlWsMessage(
+          await withTimeout(ws.receiveJson(), GRAPHQL_WS_TIMEOUT_MS, "GraphQL subscription message")
+        );
+        const messageType = typeof message.type === "string" ? message.type : "";
+        const messageId = typeof message.id === "string" ? message.id : void 0;
+        const idMatches = messageId === void 0 || messageId === operationId;
+        if (messageType === "next" && idMatches) {
+          event = "payload" in message ? message.payload : null;
+          await ws.sendJson({ id: operationId, type: "complete" });
+          try {
+            const maybeComplete = decodeGraphqlWsMessage(
+              await withTimeout(ws.receiveJson(), GRAPHQL_WS_TIMEOUT_MS, "GraphQL complete message")
+            );
+            const completeType = typeof maybeComplete.type === "string" ? maybeComplete.type : "";
+            const completeId = typeof maybeComplete.id === "string" ? maybeComplete.id : void 0;
+            if (completeType === "complete" && (completeId === void 0 || completeId === operationId)) {
+              completeReceived = true;
+            }
+          } catch {
+          }
+          break;
+        }
+        if (messageType === "error") {
+          errors.push("payload" in message ? message.payload : message);
+          break;
+        }
+        if (messageType === "complete" && idMatches) {
+          completeReceived = true;
+          break;
+        }
+        if (messageType === "ping") {
+          const pong = { type: "pong" };
+          if ("payload" in message) {
+            pong.payload = message.payload;
+          }
+          await ws.sendJson(pong);
+        }
+      }
+      if (event === null && errors.length === 0 && !completeReceived) {
+        throw new Error("No GraphQL subscription event received before timeout");
+      }
+      return {
+        operationId,
+        acknowledged,
+        event,
+        errors,
+        completeReceived
+      };
+    } finally {
+      await ws.close();
+    }
+  }
   /**
    * Cleanup resources when test client is done
    */