def test_get_short_list_trivial_works(self): # setup view = HomeView() with self.assertNumQueries(1): short_list = list(view.get_short_list()) self.assertFalse(short_list)
def test_get_short_list_returns_enrollments(self): # setup view = HomeView() institution = InstitutionFactory() EnrollmentFactory(institution=institution, year=LATEST_ENROLLMENT_YEAR) with self.assertNumQueries(1): short_list = list(view.get_short_list()) self.assertEqual(short_list[0].institution, institution)