From 1e7f9e8746d764b3fa1edf01a4efa7323a71983a Mon Sep 17 00:00:00 2001
From: Vincent Koc <vincentkoc@ieee.org>
Date: Sat, 4 Apr 2026 23:13:57 +0900
Subject: [PATCH] test(providers): cover transport family matrix

---
 src/agents/provider-transport-stream.test.ts | 150 +++++++++++++++++++
 1 file changed, 150 insertions(+)
 create mode 100644 src/agents/provider-transport-stream.test.ts

diff --git a/src/agents/provider-transport-stream.test.ts b/src/agents/provider-transport-stream.test.ts
new file mode 100644
index 00000000000..130e484fb78
--- /dev/null
+++ b/src/agents/provider-transport-stream.test.ts
@@ -0,0 +1,150 @@
+import type { Api, Model } from "@mariozechner/pi-ai";
+import { describe, expect, it } from "vitest";
+import { attachModelProviderRequestTransport } from "./provider-request-config.js";
+import {
+  buildTransportAwareSimpleStreamFn,
+  createBoundaryAwareStreamFnForModel,
+  createTransportAwareStreamFnForModel,
+  isTransportAwareApiSupported,
+  prepareTransportAwareSimpleModel,
+  resolveTransportAwareSimpleApi,
+} from "./provider-transport-stream.js";
+
+function buildModel<TApi extends Api>(
+  api: TApi,
+  params: {
+    id: string;
+    provider: string;
+    baseUrl: string;
+  },
+): Model<TApi> {
+  return {
+    id: params.id,
+    name: params.id,
+    api,
+    provider: params.provider,
+    baseUrl: params.baseUrl,
+    reasoning: true,
+    input: ["text"],
+    cost: { input: 0, output: 0, cacheRead: 0, cacheWrite: 0 },
+    contextWindow: 200_000,
+    maxTokens: 8_192,
+  };
+}
+
+describe("provider transport stream contracts", () => {
+  it("covers the supported transport api alias matrix", () => {
+    const cases = [
+      {
+        api: "openai-responses" as const,
+        provider: "openai",
+        id: "gpt-5.4",
+        baseUrl: "https://api.openai.com/v1",
+        alias: "openclaw-openai-responses-transport",
+      },
+      {
+        api: "openai-codex-responses" as const,
+        provider: "openai-codex",
+        id: "codex-mini-latest",
+        baseUrl: "https://chatgpt.com/backend-api",
+        alias: "openclaw-openai-responses-transport",
+      },
+      {
+        api: "openai-completions" as const,
+        provider: "xai",
+        id: "grok-4",
+        baseUrl: "https://api.x.ai/v1",
+        alias: "openclaw-openai-completions-transport",
+      },
+      {
+        api: "azure-openai-responses" as const,
+        provider: "azure-openai-responses",
+        id: "gpt-5.4",
+        baseUrl: "https://example.openai.azure.com/openai/v1",
+        alias: "openclaw-azure-openai-responses-transport",
+      },
+      {
+        api: "anthropic-messages" as const,
+        provider: "anthropic",
+        id: "claude-sonnet-4.6",
+        baseUrl: "https://api.anthropic.com",
+        alias: "openclaw-anthropic-messages-transport",
+      },
+      {
+        api: "google-generative-ai" as const,
+        provider: "google",
+        id: "gemini-3.1-pro-preview",
+        baseUrl: "https://generativelanguage.googleapis.com/v1beta",
+        alias: "openclaw-google-generative-ai-transport",
+      },
+    ];
+
+    for (const testCase of cases) {
+      const model = attachModelProviderRequestTransport(
+        buildModel(testCase.api, {
+          id: testCase.id,
+          provider: testCase.provider,
+          baseUrl: testCase.baseUrl,
+        }),
+        {
+          proxy: {
+            mode: "explicit-proxy",
+            url: "http://proxy.internal:8443",
+          },
+        },
+      );
+
+      expect(isTransportAwareApiSupported(testCase.api)).toBe(true);
+      expect(resolveTransportAwareSimpleApi(testCase.api)).toBe(testCase.alias);
+      expect(createBoundaryAwareStreamFnForModel(model)).toBeTypeOf("function");
+      expect(createTransportAwareStreamFnForModel(model)).toBeTypeOf("function");
+      expect(buildTransportAwareSimpleStreamFn(model)).toBeTypeOf("function");
+      expect(prepareTransportAwareSimpleModel(model)).toMatchObject({
+        api: testCase.alias,
+        provider: testCase.provider,
+        id: testCase.id,
+      });
+    }
+  });
+
+  it("fails closed when unsupported apis carry transport overrides", () => {
+    const model = attachModelProviderRequestTransport(
+      buildModel("ollama", {
+        id: "qwen3:32b",
+        provider: "ollama",
+        baseUrl: "http://localhost:11434",
+      }),
+      {
+        proxy: {
+          mode: "explicit-proxy",
+          url: "http://proxy.internal:8443",
+        },
+      },
+    );
+
+    expect(isTransportAwareApiSupported(model.api)).toBe(false);
+    expect(resolveTransportAwareSimpleApi(model.api)).toBeUndefined();
+    expect(createBoundaryAwareStreamFnForModel(model)).toBeUndefined();
+    expect(() => createTransportAwareStreamFnForModel(model)).toThrow(
+      'Model-provider request.proxy/request.tls is not yet supported for api "ollama"',
+    );
+    expect(() => buildTransportAwareSimpleStreamFn(model)).toThrow(
+      'Model-provider request.proxy/request.tls is not yet supported for api "ollama"',
+    );
+    expect(() => prepareTransportAwareSimpleModel(model)).toThrow(
+      'Model-provider request.proxy/request.tls is not yet supported for api "ollama"',
+    );
+  });
+
+  it("keeps unsupported apis unchanged when no transport overrides are attached", () => {
+    const model = buildModel("ollama", {
+      id: "qwen3:32b",
+      provider: "ollama",
+      baseUrl: "http://localhost:11434",
+    });
+
+    expect(createTransportAwareStreamFnForModel(model)).toBeUndefined();
+    expect(buildTransportAwareSimpleStreamFn(model)).toBeUndefined();
+    expect(prepareTransportAwareSimpleModel(model)).toBe(model);
+  });
+});