def test_get_student_results(self):
backend = haystack.load_backend(HAYSTACK_SEARCH_ENGINE_TEST)
with self._whoosh_settings():
with patch('haystack.backend', backend):
periodbuilder = PeriodBuilder.quickadd_ducku_duck1010_active()
group1 = periodbuilder.add_assignment('assignment1', long_name="Assignment One")\
.add_group().add_students(self.testuser).group
group2 = periodbuilder.add_assignment('assignment2', long_name="Assignment Two")\
.add_group().add_students(self.testuser).group
group_not_student = periodbuilder.add_assignment('assignment3', long_name="Assignment Three")\
.add_group().add_students(UserBuilder('nobody').user).group
self._rebuild_searchindex()
results = SearchHelper(self.testuser, 'Assignment').get_student_results()
self.assertEquals(results.count(), 2)
self.assertEquals(set(map(lambda r: r.object, results)), set([group1, group2]))
def test_get_examiner_results(self):
with self._whoosh_settings():
with patch('haystack.backend', haystack.load_backend(HAYSTACK_SEARCH_ENGINE_TEST)):
periodbuilder = PeriodBuilder.quickadd_ducku_duck1010_active()
assignment1builder = periodbuilder.add_assignment('assignment1', long_name="Assignment One")
assignment2builder = periodbuilder.add_assignment('assignment2', long_name="Assignment Two")
group1 = assignment1builder.add_group().add_examiners(self.testuser).group
group2 = assignment1builder.add_group().add_examiners(self.testuser).group
group3 = assignment2builder.add_group().add_examiners(self.testuser).group
assignment2builder.add_group().add_examiners(UserBuilder('nobody').user) # Should not math
self._rebuild_searchindex()
results = SearchHelper(self.testuser, 'Assignment').get_examiner_results()
self.assertEquals(results.count(), 5)
self.assertEquals(set(map(lambda r: r.object, results)),
set([group1, group2, group3,
assignment1builder.assignment, assignment2builder.assignment]))
