@rimori/playwright-testing 0.3.5-next.0 → 0.3.5-next.2

package/dist/core/RimoriTestEnvironment.d.ts

@@ -231,6 +231,7 @@ export declare class RimoriTestEnvironment {
         mockGetVoice: (values: Buffer, options?: MockOptions) => void;
         mockGetTextFromVoice: (text: string, options?: MockOptions) => void;
         mockGetObject: (value: Record<string, unknown>, options?: MockOptions) => void;
+        mockGetStreamedObject: (value: Record<string, unknown>, options?: MockOptions) => void;
     };
     /**
      * Helpers for tracking browser audio playback in tests.

package/dist/core/RimoriTestEnvironment.js

@@ -47,7 +47,13 @@ class RimoriTestEnvironment {
                 this.addSupabaseRoute('plugin_settings', response, { ...options, method: 'POST' });
             },
             mockGetUserInfo: (userInfo, options) => {
-                this.addSupabaseRoute('/user-info', { ...this.rimoriInfo.profile, ...userInfo }, { ...options, delay: 0 });
+                // Update the rimoriInfo.profile so that MessageChannelSimulator returns the correct user info
+                this.rimoriInfo.profile = { ...this.rimoriInfo.profile, ...userInfo };
+                // Also update the MessageChannelSimulator if it exists (setup() has been called)
+                if (this.messageChannelSimulator) {
+                    this.messageChannelSimulator.setUserInfo(this.rimoriInfo.profile);
+                }
+                this.addSupabaseRoute('/user-info', this.rimoriInfo.profile, { ...options, delay: 0 });
             },
             mockGetPluginInfo: (pluginInfo, options) => {
                 this.addSupabaseRoute('/plugin-info', pluginInfo, options);
@@ -218,6 +224,9 @@ class RimoriTestEnvironment {
             mockGetObject: (value, options) => {
                 this.addBackendRoute('/ai/llm-object', value, { ...options, method: 'POST' });
             },
+            mockGetStreamedObject: (value, options) => {
+                this.addBackendRoute('/ai/llm-object', value, { ...options, isStreaming: true });
+            },
         };
         /**
          * Helpers for tracking browser audio playback in tests.
@@ -821,10 +830,19 @@ class RimoriTestEnvironment {
         if (typeof matchingMock.value === 'function') {
             responseValue = await matchingMock.value(request);
         }
-        // Handle streaming responses (for mockGetSteamedText)
+        // Handle streaming responses (for mockGetSteamedText and mockGetStreamedObject)
         // Since Playwright requires complete body, we format as SSE without delays
-        if (matchingMock.isStreaming && typeof responseValue === 'string') {
-            const body = this.formatAsSSE(responseValue);
+        if (matchingMock.isStreaming) {
+            let body;
+            if (typeof responseValue === 'string') {
+                // Text streaming (mockGetSteamedText)
+                body = this.formatAsSSE(responseValue);
+            }
+            else {
+                // Object streaming (mockGetStreamedObject)
+                // Format as SSE with JSON payload, followed by [DONE] marker
+                body = `data: ${JSON.stringify(responseValue)}\n\ndata: [DONE]\n\n`;
+            }
             return await route.fulfill({
                 status: 200,
                 headers: { 'Content-Type': 'text/event-stream' },

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@rimori/playwright-testing",
-  "version": "0.3.5-next.0",
+  "version": "0.3.5-next.2",
   "description": "Playwright testing utilities for Rimori plugins and workers",
   "license": "Apache-2.0",
   "repository": {
@@ -23,11 +23,11 @@
   },
   "peerDependencies": {
     "@playwright/test": "^1.40.0",
-    "@rimori/client": "^2.5.8"
+    "@rimori/client": "2.5.10-next.0"
   },
   "devDependencies": {
     "@playwright/test": "^1.40.0",
-    "@rimori/client": "^2.5.8",
+    "@rimori/client": "2.5.10-next.0",
     "@types/node": "^20.12.7",
     "rimraf": "^5.0.7",
     "typescript": "^5.7.2"