darkhelm/notifications-api
1
0
Fork 0
mirror of https://github.com/GSA/notifications-api.git synced 2026-08-24 16:23:44 -04:00
Code Issues Packages Projects Releases Wiki Activity

Stuff was done.

Browse Source 
Signed-off-by: Cliff Hill <Clifford.hill@gsa.gov>
This commit is contained in:
Cliff Hill
2024-02-13 14:48:10 -05:00
parent 63aa12e4ef
commit 3624cd812b
7 changed files with 490 additions and 244 deletions
Download Patch File Download Diff File Expand all files Collapse all files

515
tests/app/dao/notification_dao/test_notification_dao.py
View File

File diff suppressed because it is too large Load Diff

110
tests/app/dao/notification_dao/test_notification_dao_delete_notifications.py
View File

@@ -7,7 +7,7 @@ from app.dao.notifications_dao import (
insert_notification_history_delete_notifications,
move_notifications_to_notification_history,
)
from app.enums import KeyType
from app.enums import KeyType, NotificationStatus, NotificationType, TemplateType
from app.models import Notification, NotificationHistory
from tests.app.db import (
create_notification,
@@ -23,23 +23,26 @@ def test_move_notifications_does_nothing_if_notification_history_row_already_exi
notification = create_notification(
template=sample_email_template,
created_at=datetime.utcnow() - timedelta(days=8),
status="temporary-failure",
status=NotificationStatus.TEMPORARY_FAILURE,
)
create_notification_history(
id=notification.id,
template=sample_email_template,
created_at=datetime.utcnow() - timedelta(days=8),
status="delivered",
status=NotificationStatus.DELIVERED,
)
move_notifications_to_notification_history(
"email", sample_email_template.service_id, datetime.utcnow(), 1
NotificationType.EMAIL,
sample_email_template.service_id,
datetime.utcnow(),
1,
)
assert Notification.query.count() == 0
history = NotificationHistory.query.all()
assert len(history) == 1
assert history[0].status == "delivered"
assert history[0].status == NotificationStatus.DELIVERED
def test_move_notifications_only_moves_notifications_older_than_provided_timestamp(
@@ -59,7 +62,9 @@ def test_move_notifications_only_moves_notifications_older_than_provided_timesta
old_notification_id = old_notification.id
result = move_notifications_to_notification_history(
"sms", sample_template.service_id, delete_time
NotificationType.SMS,
sample_template.service_id,
delete_time,
)
assert result == 1
@@ -78,12 +83,15 @@ def test_move_notifications_keeps_calling_until_no_more_to_delete_and_then_retur
timestamp = datetime(2021, 1, 1)
result = move_notifications_to_notification_history(
"sms", service_id, timestamp, qry_limit=5
NotificationType.SMS,
service_id,
timestamp,
qry_limit=5,
)
assert result == 11
mock_insert.asset_called_with(
notification_type="sms",
notification_type=NotificationType.SMS,
service_id=service_id,
timestamp_to_delete_backwards_from=timestamp,
qry_limit=5,
@@ -95,17 +103,19 @@ def test_move_notifications_only_moves_for_given_notification_type(sample_servic
delete_time = datetime(2020, 6, 1, 12)
one_second_before = delete_time - timedelta(seconds=1)
sms_template = create_template(sample_service, "sms")
email_template = create_template(sample_service, "email")
sms_template = create_template(sample_service, TemplateType.SMS)
email_template = create_template(sample_service, TemplateType.EMAIL)
create_notification(sms_template, created_at=one_second_before)
create_notification(email_template, created_at=one_second_before)
result = move_notifications_to_notification_history(
"sms", sample_service.id, delete_time
NotificationType.SMS,
sample_service.id,
delete_time,
)
assert result == 1
assert {x.notification_type for x in Notification.query} == {"email"}
assert NotificationHistory.query.one().notification_type == "sms"
assert {x.notification_type for x in Notification.query} == {NotificationType.EMAIL}
assert NotificationHistory.query.one().notification_type == NotificationType.SMS
def test_move_notifications_only_moves_for_given_service(notify_db_session):
@@ -115,13 +125,17 @@ def test_move_notifications_only_moves_for_given_service(notify_db_session):
service = create_service(service_name="service")
other_service = create_service(service_name="other")
template = create_template(service, "sms")
other_template = create_template(other_service, "sms")
template = create_template(service, TemplateType.SMS)
other_template = create_template(other_service, TemplateType.SMS)
create_notification(template, created_at=one_second_before)
create_notification(other_template, created_at=one_second_before)
result = move_notifications_to_notification_history("sms", service.id, delete_time)
result = move_notifications_to_notification_history(
NotificationType.SMS,
service.id,
delete_time,
)
assert result == 1
assert NotificationHistory.query.one().service_id == service.id
@@ -132,17 +146,25 @@ def test_move_notifications_just_deletes_test_key_notifications(sample_template)
delete_time = datetime(2020, 6, 1, 12)
one_second_before = delete_time - timedelta(seconds=1)
create_notification(
template=sample_template, created_at=one_second_before, key_type=KeyType.NORMAL
template=sample_template,
created_at=one_second_before,
key_type=KeyType.NORMAL,
)
create_notification(
template=sample_template, created_at=one_second_before, key_type=KeyType.TEAM
template=sample_template,
created_at=one_second_before,
key_type=KeyType.TEAM,
)
create_notification(
template=sample_template, created_at=one_second_before, key_type=KeyType.TEST
template=sample_template,
created_at=one_second_before,
key_type=KeyType.TEST,
)
result = move_notifications_to_notification_history(
"sms", sample_template.service_id, delete_time
NotificationType.SMS,
sample_template.service_id,
delete_time,
)
assert result == 2
@@ -163,58 +185,58 @@ def test_insert_notification_history_delete_notifications(sample_email_template)
n1 = create_notification(
template=sample_email_template,
created_at=datetime.utcnow() - timedelta(days=1, minutes=4),
status="delivered",
status=NotificationStatus.DELIVERED,
)
n2 = create_notification(
template=sample_email_template,
created_at=datetime.utcnow() - timedelta(days=1, minutes=20),
status="permanent-failure",
status=NotificationStatus.PERMANENT_FAILURE,
)
n3 = create_notification(
template=sample_email_template,
created_at=datetime.utcnow() - timedelta(days=1, minutes=30),
status="temporary-failure",
status=NotificationStatus.TEMPORARY_FAILURE,
)
n4 = create_notification(
template=sample_email_template,
created_at=datetime.utcnow() - timedelta(days=1, minutes=59),
status="temporary-failure",
status=NotificationStatus.TEMPORARY_FAILURE,
)
n5 = create_notification(
template=sample_email_template,
created_at=datetime.utcnow() - timedelta(days=1, hours=1),
status="sending",
status=NotificationStatus.SENDING,
)
n6 = create_notification(
template=sample_email_template,
created_at=datetime.utcnow() - timedelta(days=1, minutes=61),
status="pending",
status=NotificationStatus.PENDING,
)
n7 = create_notification(
template=sample_email_template,
created_at=datetime.utcnow() - timedelta(days=1, hours=1, seconds=1),
status="validation-failed",
status=NotificationStatus.VALIDATION_FAILED,
)
n8 = create_notification(
template=sample_email_template,
created_at=datetime.utcnow() - timedelta(days=1, minutes=20),
status="created",
status=NotificationStatus.CREATED,
)
# should NOT be deleted - wrong status
n9 = create_notification(
template=sample_email_template,
created_at=datetime.utcnow() - timedelta(hours=1),
status="delivered",
status=NotificationStatus.DELIVERED,
)
n10 = create_notification(
template=sample_email_template,
created_at=datetime.utcnow() - timedelta(hours=1),
status="technical-failure",
status=NotificationStatus.TECHNICAL_FAILURE,
)
n11 = create_notification(
template=sample_email_template,
created_at=datetime.utcnow() - timedelta(hours=23, minutes=59),
status="created",
status=NotificationStatus.CREATED,
)
ids_to_move = sorted([n1.id, n2.id, n3.id, n4.id, n5.id, n6.id, n7.id, n8.id])
@@ -239,17 +261,17 @@ def test_insert_notification_history_delete_notifications_more_notifications_tha
create_notification(
template=sample_template,
created_at=datetime.utcnow() + timedelta(minutes=4),
status="delivered",
status=NotificationStatus.DELIVERED,
)
create_notification(
template=sample_template,
created_at=datetime.utcnow() + timedelta(minutes=20),
status="permanent-failure",
status=NotificationStatus.PERMANENT_FAILURE,
)
create_notification(
template=sample_template,
created_at=datetime.utcnow() + timedelta(minutes=30),
status="temporary-failure",
status=NotificationStatus.TEMPORARY_FAILURE,
)
del_count = insert_notification_history_delete_notifications(
@@ -272,14 +294,16 @@ def test_insert_notification_history_delete_notifications_only_insert_delete_for
notification_to_move = create_notification(
template=sample_email_template,
created_at=datetime.utcnow() + timedelta(minutes=4),
status="delivered",
status=NotificationStatus.DELIVERED,
)
another_service = create_service(service_name="Another service")
another_template = create_template(service=another_service, template_type="email")
another_template = create_template(
service=another_service, template_type=TemplateType.EMAIL
)
notification_to_stay = create_notification(
template=another_template,
created_at=datetime.utcnow() + timedelta(minutes=4),
status="delivered",
status=NotificationStatus.DELIVERED,
)
del_count = insert_notification_history_delete_notifications(
@@ -303,20 +327,20 @@ def test_insert_notification_history_delete_notifications_insert_for_key_type(
create_notification(
template=sample_template,
created_at=datetime.utcnow() - timedelta(hours=4),
status="delivered",
key_type="normal",
status=NotificationStatus.DELIVERED,
key_type=KeyType.NORMAL,
)
create_notification(
template=sample_template,
created_at=datetime.utcnow() - timedelta(hours=4),
status="delivered",
key_type="team",
status=NotificationStatus.DELIVERED,
key_type=KeyType.TEAM,
)
with_test_key = create_notification(
template=sample_template,
created_at=datetime.utcnow() - timedelta(hours=4),
status="delivered",
key_type="test",
status=NotificationStatus.DELIVERED,
key_type=KeyType.TEST,
)
del_count = insert_notification_history_delete_notifications(

26
tests/app/dao/test_templates_dao.py
View File

@@ -12,6 +12,7 @@ from app.dao.templates_dao import (
dao_redact_template,
dao_update_template,
)
from app.enums import TemplateProcessType, TemplateType
from app.models import Template, TemplateHistory, TemplateRedacted
from tests.app.db import create_template
@@ -19,8 +20,8 @@ from tests.app.db import create_template
@pytest.mark.parametrize(
"template_type, subject",
[
("sms", None),
("email", "subject"),
(TemplateType.SMS, None),
(TemplateType.EMAIL, "subject"),
],
)
def test_create_template(sample_service, sample_user, template_type, subject):
@@ -43,7 +44,8 @@ def test_create_template(sample_service, sample_user, template_type, subject):
== "Sample Template"
)
assert (
dao_get_all_templates_for_service(sample_service.id)[0].process_type == "normal"
dao_get_all_templates_for_service(sample_service.id)[0].process_type
== TemplateProcessType.NORMAL
)
@@ -61,7 +63,7 @@ def test_create_template_creates_redact_entry(sample_service):
def test_update_template(sample_service, sample_user):
data = {
"name": "Sample Template",
"template_type": "sms",
"template_type": TemplateType.SMS,
"content": "Template content",
"service": sample_service,
"created_by": sample_user,
@@ -101,19 +103,19 @@ def test_get_all_templates_for_service(service_factory):
create_template(
service=service_1,
template_name="Sample Template 1",
template_type="sms",
template_type=TemplateType.SMS,
content="Template content",
)
create_template(
service=service_1,
template_name="Sample Template 2",
template_type="sms",
template_type=TemplateType.SMS,
content="Template content",
)
create_template(
service=service_2,
template_name="Sample Template 3",
template_type="sms",
template_type=TemplateType.SMS,
content="Template content",
)
@@ -125,19 +127,19 @@ def test_get_all_templates_for_service(service_factory):
def test_get_all_templates_for_service_is_alphabetised(sample_service):
create_template(
template_name="Sample Template 1",
template_type="sms",
template_type=TemplateType.SMS,
content="Template content",
service=sample_service,
)
template_2 = create_template(
template_name="Sample Template 2",
template_type="sms",
template_type=TemplateType.SMS,
content="Template content",
service=sample_service,
)
create_template(
template_name="Sample Template 3",
template_type="sms",
template_type=TemplateType.SMS,
content="Template content",
service=sample_service,
)
@@ -259,7 +261,7 @@ def test_create_template_creates_a_history_record_with_current_data(
assert TemplateHistory.query.count() == 0
data = {
"name": "Sample Template",
"template_type": "email",
"template_type": TemplateType.EMAIL,
"subject": "subject",
"content": "Template content",
"service": sample_service,
@@ -288,7 +290,7 @@ def test_update_template_creates_a_history_record_with_current_data(
assert TemplateHistory.query.count() == 0
data = {
"name": "Sample Template",
"template_type": "email",
"template_type": TemplateType.EMAIL,
"subject": "subject",
"content": "Template content",
"service": sample_service,

15
tests/app/dao/test_uploads_dao.py
View File

@@ -3,7 +3,7 @@ from datetime import datetime, timedelta
from freezegun import freeze_time
from app.dao.uploads_dao import dao_get_uploads_by_service_id
from app.enums import JobStatus, TemplateType
from app.enums import JobStatus, NotificationStatus, NotificationType, TemplateType
from tests.app.db import (
create_job,
create_notification,
@@ -13,7 +13,12 @@ from tests.app.db import (
)
def create_uploaded_letter(letter_template, service, status="created", created_at=None):
def create_uploaded_letter(
letter_template,
service,
status=NotificationStatus.CREATED,
created_at=None,
):
return create_notification(
template=letter_template,
to_field="file-name",
@@ -39,7 +44,9 @@ def create_uploaded_template(service):
@freeze_time("2020-02-02 09:00") # GMT time
def test_get_uploads_for_service(sample_template):
create_service_data_retention(sample_template.service, "sms", days_of_retention=9)
create_service_data_retention(
sample_template.service, NotificationType.SMS, days_of_retention=9
)
job = create_job(sample_template, processing_started=datetime.utcnow())
other_service = create_service(service_name="other service")
@@ -55,7 +62,7 @@ def test_get_uploads_for_service(sample_template):
job.id,
job.original_file_name,
job.notification_count,
"sms",
TemplateType.SMS,
9,
job.created_at,
job.scheduled_for,

3
tests/app/dao/test_users_dao.py
View File

@@ -1,5 +1,6 @@
import uuid
from datetime import datetime, timedelta
from types import CodeType
import pytest
from freezegun import freeze_time
@@ -137,7 +138,7 @@ def test_should_not_delete_verification_codes_less_than_one_day_old(sample_user)
def make_verify_code(user, age=None, expiry_age=None, code="12335", code_used=False):
verify_code = VerifyCode(
code_type="sms",
code_type=CodeType.SMS,
_code=code,
created_at=datetime.utcnow() - (age or timedelta(hours=0)),
expiry_datetime=datetime.utcnow() - (expiry_age or timedelta(0)),