poll

Author: Skn0tt

tests/async/test_extension.py

@@ -68,7 +68,8 @@ async def test_should_give_access_to_the_service_worker(
     )
     assert service_worker
     assert service_worker in context.service_workers
-    assert await service_worker.evaluate("globalThis.MAGIC") == 42
+    while not await service_worker.evaluate("globalThis.MAGIC") == 42:
+        await asyncio.sleep(0.1)
     await context.close()
     assert len(context.background_pages) == 0
 
@@ -91,7 +92,8 @@ async def test_should_give_access_to_the_service_worker_when_recording_video(
     )
     assert service_worker
     assert service_worker in context.service_workers
-    assert await service_worker.evaluate("globalThis.MAGIC") == 42
+    while not await service_worker.evaluate("globalThis.MAGIC") == 42:
+        await asyncio.sleep(0.1)
     await context.close()
     assert len(context.background_pages) == 0
 

tests/sync/test_extension.py

@@ -13,6 +13,7 @@
 # limitations under the License.
 
 from pathlib import Path
+from time import sleep
 from typing import Any, Callable, Dict, Generator, Optional
 
 import pytest
@@ -65,7 +66,8 @@ def test_should_give_access_to_the_service_worker(
     )
     assert service_worker
     assert service_worker in context.service_workers
-    assert service_worker.evaluate("globalThis.MAGIC") == 42
+    while not service_worker.evaluate("globalThis.MAGIC") == 42:
+        sleep(0.1)
     context.close()
     assert len(context.background_pages) == 0
 
@@ -88,6 +90,7 @@ def test_should_give_access_to_the_service_worker_when_recording_video(
     )
     assert service_worker
     assert service_worker in context.service_workers
-    assert service_worker.evaluate("globalThis.MAGIC") == 42
+    while not service_worker.evaluate("globalThis.MAGIC") == 42:
+        sleep(0.1)
     context.close()
     assert len(context.background_pages) == 0