def test_data_fetching_total_record_count_for_home_visit_api(self):
"""
test to check the total count of records that are returned from the home visit api
"""
limit = GOVERNANCE_API_RECORDS_PAGINATION
query_filters = {'aggregation_level': 5}
order = ['awc_id']
data, count = get_home_visit_data(limit, 2017, 5, order, query_filters)
expected_count = 55
self.assertEqual(count, expected_count)
def test_data_fetching_total_record_count_with_state_id_for_home_visit_api(
self):
"""
test to check the total count of records that are returned from the home visit api
"""
limit = GOVERNANCE_API_PAGE_SIZE
query_filters = {'aggregation_level': 5, 'state_id': 'st1'}
order = ['awc_id']
data, count = get_home_visit_data(limit, 2017, 5, order, query_filters)
expected_count = 26
self.assertEqual(count, expected_count)
def test_data_fetching_total_record_count_for_no_records_for_home_visit_api(
self):
"""
test to check the no records are returned from the home visit api
"""
limit = GOVERNANCE_API_PAGE_SIZE
query_filters = {'aggregation_level': 5}
order = ['awc_id']
data, count = get_home_visit_data(limit, 2017, 6, order, query_filters)
expected_count = 0
self.assertEqual(count, expected_count)
self.assertEqual(data, [])
def test_data_fetching_without_start_for_home_visit_api(self):
"""
test to check the first record that is returned from the home visit api without start parameter
"""
limit = GOVERNANCE_API_RECORDS_PAGINATION
query_filters = {'aggregation_level': 5}
order = ['awc_id']
data, count = get_home_visit_data(limit, 2017, 5, order, query_filters)
expected_first_row = {
'awc_id': 'a1',
'awc_code': 'a1',
'valid_visits': 0,
'expected_visits': 4,
}
self.assertEqual(data[0], expected_first_row)
def test_data_fetching_with_start_for_home_visit_api(self):
"""
test to check the first record that is returned from the home visit api with start parameter
"""
limit = CAS_API_PAGE_SIZE
query_filters = {'aggregation_level': 5, 'awc_id__gt': 'a1'}
order = ['awc_id']
data, count = get_home_visit_data(limit, 2017, 5, order, query_filters)
expected_first_row = {
'awc_id': 'a10',
'awc_code': 'a10',
'valid_visits': 0,
'expected_visits': 2
}
self.assertEqual(data[0], expected_first_row)
