playwright-python/tests/async/test_browser.py

82 lines
2.5 KiB
Python

# 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 re
from typing import Dict
import pytest
from playwright.async_api import Browser, BrowserType, Error
async def test_should_create_new_page(browser: Browser) -> None:
page1 = await browser.new_page()
assert len(browser.contexts) == 1
page2 = await browser.new_page()
assert len(browser.contexts) == 2
await page1.close()
assert len(browser.contexts) == 1
await page2.close()
assert len(browser.contexts) == 0
async def test_should_throw_upon_second_create_new_page(browser: Browser) -> None:
page = await browser.new_page()
with pytest.raises(Error) as exc:
await page.context.new_page()
await page.close()
assert "Please use browser.new_context()" in exc.value.message
async def test_version_should_work(browser: Browser, is_chromium: bool) -> None:
version = browser.version
if is_chromium:
assert re.match(r"^\d+\.\d+\.\d+\.\d+$", version)
else:
assert re.match(r"^\d+\.\d+", version)
async def test_should_return_browser_type(
browser: Browser, browser_type: BrowserType
) -> None:
assert browser.browser_type is browser_type
async def test_bind_should_return_endpoint_and_allow_unbind(
browser_type: BrowserType, launch_arguments: Dict
) -> None:
browser = await browser_type.launch(**launch_arguments)
try:
result = await browser.bind("test-server")
assert "endpoint" in result
assert isinstance(result["endpoint"], str)
assert len(result["endpoint"]) > 0
await browser.unbind()
finally:
await browser.close()
async def test_should_fire_context_event_on_new_context(browser: Browser) -> None:
# Ported from upstream tests/library/browser.spec.ts.
events = []
browser.on("context", lambda ctx: events.append(ctx))
context = await browser.new_context()
try:
assert events == [context]
finally:
await context.close()