# Copyright 2021 Google LLC.
# 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 pytest
from anys import ANY_STR
from test_helpers import (
    ANY_TIMESTAMP,
    ANY_UUID,
    execute_command,
    get_tree,
    goto_url,
    read_JSON_message,
    send_JSON_command,
    subscribe,
)


@pytest.mark.asyncio
async def test_nestedBrowsingContext_navigateToPageWithHash_contextInfoUpdated(
    websocket, iframe_id, html
):
    url = html("<h2>test</h2>")
    url_with_hash_1 = url + "#1"

    # Initial navigation.
    await goto_url(websocket, iframe_id, url_with_hash_1, "complete")

    result = await get_tree(websocket, iframe_id)

    assert {
        "contexts": [
            {
                "context": iframe_id,
                "children": [],
                "parent": ANY_STR,
                "url": url_with_hash_1,
                "userContext": "default",
                "originalOpener": None,
                "clientWindow": ANY_STR,
            }
        ]
    } == result


@pytest.mark.asyncio
async def test_nestedBrowsingContext_navigateWaitNone_navigated(
    websocket, iframe_id, html, read_messages
):
    url = html("<h2>iframe</h2>")

    await subscribe(
        websocket, ["browsingContext.domContentLoaded", "browsingContext.load"]
    )

    result = await goto_url(websocket, iframe_id, url, "none")
    navigation_id = result["navigation"]

    assert result == {"navigation": navigation_id, "url": url}

    [dom_content_loaded, browsing_context_load] = await read_messages(2, sort=True)
    assert dom_content_loaded == {
        "type": "event",
        "method": "browsingContext.domContentLoaded",
        "params": {
            "context": iframe_id,
            "navigation": navigation_id,
            "timestamp": ANY_TIMESTAMP,
            "url": url,
        },
    }
    assert browsing_context_load == {
        "type": "event",
        "method": "browsingContext.load",
        "params": {
            "context": iframe_id,
            "navigation": navigation_id,
            "timestamp": ANY_TIMESTAMP,
            "url": url,
        },
    }


@pytest.mark.asyncio
async def test_nestedBrowsingContext_navigateWaitInteractive_navigated(
    websocket, iframe_id, html, assert_no_more_messages
):
    url = html("<h2>iframe</h2>")

    await subscribe(
        websocket, ["browsingContext.domContentLoaded", "browsingContext.load"]
    )

    result = await goto_url(websocket, iframe_id, url, "interactive")
    navigation_id = result["navigation"]

    assert result == {"navigation": navigation_id, "url": url}

    message = await read_JSON_message(websocket)
    assert message == {
        "method": "browsingContext.load",
        "params": {
            "context": iframe_id,
            "navigation": navigation_id,
            "timestamp": ANY_TIMESTAMP,
            "url": url,
        },
        "type": "event",
    }

    await assert_no_more_messages()


@pytest.mark.asyncio
async def test_nestedBrowsingContext_navigateWaitComplete_navigated(
    websocket, iframe_id, html, assert_no_more_messages
):
    url = html("<h2>iframe</h2>")

    await subscribe(
        websocket, ["browsingContext.domContentLoaded", "browsingContext.load"]
    )

    await send_JSON_command(
        websocket,
        {
            "method": "browsingContext.navigate",
            "params": {"url": url, "wait": "complete", "context": iframe_id},
        },
    )

    response = await read_JSON_message(websocket)
    navigation_id = response["params"]["navigation"]
    assert response == {
        "type": "event",
        "method": "browsingContext.domContentLoaded",
        "params": {
            "context": iframe_id,
            "navigation": navigation_id,
            "timestamp": ANY_TIMESTAMP,
            "url": url,
        },
    }

    assert await read_JSON_message(websocket) == {
        "type": "event",
        "method": "browsingContext.load",
        "params": {
            "context": iframe_id,
            "navigation": navigation_id,
            "timestamp": ANY_TIMESTAMP,
            "url": url,
        },
    }

    response = await read_JSON_message(websocket)
    assert response["result"] == {"navigation": navigation_id, "url": url}

    await assert_no_more_messages()


@pytest.mark.asyncio
async def test_nestedBrowsingContext_navigateSameDocumentNavigation_waitNone_navigated(
    websocket, iframe_id, html
):
    url = html("<h2>test</h2>")
    url_with_hash_1 = url + "#1"
    url_with_hash_2 = url + "#2"

    # Initial navigation.
    await goto_url(websocket, iframe_id, url, "complete")

    resp = await goto_url(websocket, iframe_id, url_with_hash_1, "none")
    assert resp == {"navigation": ANY_UUID, "url": url_with_hash_1}

    resp = await goto_url(websocket, iframe_id, url_with_hash_2, "none")
    assert resp == {"navigation": ANY_UUID, "url": url_with_hash_2}


@pytest.mark.asyncio
async def test_nestedBrowsingContext_navigateSameDocumentNavigation_waitInteractive_navigated(
    websocket, iframe_id, html
):
    url = html("<h2>test</h2>")
    url_with_hash_1 = url + "#1"
    url_with_hash_2 = url + "#2"

    # Initial navigation.
    await goto_url(websocket, iframe_id, url, "complete")

    resp = await goto_url(websocket, iframe_id, url_with_hash_1, "interactive")
    assert resp == {"navigation": ANY_UUID, "url": url_with_hash_1}

    result = await get_tree(websocket, iframe_id)

    assert {
        "contexts": [
            {
                "context": iframe_id,
                "children": [],
                "parent": ANY_STR,
                "url": url_with_hash_1,
                "userContext": "default",
                "originalOpener": None,
                "clientWindow": ANY_STR,
            }
        ]
    } == result

    resp = await goto_url(websocket, iframe_id, url_with_hash_2, "interactive")
    assert resp == {"navigation": ANY_UUID, "url": url_with_hash_2}

    result = await get_tree(websocket, iframe_id)

    assert {
        "contexts": [
            {
                "context": iframe_id,
                "children": [],
                "parent": ANY_STR,
                "url": url_with_hash_2,
                "userContext": "default",
                "originalOpener": None,
                "clientWindow": ANY_STR,
            }
        ]
    } == result


@pytest.mark.asyncio
async def test_nestedBrowsingContext_navigateSameDocumentNavigation_waitComplete_navigated(
    websocket, iframe_id, html
):
    url = html("<h2>test</h2>")
    url_with_hash_1 = url + "#1"
    url_with_hash_2 = url + "#2"

    # Initial navigation.
    await goto_url(websocket, iframe_id, url, "complete")

    resp = await goto_url(websocket, iframe_id, url_with_hash_1, "complete")
    assert resp == {"navigation": ANY_UUID, "url": url_with_hash_1}

    result = await get_tree(websocket, iframe_id)

    assert {
        "contexts": [
            {
                "context": iframe_id,
                "children": [],
                "parent": ANY_STR,
                "url": url_with_hash_1,
                "userContext": "default",
                "originalOpener": None,
                "clientWindow": ANY_STR,
            }
        ]
    } == result

    resp = await goto_url(websocket, iframe_id, url_with_hash_2, "complete")
    assert resp == {"navigation": ANY_UUID, "url": url_with_hash_2}

    result = await get_tree(websocket, iframe_id)

    assert {
        "contexts": [
            {
                "context": iframe_id,
                "children": [],
                "parent": ANY_STR,
                "url": url_with_hash_2,
                "userContext": "default",
                "originalOpener": None,
                "clientWindow": ANY_STR,
            }
        ]
    } == result


@pytest.mark.asyncio
async def test_nestedBrowsingContext_afterNavigation_getTreeWithNestedCrossOriginContexts_contextsReturned(
    websocket, iframe_id, html, iframe, url_example, url_example_another_origin
):
    page_with_nested_iframe = html(iframe(url_example))
    another_page_with_nested_iframe = html(iframe(url_example_another_origin))

    await goto_url(websocket, iframe_id, page_with_nested_iframe, "complete")
    await goto_url(websocket, iframe_id, another_page_with_nested_iframe, "complete")

    result = await get_tree(websocket, iframe_id)
    assert {
        "contexts": [
            {
                "context": iframe_id,
                "children": [
                    {
                        "context": ANY_STR,
                        "url": url_example_another_origin,
                        "children": [],
                        "userContext": "default",
                        "originalOpener": None,
                        "clientWindow": ANY_STR,
                    }
                ],
                "parent": ANY_STR,
                "url": another_page_with_nested_iframe,
                "userContext": "default",
                "originalOpener": None,
                "clientWindow": ANY_STR,
            }
        ]
    } == result


@pytest.mark.asyncio
async def test_nestedBrowsingContext_afterNavigation_getTree_contextsReturned(
    websocket, iframe_id, html, iframe, url_all_origins
):
    page_with_nested_iframe = html("<h1>MAIN_PAGE</h1>" + iframe(url_all_origins))
    another_page_with_nested_iframe = html(
        "<h1>ANOTHER_MAIN_PAGE</h1>" + iframe(url_all_origins)
    )

    await goto_url(websocket, iframe_id, page_with_nested_iframe, "complete")

    result = await get_tree(websocket, iframe_id)
    assert {
        "contexts": [
            {
                "context": iframe_id,
                "url": page_with_nested_iframe,
                "children": [
                    {
                        "context": ANY_STR,
                        "url": url_all_origins,
                        "children": [],
                        "userContext": "default",
                        "originalOpener": None,
                        "clientWindow": ANY_STR,
                    }
                ],
                "parent": ANY_STR,
                "userContext": "default",
                "originalOpener": None,
                "clientWindow": ANY_STR,
            }
        ]
    } == result

    await goto_url(websocket, iframe_id, another_page_with_nested_iframe, "complete")

    result = await get_tree(websocket, iframe_id)
    assert {
        "contexts": [
            {
                "context": iframe_id,
                "url": another_page_with_nested_iframe,
                "children": [
                    {
                        "context": ANY_STR,
                        "url": url_all_origins,
                        "children": [],
                        "userContext": "default",
                        "originalOpener": None,
                        "clientWindow": ANY_STR,
                    }
                ],
                "parent": ANY_STR,
                "userContext": "default",
                "originalOpener": None,
                "clientWindow": ANY_STR,
            }
        ]
    } == result


@pytest.mark.asyncio
async def test_browsingContext_addAndRemoveNestedContext_contextAddedAndRemoved(
    websocket, context_id, url_all_origins, html, iframe
):
    page_with_nested_iframe = html(iframe(url_all_origins))
    await goto_url(websocket, context_id, page_with_nested_iframe, "complete")

    result = await get_tree(websocket)

    assert {
        "contexts": [
            {
                "context": context_id,
                "children": [
                    {
                        "context": ANY_STR,
                        "url": url_all_origins,
                        "children": [],
                        "userContext": "default",
                        "originalOpener": None,
                        "clientWindow": ANY_STR,
                    }
                ],
                "parent": None,
                "url": page_with_nested_iframe,
                "userContext": "default",
                "originalOpener": None,
                "clientWindow": ANY_STR,
            }
        ]
    } == result

    # Remove nested iframe.
    await execute_command(
        websocket,
        {
            "method": "script.evaluate",
            "params": {
                "expression": "document.querySelector('iframe').remove()",
                "awaitPromise": True,
                "target": {
                    "context": context_id,
                },
            },
        },
    )

    result = await get_tree(websocket)

    assert {
        "contexts": [
            {
                "context": context_id,
                "children": [],
                "parent": None,
                "url": page_with_nested_iframe,
                "userContext": "default",
                "originalOpener": None,
                "clientWindow": ANY_STR,
            }
        ]
    } == result