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