import re

import pytest


@pytest.mark.parametrize(
    "query_args, result", [({}, True), ({"govuk_banner": "false"}, "false")]
)
def test_renders(client_request, mocker, query_args, result):
    mocker.patch(
        "app.main.views.index.HTMLEmailTemplate.__str__", return_value="rendered"
    )

    response = client_request.get_response("main.email_template", **query_args)

    assert response.get_data(as_text=True) == "rendered"


def test_displays_both_branding(
    client_request, mock_get_email_branding_with_both_brand_type
):
    page = client_request.get(
        "main.email_template", branding_style="1", _test_page_title=False
    )

    mock_get_email_branding_with_both_brand_type.assert_called_once_with("1")

    assert page.find("img", attrs={"src": re.compile("example.png$")})
    assert (
        page.select(
            "body > table:nth-of-type(3) table > tr:nth-of-type(1) > td:nth-of-type(2)"
        )[0]
        .get_text()
        .strip()
        == "Organization text"
    )  # brand text is set


def test_displays_org_branding(client_request, mock_get_email_branding):
    # mock_get_email_branding has 'brand_type' of 'org'
    page = client_request.get(
        "main.email_template", branding_style="1", _test_page_title=False
    )

    mock_get_email_branding.assert_called_once_with("1")

    assert not page.find("a", attrs={"href": "https://www.gsa.gov"})
    assert page.find("img", attrs={"src": re.compile("example.png")})
    assert not page.select(
        "body > table > tr > td[bgcolor='#f00']"
    )  # banner colour is not set
    assert (
        page.select(
            "body > table:nth-of-type(1) > tr:nth-of-type(1) > td:nth-of-type(2)"
        )[0]
        .get_text()
        .strip()
        == "Organization text"
    )  # brand text is set


def test_displays_org_branding_with_banner(
    client_request, mock_get_email_branding_with_org_banner_brand_type
):
    page = client_request.get(
        "main.email_template", branding_style="1", _test_page_title=False
    )

    mock_get_email_branding_with_org_banner_brand_type.assert_called_once_with("1")

    assert not page.find("a", attrs={"href": "https://www.gsa.gov"})
    assert page.find("img", attrs={"src": re.compile("example.png")})
    assert page.select("body > table > tr > td[bgcolor='#f00']")  # banner colour is set
    assert (
        page.select("body > table table > tr > td > span")[0].get_text().strip()
        == "Organization text"
    )  # brand text is set


def test_displays_org_branding_with_banner_without_brand_text(
    client_request, mock_get_email_branding_without_brand_text
):
    # mock_get_email_branding_without_brand_text has 'brand_type' of 'org_banner'
    page = client_request.get(
        "main.email_template", branding_style="1", _test_page_title=False
    )

    mock_get_email_branding_without_brand_text.assert_called_once_with("1")

    assert not page.find("a", attrs={"href": "https://www.gsa.gov"})
    assert page.find("img", attrs={"src": re.compile("example.png")})
    assert page.select("body > table > tr > td[bgcolor='#f00']")  # banner colour is set
    assert (
        not page.select("body > table table > tr > td > span") == 0
    )  # brand text is not set