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_iterable(self):
"""
this is what Django does internally in django.http.request
allowed_cls is synonymous with settings.ALLOWED_HOSTS
"""
allowed_cls = AllowedSites(defaults=['yay.com'])
with self.assertNumQueries(1):
self.assertTrue(validate_host('example.com', allowed_cls))
with self.assertNumQueries(1):
self.assertTrue(validate_host('example.org', allowed_cls))
with self.assertNumQueries(1):
self.assertFalse(validate_host('djangoproject.com', allowed_cls))
# ideally this should be 0 queries, because it's a default ...
with self.assertNumQueries(1):
self.assertTrue(validate_host('yay.com', allowed_cls))
def test_inequality(self):
allowed_cls = AllowedSites()
allowed_cls2 = AllowedSites(defaults='test.com')
self.assertNotEqual(allowed_cls, allowed_cls2)
def test_equality(self):
allowed_cls = AllowedSites()
allowed_cls2 = AllowedSites()
self.assertEqual(allowed_cls, allowed_cls2)
def test_bool_false(self):
allowed_cls = AllowedSites()
Site.objects.all().delete()
self.assertFalse(allowed_cls)
def test_bool_true(self):
allowed_cls = AllowedSites()
self.assertTrue(allowed_cls)
def test_containment(self):
allowed_cls = AllowedSites(defaults=['yay.com'])
self.assertIn('example.org', allowed_cls)
self.assertNotIn('djangoproject.org', allowed_cls)
def test_length(self):
allowed_cls = AllowedSites(defaults=['yay.com'])
self.assertEqual(len(allowed_cls), 3)
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_raw_sites_unevaluated(self):
allowed_cls = AllowedSites(defaults=['yay.com'])
with self.assertNumQueries(0):
allowed_cls.get_raw_sites()
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)
