Merge pull request #210 from plotly/andrew/restarter

Andrew/restarter

Author: ayjayt

tests/test_process.py

@@ -40,7 +40,8 @@ async def test_context(headless, sandbox, gpu):
                 )
             response = await browser.send_command(command="Target.getTargets")
             assert "result" in response and "targetInfos" in response["result"]  # noqa: PT018 combined assert
-            assert len(response["result"]["targetInfos"]) != 0
+            if len(response["result"]["targetInfos"]) == 0:
+                await browser.create_tab()
             assert isinstance(browser.get_tab(), choreo.Tab)
             assert len(browser.get_tab().sessions) == 1
         # let asyncio do some cleaning up if it wants, may prevent warnings
@@ -62,7 +63,8 @@ async def test_no_context(headless, sandbox, gpu):
         async with timeout(pytest.default_timeout):
             response = await browser.send_command(command="Target.getTargets")
             assert "result" in response and "targetInfos" in response["result"]  # noqa: PT018 combined assert
-            assert len(response["result"]["targetInfos"]) != 0
+            if len(response["result"]["targetInfos"]) == 0:
+                await browser.create_tab()
             assert isinstance(browser.get_tab(), choreo.Tab)
             assert len(browser.get_tab().sessions) == 1
     finally: