def test_get_raw_sites_evaluated(self): allowed_cls = AllowedSites(defaults=['yay.com']) with self.assertNumQueries(1): data = tuple(allowed_cls.get_raw_sites()) self.assertEqual(len(data), 2) # do it again to demonstrate it uses iterator() with self.assertNumQueries(1): data = tuple(allowed_cls.get_raw_sites()) self.assertEqual(len(data), 2)
def test_get_raw_sites_unevaluated(self): allowed_cls = AllowedSites(defaults=['yay.com']) with self.assertNumQueries(0): allowed_cls.get_raw_sites()