@ai-sdk/provider-utils 1.0.2 → 1.0.4

package/test/dist/index.mjs

@@ -75,6 +75,13 @@ async function convertReadableStreamToArray(stream) {
   return result;
 }
 
+// src/test/convert-response-stream-to-array.ts
+async function convertResponseStreamToArray(response) {
+  return convertReadableStreamToArray(
+    response.body.pipeThrough(new TextDecoderStream())
+  );
+}
+
 // ../../node_modules/.pnpm/outvariant@1.4.2/node_modules/outvariant/lib/index.mjs
 var POSITIONALS_EXP = /(%?)(%([sdijo]))/g;
 function serializePositional(positional, flag) {
@@ -5200,6 +5207,10 @@ var JsonTestServer = class {
     expect(this.request).toBeDefined();
     return new URL(this.request.url).searchParams;
   }
+  async getRequestUrl() {
+    expect(this.request).toBeDefined();
+    return new URL(this.request.url).toString();
+  }
   setupTestEnvironment() {
     beforeAll(() => this.server.listen());
     beforeEach(() => {
@@ -5300,13 +5311,40 @@ function createServer({
   pushCall,
   pushController
 }) {
+  const responsesArray = Array.isArray(responses) ? responses : [responses];
+  const responsesByUrl = responsesArray.reduce((responsesByUrl2, response) => {
+    if (!responsesByUrl2[response.url]) {
+      responsesByUrl2[response.url] = [];
+    }
+    responsesByUrl2[response.url].push(response);
+    return responsesByUrl2;
+  }, {});
+  const streams = {};
+  responsesArray.filter(
+    (response) => response.type === "controlled-stream"
+  ).forEach((response) => {
+    var _a3, _b2;
+    let streamController;
+    const stream = new ReadableStream({
+      start(controller) {
+        streamController = controller;
+      }
+    });
+    pushController((_a3 = response.id) != null ? _a3 : "", () => streamController);
+    streams[(_b2 = response.id) != null ? _b2 : ""] = stream;
+  });
+  const urlInvocationCounts = Object.fromEntries(
+    Object.entries(responsesByUrl).map(([url]) => [url, 0])
+  );
   return setupServer(
-    ...(Array.isArray(responses) ? responses : [responses]).map((response) => {
-      var _a3;
-      switch (response.type) {
-        case "json-value": {
-          return http.post(response.url, ({ request: request2 }) => {
-            pushCall(new TestServerCall(request2));
+    ...Object.entries(responsesByUrl).map(([url, responses2]) => {
+      return http.post(url, ({ request: request2 }) => {
+        var _a3, _b2;
+        pushCall(new TestServerCall(request2));
+        const invocationCount = urlInvocationCounts[url]++;
+        const response = responses2[invocationCount > responses2.length ? responses2.length - 1 : invocationCount];
+        switch (response.type) {
+          case "json-value":
             return HttpResponse.json(response.content, {
               status: 200,
               headers: {
@@ -5314,11 +5352,7 @@ function createServer({
                 ...response.headers
               }
             });
-          });
-        }
-        case "stream-values": {
-          return http.post(response.url, ({ request: request2 }) => {
-            pushCall(new TestServerCall(request2));
+          case "stream-values":
             return new HttpResponse(
               convertArrayToReadableStream(response.content).pipeThrough(
                 new TextEncoderStream()
@@ -5333,20 +5367,9 @@ function createServer({
                 }
               }
             );
-          });
-        }
-        case "controlled-stream": {
-          let streamController;
-          const stream = new ReadableStream({
-            start(controller) {
-              streamController = controller;
-            }
-          });
-          pushController((_a3 = response.id) != null ? _a3 : "", () => streamController);
-          return http.post(response.url, ({ request: request2 }) => {
-            pushCall(new TestServerCall(request2));
+          case "controlled-stream": {
             return new HttpResponse(
-              stream.pipeThrough(new TextEncoderStream()),
+              streams[(_a3 = response.id) != null ? _a3 : ""].pipeThrough(new TextEncoderStream()),
               {
                 status: 200,
                 headers: {
@@ -5357,21 +5380,16 @@ function createServer({
                 }
               }
             );
-          });
-        }
-        case "error": {
-          return http.post(response.url, ({ request: request2 }) => {
-            var _a4;
-            pushCall(new TestServerCall(request2));
-            return HttpResponse.text((_a4 = response.content) != null ? _a4 : "Error", {
+          }
+          case "error":
+            return HttpResponse.text((_b2 = response.content) != null ? _b2 : "Error", {
               status: response.status,
               headers: {
                 ...response.headers
               }
             });
-          });
         }
-      }
+      });
     })
   );
 }
@@ -5446,6 +5464,7 @@ export {
   convertArrayToReadableStream,
   convertAsyncIterableToArray,
   convertReadableStreamToArray,
+  convertResponseStreamToArray,
   convertStreamToArray,
   describeWithTestServer,
   withTestServer