def get_data_retention_for_service(service_id): data_retention_list = fetch_service_data_retention(service_id) return ( jsonify([ data_retention.serialize() for data_retention in data_retention_list ]), 200, )
def test_fetch_service_data_retention_only_returns_row_for_service(sample_service): another_service = create_service(service_name="Another service") email_data_retention = insert_service_data_retention(sample_service.id, 'email', 3) letter_data_retention = insert_service_data_retention(sample_service.id, 'letter', 30) insert_service_data_retention(another_service.id, 'sms', 5) list_of_data_retention = fetch_service_data_retention(sample_service.id) assert len(list_of_data_retention) == 2 assert list_of_data_retention[0] == email_data_retention assert list_of_data_retention[1] == letter_data_retention
def test_fetch_service_data_retention(sample_service): email_data_retention = insert_service_data_retention(sample_service.id, 'email', 3) letter_data_retention = insert_service_data_retention(sample_service.id, 'letter', 30) sms_data_retention = insert_service_data_retention(sample_service.id, 'sms', 5) list_of_data_retention = fetch_service_data_retention(sample_service.id) assert len(list_of_data_retention) == 3 assert list_of_data_retention[0] == email_data_retention assert list_of_data_retention[1] == sms_data_retention assert list_of_data_retention[2] == letter_data_retention
def test_fetch_service_data_retention_returns_empty_list_when_no_rows_for_service( sample_service): empty_list = fetch_service_data_retention(sample_service.id) assert not empty_list