class SearchQuerySetWrapperTests(TestCase): def setUp(self): super(SearchQuerySetWrapperTests, self).setUp() title = 'Some random course' query = 'title:' + title CourseRunFactory.create_batch(3, title=title) self.search_queryset = SearchQuerySet().models(CourseRun).raw_search(query).load_all() self.course_runs = [e.object for e in self.search_queryset] self.wrapper = SearchQuerySetWrapper(self.search_queryset) def test_count(self): self.assertEqual(self.search_queryset.count(), self.wrapper.count()) def test_iter(self): self.assertEqual([e for e in self.course_runs], [e for e in self.wrapper]) def test_getitem(self): self.assertEqual(self.course_runs[0], self.wrapper[0])
class SearchQuerySetWrapperTests(TestCase): def setUp(self): super().setUp() title = 'Some random course' query = 'title:' + title CourseRunFactory.create_batch(3, title=title) self.search_queryset = CourseRunDocument().search().query( 'query_string', query=query) self.course_runs = [e.object for e in self.search_queryset] self.wrapper = SearchQuerySetWrapper(self.search_queryset) def test_count(self): self.assertEqual(self.search_queryset.count(), self.wrapper.count()) def test_iter(self): self.assertEqual(self.course_runs, list(self.wrapper)) def test_getitem(self): self.assertEqual(self.course_runs[0], self.wrapper[0])
class SearchQuerySetWrapperTests(TestCase): def setUp(self): super(SearchQuerySetWrapperTests, self).setUp() title = 'Some random course' query = 'title:' + title CourseRunFactory.create_batch(3, title=title) self.search_queryset = SearchQuerySet().models(CourseRun).raw_search( query).load_all() self.course_runs = [e.object for e in self.search_queryset] self.wrapper = SearchQuerySetWrapper(self.search_queryset) def test_count(self): self.assertEqual(self.search_queryset.count(), self.wrapper.count()) def test_iter(self): self.assertEqual(self.course_runs, list(self.wrapper)) def test_getitem(self): self.assertEqual(self.course_runs[0], self.wrapper[0])