def test_fetch_service_data_retention_by_id(sample_service):
email_data_retention = insert_service_data_retention(
sample_service.id, 'email', 3)
insert_service_data_retention(sample_service.id, 'sms', 13)
result = fetch_service_data_retention_by_id(sample_service.id,
email_data_retention.id)
assert result == email_data_retention
def test_fetch_service_data_retention_by_id_returns_none_if_id_not_for_service(
sample_service):
another_service = create_service(service_name="Another service")
email_data_retention = insert_service_data_retention(
sample_service.id, 'email', 3)
result = fetch_service_data_retention_by_id(another_service.id,
email_data_retention.id)
assert not result
def get_data_retention_for_service_by_id(service_id, data_retention_id):
data_retention = fetch_service_data_retention_by_id(
service_id, data_retention_id)
return jsonify(data_retention.serialize() if data_retention else {}), 200
def test_fetch_service_data_retention_by_id_returns_none_if_not_found(
sample_service):
result = fetch_service_data_retention_by_id(sample_service.id,
uuid.uuid4())
assert not result
