test(extension): exercise larger page batches and shared budgets

Author: waleedlatif1

apps/sim/lib/mothership/chat/process-contents.test.ts

@@ -714,17 +714,42 @@ describe('processContextsServer - browser and terminal selections', () => {
     }
   )
 
-  it('refuses more than three page snapshots before resolving context', async () => {
+  it('delivers ten short page snapshots without dropping their sources', async () => {
+    const snapshots = Array.from({ length: 10 }, (_, index) => ({
+      ...snapshot,
+      id: `11111111-1111-4111-8111-11111111111${index}`,
+      url: `https://example.com/page-${index}`,
+      text: `Source ${index}`,
+    }))
+    const result = await processContextsServer(
+      snapshots.map((page, index) => ({
+        kind: 'browser_page' as const,
+        label: `Page (${index + 1})`,
+        snapshot: page,
+      })),
+      'user-1'
+    )
+    expect(result).toHaveLength(10)
+    for (const [index, context] of result.entries()) {
+      const payload = context.content.split('\n').find((line) => line.startsWith('{'))
+      expect(payload && JSON.parse(payload)).toEqual(snapshots[index])
+    }
+  })
+
+  it.each([
+    { name: 'combined text', text: 'x'.repeat(37_501) },
+    { name: 'combined serialized bytes', text: '\u0000'.repeat(18_751) },
+  ])('refuses an oversized page batch before resolving context: $name', async ({ text }) => {
     await expect(
       processContextsServer(
         Array.from({ length: 4 }, (_, index) => ({
           kind: 'browser_page' as const,
           label: `Page ${index}`,
-          snapshot: { ...snapshot, id: `11111111-1111-4111-8111-11111111111${index}` },
+          snapshot: { ...snapshot, text, id: `11111111-1111-4111-8111-11111111111${index}` },
         })),
         'user-1'
       )
-    ).rejects.toThrow('At most 3 page snapshots')
+    ).rejects.toThrow('These pages contain too much text')
   })
 
   it('points every browser and terminal mention at its exact tab', async () => {

apps/sim/scripts/test-browser-extension-e2e.ts

@@ -891,7 +891,7 @@ try {
     }
   }
   await checkNavigation(
-    'Capture, preview, removal, and the three-page cap use the real composer',
+    'Capture, preview, and removal support ten pages in the real composer',
     async () => {
       await capture()
       await expect(attachments()).toHaveCount(1)
@@ -1007,13 +1007,16 @@ try {
         await focusBarrier.dispose()
         await focusObserver.dispose()
       }
-      await capture()
-      await capture()
-      await capture()
-      await expect(attachments()).toHaveCount(3)
-      await expect(frame().getByRole('alert').filter({ hasText: 'Remove a page' })).toBeVisible()
-      await screenshot('page-limit')
-      await attachments().first().getByRole('button', { name: 'Remove page context' }).click()
+      for (let count = 2; count <= 10; count++) {
+        await capture()
+        await expect(attachments()).toHaveCount(count)
+      }
+      await expect(frame().getByRole('alert').filter({ hasText: 'Remove a page' })).toHaveCount(0)
+      await screenshot('many-pages')
+      for (let count = 10; count > 2; count--) {
+        await attachments().first().getByRole('button', { name: 'Remove page context' }).click()
+        await expect(attachments()).toHaveCount(count - 1)
+      }
       await expect(attachments()).toHaveCount(2)
       const stored = await frame().evaluate(() => JSON.stringify(localStorage))
       assert(