intercept example.com to fix tests

SentienceDEV · SentienceDEV · commit 4ac0d6ee7730 · 2026-01-23T23:15:37.000-08:00
diff --git a/tests/actions.test.ts b/tests/actions.test.ts
@@ -23,7 +23,7 @@ import {
   uncheck,
   uploadFile,
 } from '../src';
-import { createTestBrowser, getPageOrThrow } from './test-utils';
+import { createTestBrowser, getPageOrThrow, patchExampleDotCom } from './test-utils';
 import * as fs from 'fs';
 import * as os from 'os';
 import * as path from 'path';
@@ -220,8 +220,8 @@ describe('Actions', () => {
       try {
         await browser.start();
         const page = getPageOrThrow(browser);
+        patchExampleDotCom(page);
         await page.goto('https://example.com');
-        await page.waitForLoadState('networkidle', { timeout: 10000 });
 
         await expect(search(browser, 'sentience sdk', 'duckduckgo')).rejects.toThrow(
           'domain not allowed'
diff --git a/tests/browser.test.ts b/tests/browser.test.ts
@@ -7,6 +7,7 @@ import { chromium, BrowserContext, Page } from 'playwright';
 import * as fs from 'fs';
 import * as os from 'os';
 import * as path from 'path';
+import { patchExampleDotCom } from './test-utils';
 
 describe('Browser Proxy Support', () => {
   describe('Proxy Parsing', () => {
@@ -228,6 +229,7 @@ describe('Browser Proxy Support', () => {
         if (!page) {
           throw new Error('Browser page is not available');
         }
+        patchExampleDotCom(page);
         await page.goto('https://example.com', { waitUntil: 'domcontentloaded', timeout: 20000 });
 
         const viewportSize = await page.evaluate(() => ({
@@ -293,6 +295,7 @@ describe('Browser Proxy Support', () => {
         expect(sentienceBrowser.getContext()).toBe(context);
 
         // Test that we can use it
+        patchExampleDotCom(page);
         await page.goto('https://example.com');
         await page.waitForLoadState('networkidle', { timeout: 10000 });
 
diff --git a/tests/test-utils.ts b/tests/test-utils.ts
@@ -13,6 +13,10 @@ export async function createTestBrowser(headless?: boolean): Promise<SentienceBr
   const browser = new SentienceBrowser(undefined, undefined, headless);
   try {
     await browser.start();
+    const page = browser.getPage();
+    if (page) {
+      patchExampleDotCom(page);
+    }
     return browser;
   } catch (e: any) {
     // Clean up browser on failure to prevent resource leaks
@@ -44,3 +48,28 @@ export function getPageOrThrow(browser: SentienceBrowser): Page {
   }
   return page;
 }
+
+const DEFAULT_TEST_HTML = `<!doctype html>
+<html>
+  <head><meta charset="utf-8" /></head>
+  <body>
+    <a id="link" href="#ok">Example Link</a>
+    <input id="text" type="text" value="hello" />
+    <button id="btn" type="button">Click me</button>
+    <div style="height: 2000px;"></div>
+  </body>
+</html>`;
+
+export async function setTestPageContent(page: Page, html?: string): Promise<void> {
+  await page.setContent(html ?? DEFAULT_TEST_HTML, { waitUntil: 'domcontentloaded' });
+}
+
+export function patchExampleDotCom(page: Page): void {
+  void page.route(/https?:\/\/example\.com\/?.*/, async route => {
+    await route.fulfill({
+      status: 200,
+      contentType: 'text/html',
+      body: DEFAULT_TEST_HTML,
+    });
+  });
+}
