notifications-api/tests/app/template/test_rest_history.py

import json
from datetime import datetime

from flask import url_for

from app.dao.templates_dao import dao_update_template
from app.enums import TemplateProcessType
from app.utils import utc_now
from tests import create_admin_authorization_header


def test_template_history_version(notify_api, sample_user, sample_template):
    with notify_api.test_request_context():
        with notify_api.test_client() as client:
            auth_header = create_admin_authorization_header()
            endpoint = url_for(
                "template.get_template_version",
                service_id=sample_template.service.id,
                template_id=sample_template.id,
                version=1,
            )
            resp = client.get(
                endpoint, headers=[("Content-Type", "application/json"), auth_header]
            )
            assert resp.status_code == 200
            json_resp = json.loads(resp.get_data(as_text=True))
            assert json_resp["data"]["id"] == str(sample_template.id)
            assert json_resp["data"]["content"] == sample_template.content
            assert json_resp["data"]["version"] == 1
            assert json_resp["data"]["process_type"] == TemplateProcessType.NORMAL
            assert json_resp["data"]["created_by"]["name"] == sample_user.name
            assert (
                datetime.strptime(
                    json_resp["data"]["created_at"], "%Y-%m-%d %H:%M:%S.%f"
                ).date()
                == utc_now().date()
            )


def test_previous_template_history_version(notify_api, sample_template):
    old_content = sample_template.content
    sample_template.content = "New content"
    sample_template.process_type = TemplateProcessType.PRIORITY
    dao_update_template(sample_template)
    with notify_api.test_request_context():
        with notify_api.test_client() as client:
            auth_header = create_admin_authorization_header()
            endpoint = url_for(
                "template.get_template_version",
                service_id=sample_template.service.id,
                template_id=sample_template.id,
                version=1,
            )
            resp = client.get(
                endpoint, headers=[("Content-Type", "application/json"), auth_header]
            )
            assert resp.status_code == 200
            json_resp = json.loads(resp.get_data(as_text=True))
            assert json_resp["data"]["id"] == str(sample_template.id)
            assert json_resp["data"]["version"] == 1
            assert json_resp["data"]["content"] == old_content
            assert json_resp["data"]["process_type"] == TemplateProcessType.NORMAL


def test_404_missing_template_version(notify_api, sample_template):
    with notify_api.test_request_context():
        with notify_api.test_client() as client:
            auth_header = create_admin_authorization_header()
            endpoint = url_for(
                "template.get_template_version",
                service_id=sample_template.service.id,
                template_id=sample_template.id,
                version=2,
            )
            resp = client.get(
                endpoint, headers=[("Content-Type", "application/json"), auth_header]
            )
            assert resp.status_code == 404


def test_all_versions_of_template(notify_api, sample_template):
    with notify_api.test_request_context():
        with notify_api.test_client() as client:
            old_content = sample_template.content
            newer_content = "Newer content"
            newest_content = "Newest content"
            sample_template.content = newer_content
            dao_update_template(sample_template)
            sample_template.content = newest_content
            dao_update_template(sample_template)
            auth_header = create_admin_authorization_header()
            endpoint = url_for(
                "template.get_template_versions",
                service_id=sample_template.service.id,
                template_id=sample_template.id,
            )
            resp = client.get(
                endpoint, headers=[("Content-Type", "application/json"), auth_header]
            )
            json_resp = json.loads(resp.get_data(as_text=True))
            assert len(json_resp["data"]) == 3
            assert json_resp["data"][0]["content"] == newest_content
            assert json_resp["data"][0]["updated_at"]
            assert json_resp["data"][1]["content"] == newer_content
            assert json_resp["data"][1]["updated_at"]
            assert json_resp["data"][2]["content"] == old_content