More Enum goodness.

Signed-off-by: Cliff Hill <Clifford.hill@gsa.gov>
This commit is contained in:
Cliff Hill
2024-01-18 10:26:40 -05:00
parent 43f18eed6a
commit df866e40f7
26 changed files with 379 additions and 278 deletions

View File

@@ -11,7 +11,7 @@ from notifications_utils.recipients import (
)
from sqlalchemy.exc import SQLAlchemyError
from app.models import SMS_TYPE, Notification, NotificationHistory
from app.models import Notification, NotificationHistory, ServicePermissionType, TemplateType
from app.notifications.process_notifications import (
create_content_for_notification,
persist_notification,
@@ -403,8 +403,8 @@ def test_persist_email_notification_stores_normalised_email(
def test_persist_notification_with_billable_units_stores_correct_info(mocker):
service = create_service(service_permissions=[SMS_TYPE])
template = create_template(service, template_type=SMS_TYPE)
service = create_service(service_permissions=[ServicePermissionType.SMS])
template = create_template(service, template_type=TemplateType.SMS)
mocker.patch("app.dao.templates_dao.dao_get_template_by_id", return_value=template)
persist_notification(
template_id=template.id,