adapt to async tests

Author: Skn0tt

tests/async/test_extension.py

@@ -0,0 +1,93 @@
+# Copyright (c) Microsoft Corporation.
+#
+# Licensed under the Apache License, Version 2.0 (the "License");
+# you may not use this file except in compliance with the License.
+# You may obtain a copy of the License at
+#
+# http://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing, software
+# distributed under the License is distributed on an "AS IS" BASIS,
+# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+# See the License for the specific language governing permissions and
+# limitations under the License.
+
+from pathlib import Path
+from typing import Any, Awaitable, Callable, Dict, Generator, Optional
+
+import pytest
+
+from playwright.async_api import BrowserContext, BrowserType
+
+
+@pytest.fixture()
+def launch_persistent_context(
+    browser_type: BrowserType,
+    browser_channel: Optional[str],
+    tmp_path: Path,
+    launch_arguments: Dict[str, Any],
+    is_headless_shell: bool,
+) -> Generator[Callable[[str], Awaitable[BrowserContext]], None, None]:
+    if browser_channel and browser_channel.startswith("chrome"):
+        pytest.skip(
+            "--load-extension is not supported in Chrome anymore. https://groups.google.com/a/chromium.org/g/chromium-extensions/c/1-g8EFx2BBY/m/S0ET5wPjCAAJ"
+        )
+    if is_headless_shell:
+        pytest.skip("Headless Shell has no support for extensions")
+
+    async def launch(extension_path: str, **kwargs: Any) -> BrowserContext:
+        context = await browser_type.launch_persistent_context(
+            str(tmp_path),
+            **launch_arguments,
+            **kwargs,
+            args=[
+                f"--disable-extensions-except={extension_path}",
+                f"--load-extension={extension_path}",
+            ],
+        )
+        return context
+
+    yield launch
+
+
+@pytest.mark.only_browser("chromium")
+async def test_should_give_access_to_the_service_worker(
+    launch_persistent_context: Any,
+    assetdir: Path,
+) -> None:
+    extension_path = str(assetdir / "extension-mv3-simple")
+    context: BrowserContext = await launch_persistent_context(extension_path)
+    service_workers = context.service_workers
+    service_worker = (
+        service_workers[0]
+        if len(service_workers)
+        else await context.wait_for_event("backgroundpage")
+    )
+    assert service_worker
+    assert service_worker in context.service_workers
+    assert await service_worker.evaluate("globalThis.MAGIC") == 42
+    await context.close()
+    assert len(context.background_pages) == 0
+
+
+@pytest.mark.only_browser("chromium")
+async def test_should_give_access_to_the_service_worker_when_recording_video(
+    launch_persistent_context: Any,
+    tmp_path: Path,
+    assetdir: Path,
+) -> None:
+    extension_path = str(assetdir / "extension-mv3-simple")
+    context: BrowserContext = await launch_persistent_context(
+        extension_path, record_video_dir=(tmp_path / "videos")
+    )
+    service_workers = context.service_workers
+    service_worker = (
+        service_workers[0]
+        if len(service_workers)
+        else await context.wait_for_event("backgroundpage")
+    )
+    assert service_worker
+    assert service_worker in context.service_workers
+    assert await service_worker.evaluate("globalThis.MAGIC") == 42
+    await context.close()
+    assert len(context.background_pages) == 0