def test_whitelist_ip_returns_response_fn_on_not_okay(self): strat = Strategy() ip = fmt_ip(random.randint(1, 0xFFFFFFFF)) whitelist = Whitelist.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 = Strategy() self.assertEqual(strat.whitelist_ip(None, [], lambda:None), None)