import re

import pytest
from bs4 import BeautifulSoup
from flask import url_for

from app.main.views.index import _set_colour


@pytest.mark.parametrize(
    "query_args, result", [
        ({}, True),
        ({'govuk_banner': 'false'}, 'false')
    ]
)
def test_renders(client, mocker, query_args, result):

    mocker.patch('app.main.views.index.HTMLEmailTemplate.__str__', return_value='rendered')

    response = client.get(url_for('main.email_template', **query_args))

    assert response.status_code == 200
    assert response.get_data(as_text=True) == 'rendered'


def test_displays_govuk_branding_by_default(client):

    response = client.get(url_for('main.email_template'))

    page = BeautifulSoup(response.data.decode("utf-8"), "html.parser")

    assert response.status_code == 200

    assert page.find("a", attrs={"href": "https://www.gov.uk"})


def test_displays_govuk_branding(client):

    response = client.get(url_for('main.email_template', branding_type="govuk", branding_style="1"))

    page = BeautifulSoup(response.data.decode("utf-8"), "html.parser")

    assert response.status_code == 200

    assert page.find("a", attrs={"href": "https://www.gov.uk"})


def test_displays_both_branding(client, mock_get_email_branding):

    response = client.get(url_for('main.email_template', branding_type="both", branding_style="1"))

    page = BeautifulSoup(response.data.decode("utf-8"), "html.parser")

    assert response.status_code == 200
    mock_get_email_branding.assert_called_once_with('1')

    assert page.find("a", attrs={"href": "https://www.gov.uk"})
    assert page.find("img", attrs={"src": re.compile("example.png$")})
    assert page.select("body > table table > tr:nth-of-type(2) > td:nth-of-type(5)")[0]\
        .get_text().strip() == 'Organisation text'  # brand text is set


def test_displays_org_branding(client, mock_get_email_branding):

    response = client.get(url_for('main.email_template', branding_type="org", branding_style="1"))

    page = BeautifulSoup(response.data.decode("utf-8"), "html.parser")

    assert response.status_code == 200
    mock_get_email_branding.assert_called_once_with('1')

    assert not page.find("a", attrs={"href": "https://www.gov.uk"})
    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 table > tr:nth-of-type(2) > td:nth-of-type(5)")[0]\
        .get_text().strip() == 'Organisation text'  # brand text is set


def test_displays_org_branding_with_banner(client, mock_get_email_branding):

    response = client.get(url_for('main.email_template', branding_type="org_banner",
                                  branding_style="1"))

    page = BeautifulSoup(response.data.decode("utf-8"), "html.parser")

    assert response.status_code == 200
    mock_get_email_branding.assert_called_once_with('1')

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


def test_displays_org_branding_with_banner_without_brand_text(
        client, mock_get_email_branding_without_brand_text):

    response = client.get(url_for('main.email_template', branding_type="org_banner",
                                  branding_style="1"))

    page = BeautifulSoup(response.data.decode("utf-8"), "html.parser")

    assert response.status_code == 200
    mock_get_email_branding_without_brand_text.assert_called_once_with('1')

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


@pytest.mark.parametrize('colour, banner_colour, single_id_colour, branding_type, expected_colour', [
    ('black', 'yellow', 'red', 'org', 'red'),
    ('black', 'yellow', None, 'org', 'black'),
    ('black', 'yellow', 'red', 'org_banner', 'yellow'),
    ('black', None, 'red', 'org_banner', 'black'),
    ('black', 'yellow', 'red', 'govuk', None),
    ('black', 'yellow', 'red', 'both', 'red'),
    ('black', 'yellow', None, 'both', 'black'),
])
def test_set_colour(colour, banner_colour, single_id_colour, branding_type, expected_colour):
    email_branding = {
        'logo': None,
        'colour': colour,
        'text': 'new text',
        'name': 'new name',
        'domain': 'sample.com',
        'banner_colour': banner_colour,
        'single_id_colour': single_id_colour,
    }
    colour = _set_colour(branding_type, email_branding)
    assert colour == expected_colour