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_whitelist_ip_automatically_fails_on_no_ip(self):
strat = CachedStrategy()
self.assertEqual(strat.whitelist_ip(None, [], lambda:None), None)
