from copy import deepcopy

import pytest
from support.network import get_host, http_request, websocket_request

pytestmark = pytest.mark.asyncio


@pytest.mark.parametrize(
    "allow_hosts, hostname, port_type, status",
    [
        # Valid hosts
        (["localhost.localdomain", "localhost"], "localhost", "server_port", 200),
        (["localhost.localdomain", "localhost"], "127.0.0.1", "server_port", 200),
        # Invalid hosts
        (["localhost.localdomain"], "localhost", "server_port", 500),
        (["localhost"], "localhost", "wrong_port", 500),
        (["www.localhost"], "localhost", "server_port", 500),
    ],
)
def test_allow_hosts(geckodriver, allow_hosts, hostname, port_type, status):
    extra_args = ["--allow-hosts"] + allow_hosts

    driver = geckodriver(hostname=hostname, extra_args=extra_args)
    host = get_host(port_type, hostname, driver.port)
    response = http_request(driver.hostname, driver.port, host=host)

    assert response.status == status


@pytest.mark.parametrize(
    "allow_hosts, hostname, status",
    [
        (["mozilla.org", "testhost"], "testhost", 101),
        (["mozilla.org"], "testhost", 400),
    ],
    ids=["allowed", "not allowed"],
)
async def test_allow_hosts_passed_to_remote_agent(
    configuration, geckodriver, allow_hosts, hostname, status
):
    config = deepcopy(configuration)
    config["capabilities"]["webSocketUrl"] = True

    extra_args = ["--allow-hosts"] + allow_hosts

    driver = geckodriver(config=config, extra_args=extra_args)

    driver.new_session()

    host = get_host("default_port", hostname, driver.remote_agent_port)
    response = websocket_request("127.0.0.1", driver.remote_agent_port, host=host)
    assert response.status == status

    await driver.delete_session()