fix: query_selector strict mode

mxschmitt · mxschmitt · commit e2093235f98e · 2021-08-26T11:42:32.000+02:00
diff --git a/playwright/_impl/_element_handle.py b/playwright/_impl/_element_handle.py
@@ -150,6 +150,7 @@ async def select_option(
             dict(
                 timeout=timeout,
                 noWaitAfter=noWaitAfter,
+                force=force,
                 **convert_select_option_values(value, index, label, element)
             )
         )
diff --git a/playwright/_impl/_frame.py b/playwright/_impl/_frame.py
@@ -259,7 +259,7 @@ async def query_selector(
         self, selector: str, strict: bool = None
     ) -> Optional[ElementHandle]:
         return from_nullable_channel(
-            await self._channel.send("querySelector", dict(selector=selector))
+            await self._channel.send("querySelector", locals_to_params(locals()))
         )
 
     async def query_selector_all(self, selector: str) -> List[ElementHandle]:
@@ -321,7 +321,13 @@ async def dispatch_event(
     ) -> None:
         await self._channel.send(
             "dispatchEvent",
-            dict(selector=selector, type=type, eventInit=serialize_argument(eventInit)),
+            dict(
+                selector=selector,
+                type=type,
+                eventInit=serialize_argument(eventInit),
+                strict=strict,
+                timeout=timeout,
+            ),
         )
 
     async def eval_on_selector(
@@ -338,6 +344,7 @@ async def eval_on_selector(
                     selector=selector,
                     expression=expression,
                     arg=serialize_argument(arg),
+                    strict=strict,
                 ),
             )
         )
@@ -549,6 +556,8 @@ async def select_option(
                 selector=selector,
                 timeout=timeout,
                 noWaitAfter=noWaitAfter,
+                strict=strict,
+                force=force,
                 **convert_select_option_values(value, index, label, element),
             )
         )
diff --git a/tests/async/test_browsercontext.py b/tests/async/test_browsercontext.py
@@ -740,4 +740,6 @@ async def test_strict_selectors_on_context(browser: Browser, server: Server):
     )
     with pytest.raises(Error):
         await page.text_content("button")
+    with pytest.raises(Error):
+        await page.query_selector("button")
     await context.close()
diff --git a/tests/async/test_frames.py b/tests/async/test_frames.py
@@ -14,7 +14,10 @@
 
 import asyncio
 
-from playwright.async_api import Error
+import pytest
+
+from playwright.async_api import Error, Page
+from tests.server import Server
 
 
 async def test_evaluate_handle(page, server):
@@ -253,3 +256,17 @@ async def test_should_report_different_frame_instance_when_frame_re_attaches(
     frame2 = await frame2_info.value
     assert frame2.is_detached() is False
     assert frame1 != frame2
+
+
+async def test_strict_mode(page: Page, server: Server):
+    await page.goto(server.EMPTY_PAGE)
+    await page.set_content(
+        """
+        <button>Hello</button>
+        <button>Hello</button>
+    """
+    )
+    with pytest.raises(Error):
+        await page.text_content("button", strict=True)
+    with pytest.raises(Error):
+        await page.query_selector("button", strict=True)

Original file line number	Diff line number	Diff line change
`@@ -150,6 +150,7 @@ async def select_option(`
`150`	`150`	`dict(`
`151`	`151`	`timeout=timeout,`
`152`	`152`	`noWaitAfter=noWaitAfter,`
	`153`	`+ force=force,`
`153`	`154`	`**convert_select_option_values(value, index, label, element)`
`154`	`155`	`)`
`155`	`156`	`)`
Original file line number	Diff line number	Diff line change
`@@ -740,4 +740,6 @@ async def test_strict_selectors_on_context(browser: Browser, server: Server):`
`740`	`740`	`)`
`741`	`741`	`with pytest.raises(Error):`
`742`	`742`	`await page.text_content("button")`
	`743`	`+ with pytest.raises(Error):`
	`744`	`+ await page.query_selector("button")`
`743`	`745`	`await context.close()`