def setUp(self):
super(SimpleSearchBackendTestCase, self).setUp()
self.site = SearchSite()
self.backend = SearchBackend(site=self.site)
self.index = SimpleMockSearchIndex(MockModel, backend=self.backend)
self.site.register(MockModel, SimpleMockSearchIndex)
self.sample_objs = MockModel.objects.all()
self.backend.update(self.index, self.sample_objs)
class SimpleSearchBackendTestCase(TestCase):
fixtures = ['bulk_data.json']
def setUp(self):
super(SimpleSearchBackendTestCase, self).setUp()
self.site = SearchSite()
self.backend = SearchBackend(site=self.site)
self.index = SimpleMockSearchIndex(MockModel, backend=self.backend)
self.site.register(MockModel, SimpleMockSearchIndex)
self.sample_objs = MockModel.objects.all()
self.backend.update(self.index, self.sample_objs)
def test_update(self):
self.backend.update(self.index, self.sample_objs)
self.assertEqual(len(self.sample_objs), SearchableObject.objects.count())
def test_remove(self):
self.backend.remove(self.sample_objs[0])
self.assertEqual(len(self.sample_objs)-1, SearchableObject.objects.count())
def test_clear(self):
self.backend.clear()
self.assertEqual(0, SearchableObject.objects.count())
def test_search(self):
# No query string should always yield zero results.
self.assertEqual(self.backend.search(None), {'hits': 0, 'results': []})
EMPTY = MultiValueDict()
DANIEL = MultiValueDict({'content':['daniel1']})
STRING = MultiValueDict({'content':['should be a string']})
INDX = MultiValueDict({'content':['Indx']})
search_results = self.backend.search(EMPTY)
self.assertEqual(search_results['hits'], 23)
self.assertEqual([result.pk for result in search_results['results']], [1, 2, 3, 4, 5, 6, 7, 8, 9, 10, 11, 12, 13, 14, 15, 16, 17, 18, 19, 20, 21, 22, 23])
search_results = self.backend.search(DANIEL)
#print SearchableObject.objects.filter(search_text__icontains='daniel')
self.assertEqual(search_results['hits'], 7)
self.assertEqual([result.pk for result in search_results['results']], [1, 5, 18, 6, 11, 7, 9])
search_results = self.backend.search(STRING)
self.assertEqual(search_results['hits'], 1)
self.assertEqual([result.pk for result in search_results['results']], [8])
# Ensure the results are ``SearchResult`` instances...
self.assertEqual(search_results['results'][0].score, 0)
'''
#Ands are not supported currently
search_results = self.backend.search(u'[["index", "document"]]', debug=True)
self.assertEqual(search_results['hits'], 6)
self.assertEqual([result.pk for result in search_results['results']], [2, 3, 15, 16, 17, 18])
# Regression-ville
self.assertEqual([result.object.id for result in search_results['results']], [2, 3, 15, 16, 17, 18])
self.assertEqual(search_results['results'][0].model, MockModel)
'''
# No support for spelling suggestions
self.assertEqual(self.backend.search(INDX)['hits'], 0)
self.assertFalse(self.backend.search(INDX).get('spelling_suggestion'))
# No support for facets
#self.assertEqual(self.backend.search(MultiValueDict(), facets=['name']), {'hits': 0, 'results': []})
self.assertEqual(self.backend.search(DANIEL, facets=['name'])['hits'], 7)
self.assertEqual(self.backend.search(EMPTY, date_facets={'pub_date': {'start_date': date(2008, 2, 26), 'end_date': date(2008, 2, 26), 'gap': '/MONTH'}}), {'hits': 0, 'results': []})
self.assertEqual(self.backend.search(DANIEL, date_facets={'pub_date': {'start_date': date(2008, 2, 26), 'end_date': date(2008, 2, 26), 'gap': '/MONTH'}})['hits'], 7)
self.assertEqual(self.backend.search(EMPTY, query_facets={'name': '[* TO e]'}), {'hits': 0, 'results': []})
self.assertEqual(self.backend.search(DANIEL, query_facets={'name': '[* TO e]'})['hits'], 7)
self.assertFalse(self.backend.search(EMPTY).get('facets'))
self.assertFalse(self.backend.search(DANIEL).get('facets'))
# Contextual fields are lumped in
#self.assertEqual(self.backend.search(u'["2009-06-18"]', debug=True)['hits'], 2)
def test_more_like_this(self):
self.assertEqual(self.backend.search(MultiValueDict())['hits'], 23)
# Unsupported by 'dummy'. Should see empty results.
self.assertEqual(self.backend.more_like_this(self.sample_objs[0])['hits'], 0)
def setUp(self):
super(SimpleSearchQueryTestCase, self).setUp()
self.sq = SearchQuery(backend=SearchBackend())
