mirror of
https://github.com/GSA/notifications-api.git
synced 2025-12-21 16:01:15 -05:00
107 lines
4.5 KiB
Python
107 lines
4.5 KiB
Python
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
|