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)
