mirror of
https://forge.fsky.io/oneflux/omegafox.git
synced 2026-02-10 15:32:06 -08:00
145 lines
4.8 KiB
Text
145 lines
4.8 KiB
Text
# 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.
|
|
|
|
import asyncio
|
|
import os
|
|
from pathlib import Path
|
|
from typing import Dict, Optional
|
|
|
|
import pytest
|
|
|
|
from playwright.async_api import BrowserType, Error
|
|
from tests.utils import TARGET_CLOSED_ERROR_MESSAGE
|
|
|
|
|
|
async def test_browser_type_launch_should_reject_all_promises_when_browser_is_closed(
|
|
browser_type: BrowserType, launch_arguments: Dict
|
|
) -> None:
|
|
browser = await browser_type.launch(**launch_arguments)
|
|
page = await (await browser.new_context()).new_page()
|
|
never_resolves = asyncio.create_task(page.evaluate("() => new Promise(r => {})"))
|
|
await page.close()
|
|
with pytest.raises(Error) as exc:
|
|
await never_resolves
|
|
assert TARGET_CLOSED_ERROR_MESSAGE in exc.value.message
|
|
|
|
|
|
@pytest.mark.skip_browser("firefox")
|
|
async def test_browser_type_launch_should_throw_if_page_argument_is_passed(
|
|
browser_type: BrowserType, launch_arguments: Dict
|
|
) -> None:
|
|
with pytest.raises(Error) as exc:
|
|
await browser_type.launch(**launch_arguments, args=["http://example.com"])
|
|
assert "can not specify page" in exc.value.message
|
|
|
|
|
|
async def test_browser_type_launch_should_reject_if_launched_browser_fails_immediately(
|
|
browser_type: BrowserType, launch_arguments: Dict, assetdir: Path
|
|
) -> None:
|
|
with pytest.raises(Error):
|
|
await browser_type.launch(
|
|
**launch_arguments,
|
|
executable_path=assetdir / "dummy_bad_browser_executable.js",
|
|
)
|
|
|
|
|
|
async def test_browser_type_launch_should_reject_if_executable_path_is_invalid(
|
|
browser_type: BrowserType, launch_arguments: Dict
|
|
) -> None:
|
|
with pytest.raises(Error) as exc:
|
|
await browser_type.launch(
|
|
**launch_arguments, executable_path="random-invalid-path"
|
|
)
|
|
assert "executable doesn't exist" in exc.value.message
|
|
|
|
|
|
async def test_browser_type_executable_path_should_work(
|
|
browser_type: BrowserType, browser_channel: str
|
|
) -> None:
|
|
if browser_channel:
|
|
return
|
|
executable_path = browser_type.executable_path
|
|
assert os.path.exists(executable_path)
|
|
assert os.path.realpath(executable_path) == os.path.realpath(executable_path)
|
|
|
|
|
|
async def test_browser_type_name_should_work(
|
|
browser_type: BrowserType, is_webkit: bool, is_firefox: bool, is_chromium: bool
|
|
) -> None:
|
|
if is_webkit:
|
|
assert browser_type.name == "webkit"
|
|
elif is_firefox:
|
|
assert browser_type.name == "firefox"
|
|
elif is_chromium:
|
|
assert browser_type.name == "chromium"
|
|
else:
|
|
raise ValueError("Unknown browser")
|
|
|
|
|
|
async def test_browser_close_should_fire_close_event_for_all_contexts(
|
|
browser_type: BrowserType, launch_arguments: Dict
|
|
) -> None:
|
|
browser = await browser_type.launch(**launch_arguments)
|
|
context = await browser.new_context()
|
|
closed = []
|
|
context.on("close", lambda _: closed.append(True))
|
|
await browser.close()
|
|
assert closed == [True]
|
|
|
|
|
|
async def test_browser_close_should_be_callable_twice(
|
|
browser_type: BrowserType, launch_arguments: Dict
|
|
) -> None:
|
|
browser = await browser_type.launch(**launch_arguments)
|
|
await asyncio.gather(
|
|
browser.close(),
|
|
browser.close(),
|
|
)
|
|
await browser.close()
|
|
|
|
|
|
@pytest.mark.only_browser("chromium")
|
|
async def test_browser_launch_should_return_background_pages(
|
|
browser_type: BrowserType,
|
|
tmpdir: Path,
|
|
browser_channel: Optional[str],
|
|
assetdir: Path,
|
|
launch_arguments: Dict,
|
|
) -> None:
|
|
if browser_channel:
|
|
pytest.skip()
|
|
|
|
extension_path = str(assetdir / "simple-extension")
|
|
context = await browser_type.launch_persistent_context(
|
|
str(tmpdir),
|
|
**{
|
|
**launch_arguments,
|
|
"headless": False,
|
|
"args": [
|
|
f"--disable-extensions-except={extension_path}",
|
|
f"--load-extension={extension_path}",
|
|
],
|
|
},
|
|
)
|
|
background_page = None
|
|
if len(context.background_pages):
|
|
background_page = context.background_pages[0]
|
|
else:
|
|
background_page = await context.wait_for_event("backgroundpage")
|
|
assert background_page
|
|
assert background_page in context.background_pages
|
|
assert background_page not in context.pages
|
|
await context.close()
|
|
assert len(context.background_pages) == 0
|
|
assert len(context.pages) == 0
|