Hot fixed (#121)

Co-authored-by: DESKTOP\atthaboon.s <atthaboon.s@qahive.com>
qahive · Jul 14, 2021 · 88a3a5b · 88a3a5b
1 parent 0d9dadb
commit 88a3a5b
Showing 1 changed file with 10 additions and 4 deletions.
diff --git a/PuppeteerLibrary/playwright/custom_elements/playwright_page.py b/PuppeteerLibrary/playwright/custom_elements/playwright_page.py
@@ -78,10 +78,16 @@ async def waitForSelector_with_selenium_locator(self, selenium_locator: str, tim
             options['state'] = 'hidden'
 
         selector_value = SelectorAbstraction.get_selector(selenium_locator)
-        return await self.get_page().wait_for_selector(
-            selector=selector_value, 
-            timeout=options['timeout'], 
-            state=options['state'])
+        if self.selected_iframe is not None:
+            return await self.selected_iframe.wait_for_selector(
+                selector=selector_value, 
+                timeout=options['timeout'], 
+                state=options['state'])
+        else:
+            return await self.get_page().wait_for_selector(
+                selector=selector_value, 
+                timeout=options['timeout'], 
+                state=options['state'])
 
     ############
     # Query