@ai-sdk/provider-utils 2.2.0 → 2.2.2

package/test/dist/index.mjs

@@ -18569,6 +18569,10 @@ var TestServerCall = class {
     expect(this.request).toBeDefined();
     return JSON.parse(await this.request.text());
   }
+  getRequestCredentials() {
+    expect(this.request).toBeDefined();
+    return this.request.credentials;
+  }
   getRequestHeaders() {
     expect(this.request).toBeDefined();
     const requestHeaders = this.request.headers;
@@ -18768,10 +18772,11 @@ function createTestServer(routes) {
   const originalRoutes = structuredClone(routes);
   const mswServer = setupServer(
     ...Object.entries(routes).map(([url, handler]) => {
-      return http.all(url, ({ request, params }) => {
-        var _a4, _b2, _c2, _d, _e;
+      return http.all(url, ({ request }) => {
+        var _a4, _b2, _c2;
+        const callNumber = calls.length;
         calls.push(new TestServerCall2(request));
-        const response = handler.response;
+        const response = typeof handler.response === "function" ? handler.response({ callNumber }) : Array.isArray(handler.response) ? handler.response[callNumber] : handler.response;
         if (response === void 0) {
           return HttpResponse.json({ error: "Not Found" }, { status: 404 });
         }
@@ -18782,7 +18787,7 @@ function createTestServer(routes) {
               status: 200,
               headers: {
                 "Content-Type": "application/json",
-                ...(_a4 = handler.response) == null ? void 0 : _a4.headers
+                ...response.headers
               }
             });
           case "stream-chunks":
@@ -18800,9 +18805,9 @@ function createTestServer(routes) {
                 }
               }
             );
-          case "readable-stream": {
+          case "controlled-stream": {
             return new HttpResponse(
-              response.stream.pipeThrough(new TextEncoderStream()),
+              response.controller.stream.pipeThrough(new TextEncoderStream()),
               {
                 status: 200,
                 headers: {
@@ -18817,17 +18822,17 @@ function createTestServer(routes) {
           case "binary": {
             return HttpResponse.arrayBuffer(response.body, {
               status: 200,
-              headers: (_b2 = handler.response) == null ? void 0 : _b2.headers
+              headers: response.headers
             });
           }
           case "error":
-            return HttpResponse.text((_c2 = response.body) != null ? _c2 : "Error", {
-              status: (_d = response.status) != null ? _d : 500,
+            return HttpResponse.text((_a4 = response.body) != null ? _a4 : "Error", {
+              status: (_b2 = response.status) != null ? _b2 : 500,
               headers: response.headers
             });
           case "empty":
             return new HttpResponse(null, {
-              status: (_e = response.status) != null ? _e : 200
+              status: (_c2 = response.status) != null ? _c2 : 200
             });
           default: {
             const _exhaustiveCheck = handlerType;
@@ -18858,9 +18863,25 @@ function createTestServer(routes) {
     }
   };
 }
+var TestResponseController = class {
+  constructor() {
+    this.transformStream = new TransformStream();
+    this.writer = this.transformStream.writable.getWriter();
+  }
+  get stream() {
+    return this.transformStream.readable;
+  }
+  async write(chunk) {
+    await this.writer.write(chunk);
+  }
+  async close() {
+    await this.writer.close();
+  }
+};
 export {
   JsonTestServer,
   StreamingTestServer,
+  TestResponseController,
   convertArrayToAsyncIterable,
   convertArrayToReadableStream,
   convertAsyncIterableToArray,