def test_process_view_passes_curried_mithril_reset_view(self):
expected = fmt_ip(random.randint(1, 0xFFFFFFFF))
req = self.fake_request(expected)
strat = CachedStrategy()
wl = CachedWhitelist.objects.create(
name='okay',
slug='okay'
)
wl.range_set.create(
ip='0.0.0.0',
cidr=32
)
strat.return_one = lambda *a, **kw: wl.pk
strat.actions = [['return_one', 'pk']]
view = lambda:None
view.mithril_reset = \
lambda request, view, args, kwargs: check(
request, view, args, kwargs)
def check(request, target_view, target_args, target_kwargs):
self.assertTrue(request is req)
self.assertEqual(target_view, view)
return expected
self.assertEqual(
strat.process_view(req, view, (), {}),
expected
)
def test_super_user_exempt(self):
"""
Make sure that superuser is exempt from IP whitelists.
"""
expected = '10.112.12.12'
req = self.fake_request(expected)
req.user.is_superuser = False
whitelist = CachedWhitelist.objects.create(name='asdf', slug='asdf')
whitelist.range_set.create(ip='127.0.0.1', cidr=32)
strat = CachedStrategy()
strat.return_one = lambda *a, **kw: whitelist.pk
strat.actions = [['return_one', 'pk']]
resp = strat.process_view(req, None, (), {})
self.assertTrue(
isinstance(resp, HttpResponseForbidden)
)
req.user.is_superuser = True
self.assertEqual(
strat.process_view(req, None, (), {}),
None
)
def test_process_view_skips_lookups_raising_fielderror(self):
expected = fmt_ip(random.randint(1, 0xFFFFFFFF))
req = self.fake_request(expected)
strat = CachedStrategy()
strat.return_one = lambda *a, **kw: 1
strat.actions = [['return_one', 'anything']]
self.assertEqual(
strat.process_view(req, None, (), {}),
None
)
def test_get_authentication_backend_meta(self):
base_backend = type('Any', (object,), {})
new_backend = CachedStrategy.get_authentication_backend(base_backend)
self.assertTrue(isinstance(new_backend(), base_backend))
self.assertEqual(new_backend().__module__, base_backend.__module__)
self.assertEqual(new_backend().__class__, base_backend)
def test_whitelist_ip_returns_response_fn_on_not_okay(self):
strat = CachedStrategy()
ip = fmt_ip(random.randint(1, 0xFFFFFFFF))
whitelist = CachedWhitelist.objects.create(name='asdf', slug='asdf')
whitelist.range_set.create(
ip='0.0.0.0',
cidr=32
)
Expected = object()
self.assertEqual(
strat.whitelist_ip(ip, [whitelist], lambda:Expected),
Expected
)
def test_process_view_skips_methods_returning_no_whitelists(self):
expected = fmt_ip(random.randint(1, 0xFFFFFFFF))
req = self.fake_request(expected)
strat = CachedStrategy()
# make sure that the cache is empty...
cache.delete('whitelist:pk:1')
# there are no whitelists at this point,
# so a 'pk' lookup will return nothing!
strat.return_one = lambda *a, **kw: 1
strat.actions = [['return_one', 'pk']]
self.assertEqual(
strat.process_view(req, None, (), {}),
None
)
def test_get_ip_from_request(self):
strat = CachedStrategy()
expected = random.randint(1, 10)
req = self.fake_request(expected)
strat.request_ip_headers = ['REMOTE_ADDR', 'LOL']
self.assertEqual(strat.get_ip_from_request(req), expected)
strat.request_ip_headers = ['LOL']
self.assertEqual(strat.get_ip_from_request(req), 0)
strat.request_ip_headers = ['LOL', 'REMOTE_ADDR']
self.assertEqual(strat.get_ip_from_request(req), 0)
strat.request_ip_headers = ['DNE']
self.assertEqual(strat.get_ip_from_request(req), None)
def test_process_view_passes_forbidden_response_class(self):
expected = fmt_ip(random.randint(1, 0xFFFFFFFF))
req = self.fake_request(expected)
strat = CachedStrategy()
wl = CachedWhitelist.objects.create(
name='okay',
slug='okay'
)
wl.range_set.create(
ip='0.0.0.0',
cidr=32
)
strat.return_one = lambda *a, **kw: wl.pk
strat.actions = [['return_one', 'pk']]
self.assertTrue(
isinstance(strat.process_view(req, None, (), {}),
HttpResponseForbidden)
)
def test_whitelist_ip_automatically_fails_on_no_ip(self):
strat = CachedStrategy()
self.assertEqual(strat.whitelist_ip(None, [], lambda:None), None)
