class TestSearchFunctions(ViewTestBase):
def setUp(self):
self.elastic_search_api = ElasticAPI()
self.elastic_search_api.setup_index(index_name='test_index')
super(TestSearchFunctions, self).setUp()
def test_serialize_model(self):
self.maxDiff = None
facility = mommy.make(Facility)
serialized_data = serialize_model(facility)
expected_data = FacilitySerializer(facility).data
expected_data = json.dumps(expected_data, default=default)
self.assertEquals(expected_data, serialized_data.get('data'))
def test_serialize_model_serializer_not_found(self):
# There is no serializer named GroupSerializer
# Therefore the serialize model function should return None
group = mommy.make(Group)
serialized_data = serialize_model(group)
self.assertIsNone(serialized_data)
def test_default_json_dumps_function(self):
facility = mommy.make(Facility)
data = FacilitySerializer(facility).data
result = json.dumps(data, default=default)
self.assertIsInstance(result, str)
def test_search_facility(self):
url = reverse('api:facilities:facilities_list')
self.elastic_search_api = ElasticAPI()
self.elastic_search_api.setup_index(index_name='test_index')
facility = mommy.make(Facility, name='Kanyakini')
index_instance(facility, 'test_index')
url = url + "?search={}".format('Kanyakini')
response = self.client.get(url)
self.assertEquals(200, response.status_code)
self.assertIsInstance(response.data.get('results'), list)
self.elastic_search_api.delete_index('test_index')
def test_search_facility_using_material_view(self):
self.elastic_search_api = ElasticAPI()
self.elastic_search_api.setup_index(index_name='test_index')
facility = mommy.make(Facility, name='Facility ya kutest material')
url = reverse('api:facilities:material')
index_instance(facility, 'test_index')
url = url + "?search={}".format('material')
response = self.client.get(url)
self.assertEquals(200, response.status_code)
self.assertIsInstance(response.data.get('results'), list)
self.elastic_search_api.delete_index('test_index')
def test_search_facility_using_query_dsl(self):
url = reverse('api:facilities:facilities_list')
self.elastic_search_api = ElasticAPI()
self.elastic_search_api.setup_index(index_name='test_index')
facility = mommy.make(Facility, name='Kanyakini')
index_instance(facility, 'test_index')
query_dsl = {
"from": 0,
"size": 30,
"query": {
"query_string": {
"default_field": "name",
"query": "Kanyakini"
}
}
}
url = url + "?search={}".format(json.dumps(query_dsl))
response = self.client.get(url)
self.assertEquals(200, response.status_code)
self.assertIsInstance(response.data.get('results'), list)
self.elastic_search_api.delete_index('test_index')
def test_seach_auto_complete(self):
url = reverse('api:facilities:facilities_list')
self.elastic_search_api = ElasticAPI()
self.elastic_search_api.setup_index(index_name='test_index')
facility = mommy.make(Facility, name='Kanyakini')
index_instance(facility, 'test_index')
url = url + "?search_auto={}".format('Kanya')
response = self.client.get(url)
self.assertEquals(200, response.status_code)
self.assertIsInstance(response.data.get('results'), list)
self.elastic_search_api.delete_index('test_index')
def test_search_facility_multiple_filters(self):
url = reverse('api:facilities:facilities_list')
self.elastic_search_api = ElasticAPI()
self.elastic_search_api.setup_index(index_name='test_index')
facility = mommy.make(Facility, name='Mordal mountains medical clinic')
mommy.make(FacilityApproval, facility=facility)
facility.is_published = True
facility.save()
facility_2 = mommy.make(Facility,
name='Eye of mordal health center',
is_published=False)
index_instance(facility, 'test_index')
index_instance(facility_2, 'test_index')
url = url + "?search={}&is_published={}".format('mordal', 'false')
response = self.client.get(url)
self.assertEquals(200, response.status_code)
self.assertIsInstance(response.data.get('results'), list)
self.elastic_search_api.delete_index('test_index')
def test_get_search_auto_complete_fields(self):
search_fields = self.elastic_search_api.get_search_fields('facility')
self.assertIsInstance(search_fields, list)
self.assertEquals(len(search_fields), 10)
def test_get_search_auto_complete_fields_si_empty(self):
search_fields = self.elastic_search_api.get_search_fields('no_model')
self.assertIsNone(search_fields)
def test_user_search(self):
user = mommy.make(get_user_model(), is_national=True)
self.client.force_authenticate(user)
county_user = mommy.make(get_user_model())
sub_county_user = mommy.make(get_user_model(),
first_name='kijana',
last_name='mzee')
index_instance(sub_county_user, 'test_index')
url = reverse("api:users:mfl_users_list")
url = url + "?search=kijana"
response = self.client.get(url)
self.assertEquals(200, response.status_code)
# the national users are not allowed to see the sub county level users
county = mommy.make(County)
const = mommy.make(Constituency, county=county)
mommy.make(UserCounty, county=county, user=county_user)
mommy.make(UserConstituency,
user=sub_county_user,
constituency=const,
created_by=county_user,
updated_by=county_user)
filtered_response = self.client.get(url)
self.assertEquals(200, filtered_response.status_code)
def test_chu_search(self):
facility = mommy.make(Facility)
facility_2 = mommy.make(Facility)
chu_url = reverse("api:chul:community_health_units_list")
chu = mommy.make(CommunityHealthUnit,
facility=facility,
name='Jericho')
mommy.make(CommunityHealthUnit, facility=facility_2)
index_instance(chu)
chu_search_url = chu_url + "?search=Jericho"
response = self.client.get(chu_search_url)
self.assertEquals(200, response.status_code)
def test_chu_search_using_code(self):
facility = mommy.make(Facility)
facility_2 = mommy.make(Facility)
chu_url = reverse("api:chul:community_health_units_list")
chu = mommy.make(CommunityHealthUnit,
facility=facility,
name='Jericho')
mommy.make(CommunityHealthUnit, facility=facility_2)
index_instance(chu)
chu_search_url = chu_url + "?search={}".format(chu.code)
response = self.client.get(chu_search_url)
self.assertEquals(200, response.status_code)
def tearDown(self):
self.elastic_search_api.delete_index(index_name='test_index')
super(TestSearchFunctions, self).tearDown()
class TestSearchFunctions(ViewTestBase):
def setUp(self):
self.elastic_search_api = ElasticAPI()
self.elastic_search_api.setup_index(index_name='test_index')
super(TestSearchFunctions, self).setUp()
def test_serialize_model(self):
self.maxDiff = None
facility = mommy.make(Facility)
serialized_data = serialize_model(facility)
expected_data = FacilitySerializer(
facility
).data
expected_data = json.dumps(expected_data, default=default)
self.assertEquals(expected_data, serialized_data.get('data'))
def test_serialize_model_serializer_not_found(self):
# There is no serializer named GroupSerializer
# Therefore the serialize model function should return None
group = mommy.make(Group)
serialized_data = serialize_model(group)
self.assertIsNone(serialized_data)
def test_default_json_dumps_function(self):
facility = mommy.make(Facility)
data = FacilitySerializer(
facility
).data
result = json.dumps(data, default=default)
self.assertIsInstance(result, str)
def test_search_facility(self):
url = reverse('api:facilities:facilities_list')
self.elastic_search_api = ElasticAPI()
self.elastic_search_api.setup_index(index_name='test_index')
facility = mommy.make(Facility, name='Kanyakini')
index_instance(
'facilities', 'Facility', str(facility.id), 'test_index')
url = url + "?search={}".format('Kanyakini')
response = self.client.get(url)
self.assertEquals(200, response.status_code)
self.assertIsInstance(response.data.get('results'), list)
self.elastic_search_api.delete_index('test_index')
def test_search_facility_using_material_view(self):
self.elastic_search_api = ElasticAPI()
self.elastic_search_api.setup_index(index_name='test_index')
facility = mommy.make(Facility, name='Facility ya kutest material')
url = reverse('api:facilities:material')
index_instance(
'facilities', 'Facility', str(facility.id), 'test_index')
url = url + "?search={}".format('material')
response = self.client.get(url)
self.assertEquals(200, response.status_code)
self.assertIsInstance(response.data.get('results'), list)
self.elastic_search_api.delete_index('test_index')
def test_search_facility_using_query_dsl(self):
url = reverse('api:facilities:facilities_list')
self.elastic_search_api = ElasticAPI()
self.elastic_search_api.setup_index(index_name='test_index')
facility = mommy.make(Facility, name='Kanyakini')
index_instance(
'facilities', 'Facility', str(facility.id), 'test_index')
query_dsl = {
"from": 0,
"size": 30,
"query": {
"query_string": {
"default_field": "name",
"query": "Kanyakini"
}
}
}
url = url + "?search={}".format(json.dumps(query_dsl))
response = self.client.get(url)
self.assertEquals(200, response.status_code)
self.assertIsInstance(response.data.get('results'), list)
self.elastic_search_api.delete_index('test_index')
def test_seach_auto_complete(self):
url = reverse('api:facilities:facilities_list')
self.elastic_search_api = ElasticAPI()
self.elastic_search_api.setup_index(index_name='test_index')
facility = mommy.make(Facility, name='Kanyakini')
index_instance(
'facilities', 'Facility', str(facility.id), 'test_index')
url = url + "?search_auto={}".format('Kanya')
response = self.client.get(url)
self.assertEquals(200, response.status_code)
self.assertIsInstance(response.data.get('results'), list)
self.elastic_search_api.delete_index('test_index')
def test_search_facility_multiple_filters(self):
url = reverse('api:facilities:facilities_list')
self.elastic_search_api = ElasticAPI()
self.elastic_search_api.setup_index(index_name='test_index')
facility = mommy.make(
Facility, name='Mordal mountains medical clinic')
mommy.make(FacilityApproval, facility=facility)
facility.is_published = True
facility.save()
facility_2 = mommy.make(
Facility,
name='Eye of mordal health center',
is_published=False)
index_instance(
'facilities', 'Facility', str(facility.id), 'test_index')
index_instance(
'facilities', 'Facility', str(facility_2.id), 'test_index')
url = url + "?search={}&is_published={}".format('mordal', 'false')
response = self.client.get(url)
self.assertEquals(200, response.status_code)
self.assertIsInstance(response.data.get('results'), list)
self.elastic_search_api.delete_index('test_index')
def test_get_search_auto_complete_fields(self):
search_fields = self.elastic_search_api.get_search_fields('facility')
self.assertIsInstance(search_fields, list)
self.assertEquals(len(search_fields), 10)
def test_get_search_auto_complete_fields_si_empty(self):
search_fields = self.elastic_search_api.get_search_fields('no_model')
self.assertIsNone(search_fields)
def test_user_search(self):
user = mommy.make(get_user_model(), is_national=True)
self.client.force_authenticate(user)
county_user = mommy.make(get_user_model())
sub_county_user = mommy.make(
get_user_model(), first_name='kijana', last_name='mzee')
index_instance(
'users', 'MflUser', str(sub_county_user.id), 'test_index')
url = reverse("api:users:mfl_users_list")
url = url + "?search=kijana"
response = self.client.get(url)
self.assertEquals(200, response.status_code)
# the national users are not allowed to see the sub county level users
county = mommy.make(County)
const = mommy.make(Constituency, county=county)
mommy.make(UserCounty, county=county, user=county_user)
mommy.make(
UserConstituency, user=sub_county_user, constituency=const,
created_by=county_user, updated_by=county_user)
filtered_response = self.client.get(url)
self.assertEquals(200, filtered_response.status_code)
def test_chu_search(self):
facility = mommy.make(Facility)
facility_2 = mommy.make(Facility)
chu_url = reverse("api:chul:community_health_units_list")
chu = mommy.make(
CommunityHealthUnit, facility=facility, name='Jericho')
mommy.make(CommunityHealthUnit, facility=facility_2)
index_instance(
'chul', 'CommunityHealthUnit', str(chu.id), 'test_index')
chu_search_url = chu_url + "?search=Jericho"
response = self.client.get(chu_search_url)
self.assertEquals(200, response.status_code)
def test_chu_search_using_code(self):
facility = mommy.make(Facility)
facility_2 = mommy.make(Facility)
chu_url = reverse("api:chul:community_health_units_list")
chu = mommy.make(
CommunityHealthUnit, facility=facility, name='Jericho')
mommy.make(CommunityHealthUnit, facility=facility_2)
index_instance(
'chul', 'CommunityHealthUnit', str(chu.id), 'test_index')
chu_search_url = chu_url + "?search={}".format(chu.code)
response = self.client.get(chu_search_url)
self.assertEquals(200, response.status_code)
def test_record_pushed_to_error_queue(self):
with patch('search.search_utils.requests.get') as mock_get:
mock_get.side_effect = ConnectionError
mommy.make(Facility, name='Ile Noma')
self.assertEquals(1, ErrorQueue.objects.count())
def test_unique_record_in_error_queue(self):
with patch('search.search_utils.requests.get') as mock_get:
mock_get.side_effect = ConnectionError
facility = mommy.make(Facility, name='Ile Noma')
self.assertEquals(1, ErrorQueue.objects.count())
facility.name = 'Ile noma sana'
facility.save()
self.assertEquals(1, ErrorQueue.objects.count())
def test_retrying_indexing_elastic_search_off(self):
with patch('search.search_utils.requests.get') as mock_get:
mock_get.side_effect = ConnectionError
mommy.make(Facility, name='Ile Noma')
call_command('retry_indexing')
# elastic search down hence no records updated
self.assertEquals(1, ErrorQueue.objects.count())
def test_retrying_indexing_elastic_search_on(self):
mommy.make(Facility, name='Ile Noma')
self.assertEquals(0, ErrorQueue.objects.count())
call_command('retry_indexing')
# elastic search up hence records updated
self.assertEquals(0, ErrorQueue.objects.count())
def test_retrying_indexing_object_already_deleted(self):
facility = mommy.make(Facility, name='Ile Noma')
facility.delete()
ErrorQueue.objects.create(
object_pk=str(facility.pk),
error_type='SEARCH_INDEXING_ERROR',
app_label='facilities',
model_name='Facility')
call_command('retry_indexing')
self.assertEquals(1, ErrorQueue.objects.count())
def test_retry_indexing_objects_in_queue(self):
with patch('search.search_utils.requests.get') as mock_get:
mock_get.side_effect = ConnectionError
mommy.make(Facility, name='Ile Noma')
self.assertEquals(1, ErrorQueue.objects.count())
# elastic search up hence records updated
call_command('retry_indexing')
self.assertEquals(0, ErrorQueue.objects.count())
def test_index_non_indexable_model(self):
job_title = mommy.make(JobTitle)
self.assertFalse(
index_instance('users', 'JobTitle', job_title.id)
)
def test_sending_email_to_admins(self):
"""
This is for test coverage only.
Tests that emails are sent to admins if the number
of retries on a record is more than one
"""
with patch('search.search_utils.requests.get') as mock_get:
mock_get.side_effect = ConnectionError
mommy.make(Facility, name='Ile Noma')
self.assertEquals(1, ErrorQueue.objects.count())
# elastic search up hence records updated
call_command('retry_indexing')
self.assertEquals(1, ErrorQueue.objects.count())
error_queue_object = ErrorQueue.objects.all()[0]
self.assertEquals(1, error_queue_object.retries)
call_command('retry_indexing')
self.assertEquals(1, ErrorQueue.objects.count())
error_queue_object = ErrorQueue.objects.all()[0]
self.assertEquals(2, error_queue_object.retries)
call_command('retry_indexing')
self.assertEquals(1, ErrorQueue.objects.count())
error_queue_object = ErrorQueue.objects.all()[0]
self.assertEquals(3, error_queue_object.retries)
def test_search_using_facility_name_when_elastic_search_is_off(self):
with patch('search.search_utils.requests.get') as mock_get:
mock_get.side_effect = ConnectionError
mommy.make(Facility, name='Ile Noma')
url = reverse('api:facilities:facilities_list')
url = url + '?search=noma'
response = self.client.get(url)
self.assertEquals(200, response.status_code)
def test_search_using_facility_code_when_elastic_search_is_off(self):
with patch('search.search_utils.requests.get') as mock_get:
mock_get.side_effect = ConnectionError
mommy.make(Facility, name='Ile Noma', code=10000)
url = reverse('api:facilities:facilities_list')
url = url + '?search=10000'
response = self.client.get(url)
self.assertEquals(200, response.status_code)
def tearDown(self):
self.elastic_search_api.delete_index(index_name='test_index')
super(TestSearchFunctions, self).tearDown()
class TestSearchFunctions(ViewTestBase):
def setUp(self):
self.elastic_search_api = ElasticAPI()
self.elastic_search_api.setup_index(index_name='test_index')
super(TestSearchFunctions, self).setUp()
def test_serialize_model(self):
self.maxDiff = None
facility = mommy.make(Facility)
serialized_data = serialize_model(facility)
expected_data = FacilitySerializer(facility).data
expected_data = json.dumps(expected_data, default=default)
self.assertEquals(expected_data, serialized_data.get('data'))
def test_serialize_model_serializer_not_found(self):
# There is no serializer named GroupSerializer
# Therefore the serialize model function should return None
group = mommy.make(Group)
serialized_data = serialize_model(group)
self.assertIsNone(serialized_data)
def test_default_json_dumps_function(self):
facility = mommy.make(Facility)
data = FacilitySerializer(facility).data
result = json.dumps(data, default=default)
self.assertIsInstance(result, str)
def test_search_facility(self):
url = reverse('api:facilities:facilities_list')
self.elastic_search_api = ElasticAPI()
self.elastic_search_api.setup_index(index_name='test_index')
facility = mommy.make(Facility, name='Kanyakini')
index_instance('facilities', 'Facility', str(facility.id),
'test_index')
url = url + "?search={}".format('Kanyakini')
response = self.client.get(url)
self.assertEquals(200, response.status_code)
self.assertIsInstance(response.data.get('results'), list)
self.elastic_search_api.delete_index('test_index')
def test_search_facility_using_material_view(self):
self.elastic_search_api = ElasticAPI()
self.elastic_search_api.setup_index(index_name='test_index')
facility = mommy.make(Facility, name='Facility ya kutest material')
url = reverse('api:facilities:material')
index_instance('facilities', 'Facility', str(facility.id),
'test_index')
url = url + "?search={}".format('material')
response = self.client.get(url)
self.assertEquals(200, response.status_code)
self.assertIsInstance(response.data.get('results'), list)
self.elastic_search_api.delete_index('test_index')
def test_search_facility_using_query_dsl(self):
url = reverse('api:facilities:facilities_list')
self.elastic_search_api = ElasticAPI()
self.elastic_search_api.setup_index(index_name='test_index')
facility = mommy.make(Facility, name='Kanyakini')
index_instance('facilities', 'Facility', str(facility.id),
'test_index')
query_dsl = {
"from": 0,
"size": 30,
"query": {
"query_string": {
"default_field": "name",
"query": "Kanyakini"
}
}
}
url = url + "?search={}".format(json.dumps(query_dsl))
response = self.client.get(url)
self.assertEquals(200, response.status_code)
self.assertIsInstance(response.data.get('results'), list)
self.elastic_search_api.delete_index('test_index')
def test_seach_auto_complete(self):
url = reverse('api:facilities:facilities_list')
self.elastic_search_api = ElasticAPI()
self.elastic_search_api.setup_index(index_name='test_index')
facility = mommy.make(Facility, name='Kanyakini')
index_instance('facilities', 'Facility', str(facility.id),
'test_index')
url = url + "?search_auto={}".format('Kanya')
response = self.client.get(url)
self.assertEquals(200, response.status_code)
self.assertIsInstance(response.data.get('results'), list)
self.elastic_search_api.delete_index('test_index')
def test_search_facility_multiple_filters(self):
url = reverse('api:facilities:facilities_list')
self.elastic_search_api = ElasticAPI()
self.elastic_search_api.setup_index(index_name='test_index')
facility = mommy.make(Facility, name='Mordal mountains medical clinic')
mommy.make(FacilityApproval, facility=facility)
facility.is_published = True
facility.save()
facility_2 = mommy.make(Facility,
name='Eye of mordal health center',
is_published=False)
index_instance('facilities', 'Facility', str(facility.id),
'test_index')
index_instance('facilities', 'Facility', str(facility_2.id),
'test_index')
url = url + "?search={}&is_published={}".format('mordal', 'false')
response = self.client.get(url)
self.assertEquals(200, response.status_code)
self.assertIsInstance(response.data.get('results'), list)
self.elastic_search_api.delete_index('test_index')
def test_get_search_auto_complete_fields(self):
search_fields = self.elastic_search_api.get_search_fields('facility')
self.assertIsInstance(search_fields, list)
self.assertEquals(len(search_fields), 10)
def test_get_search_auto_complete_fields_si_empty(self):
search_fields = self.elastic_search_api.get_search_fields('no_model')
self.assertIsNone(search_fields)
def test_user_search(self):
user = mommy.make(get_user_model(), is_national=True)
self.client.force_authenticate(user)
county_user = mommy.make(get_user_model())
sub_county_user = mommy.make(get_user_model(),
first_name='kijana',
last_name='mzee')
index_instance('users', 'MflUser', str(sub_county_user.id),
'test_index')
url = reverse("api:users:mfl_users_list")
url = url + "?search=kijana"
response = self.client.get(url)
self.assertEquals(200, response.status_code)
# the national users are not allowed to see the sub county level users
county = mommy.make(County)
const = mommy.make(Constituency, county=county)
mommy.make(UserCounty, county=county, user=county_user)
mommy.make(UserConstituency,
user=sub_county_user,
constituency=const,
created_by=county_user,
updated_by=county_user)
filtered_response = self.client.get(url)
self.assertEquals(200, filtered_response.status_code)
def test_chu_search(self):
facility = mommy.make(Facility)
facility_2 = mommy.make(Facility)
chu_url = reverse("api:chul:community_health_units_list")
chu = mommy.make(CommunityHealthUnit,
facility=facility,
name='Jericho')
mommy.make(CommunityHealthUnit, facility=facility_2)
index_instance('chul', 'CommunityHealthUnit', str(chu.id),
'test_index')
chu_search_url = chu_url + "?search=Jericho"
response = self.client.get(chu_search_url)
self.assertEquals(200, response.status_code)
def test_chu_search_using_code(self):
facility = mommy.make(Facility)
facility_2 = mommy.make(Facility)
chu_url = reverse("api:chul:community_health_units_list")
chu = mommy.make(CommunityHealthUnit,
facility=facility,
name='Jericho')
mommy.make(CommunityHealthUnit, facility=facility_2)
index_instance('chul', 'CommunityHealthUnit', str(chu.id),
'test_index')
chu_search_url = chu_url + "?search={}".format(chu.code)
response = self.client.get(chu_search_url)
self.assertEquals(200, response.status_code)
def test_record_pushed_to_error_queue(self):
with patch('search.search_utils.requests.get') as mock_get:
mock_get.side_effect = ConnectionError
mommy.make(Facility, name='Ile Noma')
self.assertEquals(1, ErrorQueue.objects.count())
def test_unique_record_in_error_queue(self):
with patch('search.search_utils.requests.get') as mock_get:
mock_get.side_effect = ConnectionError
facility = mommy.make(Facility, name='Ile Noma')
self.assertEquals(1, ErrorQueue.objects.count())
facility.name = 'Ile noma sana'
facility.save()
self.assertEquals(1, ErrorQueue.objects.count())
def test_retrying_indexing_elastic_search_off(self):
with patch('search.search_utils.requests.get') as mock_get:
mock_get.side_effect = ConnectionError
mommy.make(Facility, name='Ile Noma')
call_command('retry_indexing')
# elastic search down hence no records updated
self.assertEquals(1, ErrorQueue.objects.count())
def test_retrying_indexing_elastic_search_on(self):
mommy.make(Facility, name='Ile Noma')
self.assertEquals(0, ErrorQueue.objects.count())
call_command('retry_indexing')
# elastic search up hence records updated
self.assertEquals(0, ErrorQueue.objects.count())
def test_retrying_indexing_object_already_deleted(self):
facility = mommy.make(Facility, name='Ile Noma')
facility.delete()
ErrorQueue.objects.create(object_pk=str(facility.pk),
error_type='SEARCH_INDEXING_ERROR',
app_label='facilities',
model_name='Facility')
call_command('retry_indexing')
self.assertEquals(1, ErrorQueue.objects.count())
def test_retry_indexing_objects_in_queue(self):
with patch('search.search_utils.requests.get') as mock_get:
mock_get.side_effect = ConnectionError
mommy.make(Facility, name='Ile Noma')
self.assertEquals(1, ErrorQueue.objects.count())
# elastic search up hence records updated
call_command('retry_indexing')
self.assertEquals(0, ErrorQueue.objects.count())
def test_index_non_indexable_model(self):
job_title = mommy.make(JobTitle)
self.assertFalse(index_instance('users', 'JobTitle', job_title.id))
def test_sending_email_to_admins(self):
"""
This is for test coverage only.
Tests that emails are sent to admins if the number
of retries on a record is more than one
"""
with patch('search.search_utils.requests.get') as mock_get:
mock_get.side_effect = ConnectionError
mommy.make(Facility, name='Ile Noma')
self.assertEquals(1, ErrorQueue.objects.count())
# elastic search up hence records updated
call_command('retry_indexing')
self.assertEquals(1, ErrorQueue.objects.count())
error_queue_object = ErrorQueue.objects.all()[0]
self.assertEquals(1, error_queue_object.retries)
call_command('retry_indexing')
self.assertEquals(1, ErrorQueue.objects.count())
error_queue_object = ErrorQueue.objects.all()[0]
self.assertEquals(2, error_queue_object.retries)
call_command('retry_indexing')
self.assertEquals(1, ErrorQueue.objects.count())
error_queue_object = ErrorQueue.objects.all()[0]
self.assertEquals(3, error_queue_object.retries)
def test_search_using_facility_name_when_elastic_search_is_off(self):
with patch('search.search_utils.requests.get') as mock_get:
mock_get.side_effect = ConnectionError
mommy.make(Facility, name='Ile Noma')
url = reverse('api:facilities:facilities_list')
url = url + '?search=noma'
response = self.client.get(url)
self.assertEquals(200, response.status_code)
def test_search_using_facility_code_when_elastic_search_is_off(self):
with patch('search.search_utils.requests.get') as mock_get:
mock_get.side_effect = ConnectionError
mommy.make(Facility, name='Ile Noma', code=10000)
url = reverse('api:facilities:facilities_list')
url = url + '?search=10000'
response = self.client.get(url)
self.assertEquals(200, response.status_code)
def tearDown(self):
self.elastic_search_api.delete_index(index_name='test_index')
super(TestSearchFunctions, self).tearDown()