import pytest

from webdriver.bidi.modules.script import ContextTarget
from . import reload_and_assert
from .. import navigate_and_assert

pytestmark = pytest.mark.asyncio


@pytest.mark.parametrize("domain", ["", "alt"], ids=["same_origin", "cross_origin"])
async def test_origin(bidi_session, new_tab, inline, domain):
    frame_url = inline("frame")
    parent_url = inline(f"<iframe src='{frame_url}'></iframe>", domain=domain)

    # Navigate and assert (top-level).
    result = await bidi_session.browsing_context.navigate(
        context=new_tab["context"], url=parent_url, wait="complete"
    )
    assert result["url"] == parent_url

    contexts = await bidi_session.browsing_context.get_tree(root=new_tab["context"])
    assert len(contexts) == 1
    assert contexts[0]["url"] == parent_url

    assert len(contexts[0]["children"]) == 1
    frame = contexts[0]["children"][0]
    assert frame["url"] == frame_url

    # Reload and assert (frame).
    await reload_and_assert(
        bidi_session, frame, last_navigation=result["navigation"], url=frame_url
    )


@pytest.mark.parametrize("domain", ["", "alt"], ids=["same_origin", "cross_origin"])
async def test_reload_resets_iframe_location(bidi_session, new_tab, inline, domain):
    url = inline("<iframe src='about:blank'></iframe>", domain=domain)
    contexts = await navigate_and_assert(bidi_session, new_tab, url)

    iframe = contexts[0]["children"][0]
    iframe_url = inline("frame")

    await bidi_session.browsing_context.navigate(
        context=iframe["context"], url=iframe_url, wait="complete"
    )

    await bidi_session.browsing_context.reload(
        context=new_tab["context"], wait="complete"
    )

    contexts = await bidi_session.browsing_context.get_tree(root=new_tab["context"])
    assert len(contexts) == 1
    frames = contexts[0]["children"]
    assert len(frames) == 1
    frame_context = frames[0]["context"]

    result = await bidi_session.script.evaluate(
        expression="document.location.href",
        target=ContextTarget(frame_context),
        await_promise=False,
    )
    assert result["value"] == "about:blank"