class DistillerySearchResultsListTestCase(TestCase):
fixtures = get_fixtures(['distilleries', 'users'])
user_model = get_user_model()
@staticmethod
def _get_instance(query):
"""
Parameters
----------
query : SearchQuery
Returns
-------
DistillerySearchResultsList
"""
with MOCK_FIND:
return DistillerySearchResultsList(query)
def setUp(self):
self.user = self.user_model.objects.get(pk=1)
def test_distillery_filter(self):
"""
Tests that the distilleries from a distillery filter are used for
getting distillery results.
"""
search_query = SearchQuery('@source="test_posts" test', self.user)
distillery_results_list = self._get_instance(search_query)
self.assertEqual(len(distillery_results_list.distilleries), 1)
self.assertEqual(distillery_results_list.distilleries[0].pk, 1)
self.assertEqual(len(distillery_results_list.results), 1)
self.assertEqual(distillery_results_list.results[0].distillery.pk, 1)
def test_result_count(self):
"""
Tests that the result list returns a combined count of all
DistillerySearchResults
"""
search_query = SearchQuery('test', self.user)
distillery_results_list = self._get_instance(search_query)
self.assertEqual(distillery_results_list.count, 6)
def test_result_length(self):
"""
Tests that an accurate amount of DistillerySearchResults is created
for the number of distilleries returned.
"""
distillery_results_list = self._get_instance(
SearchQuery('test', self.user))
self.assertEqual(len(distillery_results_list.results), 6)
def test_as_dict(self):
"""
Tests that the as_dict() function returns the correct
dictionary shape.
"""
search_query = SearchQuery('@source="test_posts" test', self.user)
distillery_results_list = self._get_instance(search_query)
factory = RequestFactory()
request = factory.get('/api/v1/search/')
self.assertEqual(
distillery_results_list.as_dict(request), {
'count':
1,
'results': [{
'count': 1,
'next': None,
'previous': None,
'results': MOCK_RESULTS_LIST,
'distillery': {
'id': 1,
'name': 'mongodb.test_database.test_posts',
'url': 'http://testserver/api/v1/distilleries/1/',
}
}]
})
class DistillerySearchResultsTestCase(TestCase):
fixtures = get_fixtures(['distilleries', 'users'])
user_model = get_user_model()
@staticmethod
def _get_instance(query, distillery, after=None, before=None):
"""
Parameters
----------
query : SearchQuery
distillery : Distillery
Returns
-------
DistillerySearchResults
"""
with MOCK_FIND:
return DistillerySearchResults(query,
distillery,
after=after,
before=before)
def setUp(self):
self.user = self.user_model.objects.get(pk=1)
def test_distillery_attribute_set(self):
"""
Makes sure that the distillery attribute is set to the passed in
distillery parameter.
"""
distillery = Distillery.objects.get(pk=1)
search_query = SearchQuery('test', self.user)
distillery_results = self._get_instance(search_query, distillery)
self.assertEqual(distillery, distillery_results.distillery)
def test_no_fieldsets(self):
"""
Tests that no fieldsets are created for an instance with
no keywords or parameters.
"""
distillery = Distillery.objects.get(pk=1)
search_query = SearchQuery('', self.user)
distillery_results = self._get_instance(search_query, distillery)
self.assertIsNone(distillery_results.engine_query)
def test_field_fieldsets_if_related(self):
"""
Tests that a list of QueryFieldsets are created when given a list of
FieldSearchParameters if the parameter is related to the distillery.
"""
distillery = Distillery.objects.get(pk=3)
search_query = SearchQuery('ip_address=13.43', self.user)
distillery_results = self._get_instance(search_query, distillery)
fieldsets = get_fieldsets(distillery_results.engine_query.subqueries)
self.assertEqual(len(fieldsets), 1)
field_fieldset = fieldsets[0]
self.assertEqual(field_fieldset.field_name, 'ip_address')
self.assertEqual(field_fieldset.field_type, 'GenericIPAddressField')
def test_field_fieldsets_if_not_related(self):
"""
Tests that an engine query is not created if the
FieldSearchParameter is not related to the distillery
"""
distillery = Distillery.objects.get(pk=1)
search_query = SearchQuery('ip_address=13.43', self.user)
distillery_results = self._get_instance(search_query, distillery)
self.assertIsNone(distillery_results.engine_query)
def test_keyword_fieldsets(self):
"""
Tests that a list of QueryFieldsets is created for each text field
on a distillery.
"""
distillery = Distillery.objects.get(pk=5)
search_query = SearchQuery('test "more testing"', self.user)
distillery_results = self._get_instance(search_query, distillery)
self.assertIsNotNone(distillery_results.engine_query)
fieldsets = get_fieldsets(distillery_results.engine_query.subqueries)
self.assertEqual(fieldsets[0].field_name, 'host')
self.assertEqual(fieldsets[0].field_type, 'GenericIPAddressField')
self.assertEqual(fieldsets[0].operator, 'regex')
self.assertEqual(fieldsets[0].value, 'test')
self.assertEqual(fieldsets[1].field_name, 'message')
self.assertEqual(fieldsets[1].field_type, 'TextField')
self.assertEqual(fieldsets[1].operator, 'regex')
self.assertEqual(fieldsets[1].value, 'test')
self.assertEqual(fieldsets[2].field_name, 'host')
self.assertEqual(fieldsets[2].field_type, 'GenericIPAddressField')
self.assertEqual(fieldsets[2].operator, 'regex')
self.assertEqual(fieldsets[2].value, 'more testing')
self.assertEqual(fieldsets[3].field_name, 'message')
self.assertEqual(fieldsets[3].field_type, 'TextField')
self.assertEqual(fieldsets[3].operator, 'regex')
self.assertEqual(fieldsets[3].value, 'more testing')
def test_results(self):
"""
Tests that the results and result count from a distillery
search get put onto the results and count attribute
"""
distillery = Distillery.objects.get(pk=5)
search_query = SearchQuery('test "more testing"', self.user)
distillery_results = self._get_instance(search_query, distillery)
self.assertEqual(distillery_results.count, 1)
self.assertEqual(distillery_results.results, MOCK_RESULTS_LIST)
def test_time_search(self):
distillery = Distillery.objects.get(pk=6)
search_query = SearchQuery('body=test', self.user)
before_iso = '2017-11-27T06:00:00+05:00'
before_date = parser.parse(before_iso)
after_iso = '2017-11-27T06:00:00+05:00'
after_date = parser.parse(before_iso)
distillery_results = self._get_instance(search_query,
distillery,
before=before_date,
after=after_date)
fieldsets = get_fieldsets(distillery_results.engine_query.subqueries)
self.assertEqual(len(fieldsets), 3)
self.assertEqual(fieldsets[1].field_name, 'date')
self.assertEqual(fieldsets[1].field_type, 'DateTimeField')
self.assertEqual(fieldsets[1].operator, 'lte')
self.assertEqual(fieldsets[1].value, before_iso)
self.assertEqual(fieldsets[2].field_name, 'date')
self.assertEqual(fieldsets[2].field_type, 'DateTimeField')
self.assertEqual(fieldsets[2].operator, 'gte')
self.assertEqual(fieldsets[2].value, after_iso)
def test_as_dict(self):
"""
Tests that the correct dictionary shape is returned from the
as_dict() function.
"""
distillery = Distillery.objects.get(pk=1)
search_query = SearchQuery('test "more testing"', self.user)
distillery_results = self._get_instance(search_query, distillery)
factory = RequestFactory()
request = factory.get('/api/v1/search/')
self.assertDictEqual(
distillery_results.as_dict(request), {
'count': 1,
'results': MOCK_RESULTS_LIST,
'next': None,
'previous': None,
'distillery': {
'id': 1,
'name': 'mongodb.test_database.test_posts',
'url': 'http://testserver/api/v1/distilleries/1/',
}
})
class DistillerySearchResultsTestCase(TestCase):
fixtures = get_fixtures(['distilleries'])
@staticmethod
def _get_instance(query, distillery):
"""
Parameters
----------
query : SearchQuery
distillery : Distillery
Returns
-------
DistillerySearchResults
"""
with MOCK_FIND:
return DistillerySearchResults(query, distillery)
def test_distillery_attribute_set(self):
"""
Makes sure that the distillery attribute is set to the passed in
distillery parameter.
"""
distillery = Distillery.objects.get(pk=1)
search_query = SearchQuery('test')
distillery_results = self._get_instance(search_query, distillery)
self.assertEqual(distillery, distillery_results.distillery)
def test_no_fieldsets(self):
"""
Tests that no fieldsets are created for an instance with
no keywords or parameters.
"""
distillery = Distillery.objects.get(pk=1)
search_query = SearchQuery('')
distillery_results = self._get_instance(search_query, distillery)
self.assertEqual(distillery_results.fieldsets, [])
self.assertIsNone(distillery_results.engine_query)
def test_field_fieldsets_if_related(self):
"""
Tests that a list of QueryFieldsets are created when given a list of
FieldSearchParameters if the parameter is related to the distillery.
"""
distillery = Distillery.objects.get(pk=3)
search_query = SearchQuery('ip_address=13.43')
distillery_results = self._get_instance(search_query, distillery)
self.assertEqual(len(distillery_results.fieldsets), 1)
field_fieldset = distillery_results.fieldsets[0]
self.assertEqual(field_fieldset.field_name, 'ip_address')
self.assertEqual(field_fieldset.field_type, 'GenericIPAddressField')
def test_field_fieldsets_if_not_related(self):
"""
Tests that a list of QueryFieldsets are not created if the
FieldSearchParameter is not related to the distillery
"""
distillery = Distillery.objects.get(pk=1)
search_query = SearchQuery('ip_address=13.43')
distillery_results = self._get_instance(search_query, distillery)
self.assertEqual(len(distillery_results.fieldsets), 0)
def test_keyword_fieldsets(self):
"""
Tests that a list of QueryFieldsets is created for each text field
on a distillery.
"""
distillery = Distillery.objects.get(pk=5)
search_query = SearchQuery('test "more testing"')
distillery_results = self._get_instance(search_query, distillery)
self.assertEqual(len(distillery_results.fieldsets), 2)
fieldset_1 = distillery_results.fieldsets[0]
fieldset_2 = distillery_results.fieldsets[1]
self.assertEqual(fieldset_1.field_name, 'host')
self.assertEqual(fieldset_1.field_type, 'GenericIPAddressField')
self.assertEqual(fieldset_1.operator, 'regex')
self.assertEqual(fieldset_1.value, 'test|more testing')
self.assertEqual(fieldset_2.field_name, 'message')
self.assertEqual(fieldset_2.field_type, 'TextField')
self.assertEqual(fieldset_2.operator, 'regex')
self.assertEqual(fieldset_2.value, 'test|more testing')
def test_results(self):
"""
Tests that the results and result count from a distillery
search get put onto the results and count attribute
"""
distillery = Distillery.objects.get(pk=5)
search_query = SearchQuery('test "more testing"')
distillery_results = self._get_instance(search_query, distillery)
self.assertEqual(distillery_results.count, 1)
self.assertEqual(distillery_results.results, MOCK_RESULTS_LIST)
def test_as_dict(self):
"""
Tests that the correct dictionary shape is returned from the
as_dict() function.
"""
distillery = Distillery.objects.get(pk=1)
search_query = SearchQuery('test "more testing"')
distillery_results = self._get_instance(search_query, distillery)
factory = RequestFactory()
request = factory.get('/api/v1/search/')
self.assertDictEqual(
distillery_results.as_dict(request), {
'count': 1,
'results': MOCK_RESULTS_LIST,
'next': None,
'previous': None,
'distillery': {
'id': 1,
'name': 'mongodb.test_database.test_posts',
'url': 'http://testserver/api/v1/distilleries/1/',
}
})