chore: use offical pw methods instead of dunder defs (microsoft#11)

osullivan-daniel · Oct 3, 2020 · 20a6e7c · 20a6e7c
1 parent 55380b3
commit 20a6e7c
Showing 1 changed file with 2 additions and 3 deletions.
diff --git a/pytest_playwright/pytest_playwright.py b/pytest_playwright/pytest_playwright.py
@@ -94,14 +94,13 @@ def launch(**kwargs: Dict[Any, Any]) -> Browser:
         launch_options = {**browser_type_launch_args, **kwargs}
         if headful_option:
             launch_options["headless"] = False
-        pw_context = sync_playwright()
-        pw = pw_context.__enter__()
+        pw = sync_playwright().start()
         browser = getattr(pw, browser_name).launch(**launch_options)
         browser._close = browser.close
 
         def _handle_close() -> None:
             browser._close()
-            pw_context.__exit__(None, None, None)
+            pw.stop()
 
         browser.close = _handle_close
         return browser