Source code
Revision control
Copy as Markdown
Other Tools
import pytest
from webdriver.bidi.modules.script import ContextTarget
pytestmark = pytest.mark.asyncio
@pytest.mark.parametrize("match_type", ["alwaysMatch", "firstMatch"])
async def test_platform_name(
new_session, match_capabilities, target_platform, match_type
):
capabilities = match_capabilities(match_type, "platformName", target_platform)
bidi_session = await new_session(capabilities=capabilities)
assert bidi_session.capabilities["platformName"] == target_platform
@pytest.mark.parametrize("match_type", ["alwaysMatch", "firstMatch"])
async def test_proxy(
new_session, match_capabilities, server_config, inline, match_type
):
domain = server_config["domains"][""][""]
port = server_config["ports"]["http"][0]
proxy_url = f"{domain}:{port}"
proxy_capability = {"proxyType": "manual", "httpProxy": proxy_url}
capabilities = match_capabilities(match_type, "proxy", proxy_capability)
bidi_session = await new_session(capabilities=capabilities)
assert bidi_session.capabilities["proxy"] == proxy_capability
page_content = "proxy"
page_url = inline(f"<div>{page_content}</div>")
test_url = page_url.replace(proxy_url, "example.com")
contexts = await bidi_session.browsing_context.get_tree()
await bidi_session.browsing_context.navigate(
context=contexts[0]["context"], url=test_url, wait="complete"
)
# Check that content is expected
response = await bidi_session.script.evaluate(
expression="""document.querySelector('div').textContent""",
target=ContextTarget(contexts[0]["context"]),
await_promise=False,
)
assert response == {"type": "string", "value": page_content}
@pytest.mark.parametrize(
"value",
[
"accept",
"accept and notify",
"dismiss",
"dismiss and notify",
"ignore",
],
)
async def test_unhandledPromptBehavior_string(new_session, match_capabilities, value):
capabilities = match_capabilities("alwaysMatch", "unhandledPromptBehavior", value)
bidi_session = await new_session(capabilities=capabilities)
assert bidi_session.capabilities.get("unhandledPromptBehavior") == value
async def test_unhandledPromptBehavior_without_value(new_session, match_capabilities):
capabilities = match_capabilities("alwaysMatch", "unhandledPromptBehavior", None)
bidi_session = await new_session(capabilities=capabilities)
assert "unhandledPromptBehavior" not in bidi_session.capabilities
@pytest.mark.parametrize("handler", ["accept", "dismiss", "ignore"])
@pytest.mark.parametrize("prompt_type", ["alert", "beforeUnload", "confirm", "prompt"])
async def test_unhandledPromptBehavior_object(
new_session, match_capabilities, handler, prompt_type
):
capabilities = match_capabilities(
"alwaysMatch", "unhandledPromptBehavior", {prompt_type: handler}
)
bidi_session = await new_session(capabilities=capabilities)
assert bidi_session.capabilities.get("unhandledPromptBehavior") == {
prompt_type: handler
}
@pytest.mark.parametrize("match_type", ["alwaysMatch", "firstMatch"])
async def test_websocket_url(new_session, match_capabilities, match_type):
capabilities = match_capabilities(match_type, "webSocketUrl", True)
bidi_session = await new_session(capabilities=capabilities)
assert bidi_session.capabilities.get("webSocketUrl") is None