def test_get_domains(self):
allowed_cls = AllowedSites(defaults=['yay.com'])
with self.assertNumQueries(1):
data = allowed_cls.get_domains()
self.assertEqual(data, frozenset(['example.com', 'example.org']))
def test_get_domains_with_public_ip(self):
allowed_cls = AllowedSites(defaults=['yay.com'],
dynamic_public_ip=True)
with self.assertNumQueries(1):
data = allowed_cls.get_domains()
self.assertEqual(len(data), 3)
