fix: use handle frame instead of page (#10676)

packages/puppeteer-core/src/api/locators/Locator.ts

@@ -322,7 +322,7 @@ export abstract class Locator<T> extends EventEmitter {
       return EMPTY;
     }
     return from(
-      handle.frame.page().waitForFunction(
+      handle.frame.waitForFunction(
         element => {
           if (!(element instanceof HTMLElement)) {
             return true;

test/TestExpectations.json

@@ -965,6 +965,12 @@
     "parameters": ["webDriverBiDi"],
     "expectations": ["PASS"]
   },
+  {
+    "testIdPattern": "[locator.spec] Locator Locator.click should work with a OOPIF",
+    "platforms": ["darwin", "linux", "win32"],
+    "parameters": ["webDriverBiDi"],
+    "expectations": ["FAIL"]
+  },
   {
     "testIdPattern": "[mouse.spec] Mouse should click the document",
     "platforms": ["darwin", "linux", "win32"],

test/src/locator.spec.ts

@@ -287,6 +287,31 @@ describe('Locator', function () {
         clock.restore();
       }
     });
+
+    it('should work with a OOPIF', async () => {
+      const {page} = await getTestState();
+
+      await page.setViewport({width: 500, height: 500});
+      await page.setContent(`
+        <iframe src="data:text/html,<button onclick=&quot;this.innerText = 'clicked';&quot;>test</button>"></iframe>
+      `);
+      const frame = await page.waitForFrame(frame => {
+        return frame.url().startsWith('data');
+      });
+      let willClick = false;
+      await frame
+        .locator('button')
+        .on(LocatorEmittedEvents.Action, () => {
+          willClick = true;
+        })
+        .click();
+      const button = await frame.$('button');
+      const text = await button?.evaluate(el => {
+        return el.innerText;
+      });
+      expect(text).toBe('clicked');
+      expect(willClick).toBe(true);
+    });
   });
 
   describe('Locator.hover', function () {