add test

Author: AlinaVarkki

src/tools/snapshot.ts

@@ -36,11 +36,14 @@ export const waitFor = defineTool({
   },
   handler: async (request, response, context) => {
     const page = context.getSelectedPage();
+    const frames = page.frames();
 
-    await Locator.race([
-      page.locator(`aria/${request.params.text}`),
-      page.locator(`text/${request.params.text}`),
-    ]).wait();
+    const locators = frames.flatMap(frame => [
+      frame.locator(`aria/${request.params.text}`),
+      frame.locator(`text/${request.params.text}`),
+    ]);
+
+    await Locator.race(locators).wait();
 
     response.appendResponseLine(
       `Element with text "${request.params.text}" found.`,

tests/tools/snapshot.test.ts

@@ -95,5 +95,33 @@ describe('snapshot', () => {
         assert.ok(response.includeSnapshot);
       });
     });
+
+
+    it('should work with iframe content', async () => {
+      await withBrowser(async (response, context) => {
+        const page = await context.getSelectedPage();
+
+        await page.setContent(
+          html`<h1>Top level</h1>
+            <iframe srcdoc="<p>Hello iframe</p>"></iframe>`,
+        );
+
+        await waitFor.handler(
+          {
+            params: {
+              text: 'Hello iframe',
+            },
+          },
+          response,
+          context,
+        );
+
+        assert.equal(
+          response.responseLines[0],
+          'Element with text "Hello iframe" found.',
+        );
+        assert.ok(response.includeSnapshot);
+      });
+    });
   });
 });