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)