def get_guest_list(service_id): from app.models import EMAIL_TYPE, MOBILE_TYPE service = dao_fetch_service_by_id(service_id) if not service: raise InvalidRequest("Service does not exist", status_code=404) guest_list = dao_fetch_service_guest_list(service.id) return jsonify( email_addresses=[item.recipient for item in guest_list if item.recipient_type == EMAIL_TYPE], phone_numbers=[item.recipient for item in guest_list if item.recipient_type == MOBILE_TYPE] )
def test_fetch_service_guest_list_ignores_other_service( sample_service_guest_list): assert len(dao_fetch_service_guest_list(uuid.uuid4())) == 0
def test_fetch_service_guest_list_gets_guest_lists(sample_service_guest_list): guest_list = dao_fetch_service_guest_list( sample_service_guest_list.service_id) assert len(guest_list) == 1 assert guest_list[0].id == sample_service_guest_list.id