def test_cidr_net(self):
"""Some nets are set in settings.py; Try them."""
mw = middleware.AllowCIDRMiddleware()
self.assertEqual(settings.ALLOWED_HOSTS, ['*'])
req = self.rf.get('/', HTTP_HOST='192.168.1.11')
self.assertIsNone(mw.process_request(req))
req = self.rf.get('/', HTTP_HOST='192.168.2.200')
self.assertIsNone(mw.process_request(req))
req = self.rf.get('/', HTTP_HOST='192.168.2.200:8000')
self.assertIsNone(mw.process_request(req))
with self.assertRaises(DisallowedHost):
req = self.rf.get('/', HTTP_HOST='192.168.3.200')
mw.process_request(req)
def test_other_hosts(self):
"""The hosts defined in ALLOWED_HOSTS should still work."""
mw = middleware.AllowCIDRMiddleware()
self.assertEqual(settings.ALLOWED_HOSTS, ['*'])
req = self.rf.get('/', HTTP_HOST='thedude.abides.com')
self.assertIsNone(mw.process_request(req))
req = self.rf.get('/', HTTP_HOST='thebig.lebowski.biz')
self.assertIsNone(mw.process_request(req))
req = self.rf.get('/', HTTP_HOST='thebig.lebowski.biz:8000')
self.assertIsNone(mw.process_request(req))
with self.assertRaises(DisallowedHost):
req = self.rf.get('/', HTTP_HOST='donnie.net')
mw.process_request(req)
def test_hosts_star_disable(self):
"""The middleware should deactivate if ALLOWED_HOSTS is '*'"""
with self.assertRaises(MiddlewareNotUsed):
middleware.AllowCIDRMiddleware()
self.assertEqual(settings.ALLOWED_HOSTS, ['*'])
def test_other_hosts_no_nets(self):
"""The hosts defined in ALLOWED_HOSTS should work when middleware is off."""
with self.assertRaises(MiddlewareNotUsed):
middleware.AllowCIDRMiddleware()
self.assertEqual(settings.ALLOWED_HOSTS, ['thedude.abides.com', '.lebowski.biz'])
def _get_middleware(self):
if StrictVersion(django.get_version()) < StrictVersion('1.10'):
return middleware.AllowCIDRMiddleware().process_request
else:
return middleware.AllowCIDRMiddleware(get_response=lambda x: None)