def test_facets(self): queryset = TestModel.es.all() filter_backend = ElasticsearchFilterBackend() s = filter_backend.filter_queryset(self.fake_request, queryset, self.fake_view) expected = [{u'doc_count': 1, u'key': u'test'}] self.assertEqual(s.facets['doc_count'], 3) self.assertEqual(s.facets['first_name']['buckets'], expected)
def test_facets(self): queryset = TestModel.es.all() filter_backend = ElasticsearchFilterBackend() s = filter_backend.filter_queryset(self.fake_request, queryset, self.fake_view) expected = [{u'doc_count': 1, u'key': u'test'}] self.assertEqual(s.facets['doc_count'], 3) self.assertEqual(s.facets['first_name']['buckets'], expected)
def test_filter_backend_no_list(self): queryset = TestModel.es.all() filter_backend = ElasticsearchFilterBackend() self.fake_view.action = 'create' queryset = filter_backend.filter_queryset(self.fake_request, queryset, self.fake_view) # the 'normal' dataflow continues self.assertTrue(isinstance(queryset, QuerySet)) self.fake_view.action = 'list'
def test_filter_backend_no_list(self): queryset = TestModel.es.all() filter_backend = ElasticsearchFilterBackend() self.fake_view.action = 'create' queryset = filter_backend.filter_queryset(self.fake_request, queryset, self.fake_view) # the 'normal' dataflow continues self.assertTrue(isinstance(queryset, QuerySet)) self.fake_view.action = 'list'
def test_filter_backend_ordering(self): queryset = TestModel.es.all() filter_backend = ElasticsearchFilterBackend() self.fake_view.ordering = ('-username',) queryset = filter_backend.filter_queryset(self.fake_request, queryset, self.fake_view).deserialize() self.assertEqual(queryset[0].id, self.model2.id) self.assertEqual(queryset[1].id, self.model1.id) del self.fake_view.ordering
def _test_filter_backend(self): queryset = TestModel.es.all() filter_backend = ElasticsearchFilterBackend() queryset = filter_backend.filter_queryset(self.fake_request, queryset, self.fake_view) l = queryset.deserialize() self.assertTrue(self.model1 in l) self.assertTrue(self.model2 in l) self.assertFalse(self.model3 in l)
def test_filter_backend_ordering(self): queryset = TestModel.es.all() filter_backend = ElasticsearchFilterBackend() self.fake_view.ordering = ('-username',) queryset = filter_backend.filter_queryset(self.fake_request, queryset, self.fake_view).deserialize() self.assertEqual(queryset[0].id, self.model2.id) self.assertEqual(queryset[1].id, self.model1.id) del self.fake_view.ordering
def _test_filter_backend(self): queryset = TestModel.es.all() filter_backend = ElasticsearchFilterBackend() queryset = filter_backend.filter_queryset(self.fake_request, queryset, self.fake_view) x = queryset.deserialize() self.assertTrue(self.model1 in x) self.assertTrue(self.model2 in x) self.assertFalse(self.model3 in x)
def test_filter_backend_on_normal_model(self): filter_backend = ElasticsearchFilterBackend() with self.assertRaises(ValueError): filter_backend.filter_queryset(self.fake_request, User.objects.all(), self.fake_view)
def test_filter_backend_on_normal_model(self): filter_backend = ElasticsearchFilterBackend() with self.assertRaises(ValueError): filter_backend.filter_queryset(self.fake_request, User.objects.all(), self.fake_view)