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

@@ -2,7 +2,7 @@ import json
import uuid
from app.dao.service_guest_list_dao import dao_add_and_commit_guest_list_contacts
from app.models import EMAIL_TYPE, MOBILE_TYPE, ServiceGuestList
from app.models import GuestListRecipientType, ServiceGuestList
from tests import create_admin_authorization_header
@@ -24,11 +24,15 @@ def test_get_guest_list_separates_emails_and_phones(client, sample_service):
dao_add_and_commit_guest_list_contacts(
[
ServiceGuestList.from_string(
sample_service.id, EMAIL_TYPE, "service@example.com"
sample_service.id,
GuestListRecipientType.EMAIL,
"service@example.com",
),
ServiceGuestList.from_string(sample_service.id, MOBILE_TYPE, "2028675309"),
ServiceGuestList.from_string(sample_service.id, GuestListRecipientType.MOBILE, "2028675309"),
ServiceGuestList.from_string(
sample_service.id, MOBILE_TYPE, "+1800-555-5555"
sample_service.id,
GuestListRecipientType.MOBILE,
"+1800-555-5555",
),
]
)