def get_whitelist(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) whitelist = dao_fetch_service_whitelist(service.id) return jsonify( email_addresses=[item.recipient for item in whitelist if item.recipient_type == EMAIL_TYPE], phone_numbers=[item.recipient for item in whitelist if item.recipient_type == MOBILE_TYPE] )
def test_fetch_service_whitelist_ignores_other_service(sample_service_whitelist): assert len(dao_fetch_service_whitelist(uuid.uuid4())) == 0
def test_fetch_service_whitelist_gets_whitelists(sample_service_whitelist): whitelist = dao_fetch_service_whitelist(sample_service_whitelist.service_id) assert len(whitelist) == 1 assert whitelist[0].id == sample_service_whitelist.id
def test_fetch_service_whitelist_ignores_other_service(sample_service_whitelist): assert len(dao_fetch_service_whitelist(uuid.uuid4())) == 0
def test_fetch_service_whitelist_gets_whitelists(sample_service_whitelist): whitelist = dao_fetch_service_whitelist(sample_service_whitelist.service_id) assert len(whitelist) == 1 assert whitelist[0].id == sample_service_whitelist.id