from itertools import product

import pytest
from flask import json

from app.models import EMAIL_TYPE, TEMPLATE_TYPES
from tests import create_service_authorization_header
from tests.app.db import create_template


def test_get_all_templates_returns_200(client, sample_service):
    templates = [
        create_template(
            sample_service,
            template_type=tmp_type,
            subject="subject_{}".format(name) if tmp_type == EMAIL_TYPE else "",
            template_name=name,
        )
        for name, tmp_type in product(("A", "B", "C"), TEMPLATE_TYPES)
    ]

    auth_header = create_service_authorization_header(service_id=sample_service.id)

    response = client.get(
        path="/v2/templates",
        headers=[("Content-Type", "application/json"), auth_header],
    )

    assert response.status_code == 200
    assert response.headers["Content-type"] == "application/json"

    json_response = json.loads(response.get_data(as_text=True))

    assert len(json_response["templates"]) == len(templates)

    for index, template in enumerate(json_response["templates"]):
        assert template["id"] == str(templates[index].id)
        assert template["body"] == templates[index].content
        assert template["type"] == templates[index].template_type
        if templates[index].template_type == EMAIL_TYPE:
            assert template["subject"] == templates[index].subject


@pytest.mark.parametrize("tmp_type", TEMPLATE_TYPES)
def test_get_all_templates_for_valid_type_returns_200(client, sample_service, tmp_type):
    templates = [
        create_template(
            sample_service,
            template_type=tmp_type,
            template_name="Template {}".format(i),
            subject="subject_{}".format(i) if tmp_type == EMAIL_TYPE else "",
        )
        for i in range(3)
    ]

    auth_header = create_service_authorization_header(service_id=sample_service.id)

    response = client.get(
        path="/v2/templates?type={}".format(tmp_type),
        headers=[("Content-Type", "application/json"), auth_header],
    )

    assert response.status_code == 200
    assert response.headers["Content-type"] == "application/json"

    json_response = json.loads(response.get_data(as_text=True))

    assert len(json_response["templates"]) == len(templates)

    for index, template in enumerate(json_response["templates"]):
        assert template["id"] == str(templates[index].id)
        assert template["body"] == templates[index].content
        assert template["type"] == tmp_type
        if templates[index].template_type == EMAIL_TYPE:
            assert template["subject"] == templates[index].subject


@pytest.mark.parametrize("tmp_type", TEMPLATE_TYPES)
def test_get_correct_num_templates_for_valid_type_returns_200(
    client, sample_service, tmp_type
):
    num_templates = 3

    templates = []
    for _ in range(num_templates):
        templates.append(create_template(sample_service, template_type=tmp_type))

    for other_type in TEMPLATE_TYPES:
        if other_type != tmp_type:
            templates.append(create_template(sample_service, template_type=other_type))

    auth_header = create_service_authorization_header(service_id=sample_service.id)

    response = client.get(
        path="/v2/templates?type={}".format(tmp_type),
        headers=[("Content-Type", "application/json"), auth_header],
    )

    assert response.status_code == 200

    json_response = json.loads(response.get_data(as_text=True))

    assert len(json_response["templates"]) == num_templates


def test_get_all_templates_for_invalid_type_returns_400(client, sample_service):
    auth_header = create_service_authorization_header(service_id=sample_service.id)

    invalid_type = "coconut"

    response = client.get(
        path="/v2/templates?type={}".format(invalid_type),
        headers=[("Content-Type", "application/json"), auth_header],
    )

    assert response.status_code == 400
    assert response.headers["Content-type"] == "application/json"

    json_response = json.loads(response.get_data(as_text=True))

    assert json_response == {
        "status_code": 400,
        "errors": [
            {
                "message": "type coconut is not one of [sms, email]",
                "error": "ValidationError",
            }
        ],
    }