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()
