def test_get_ip_from_request(self): strat = Strategy() 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)