Python Strategy.request_ip_headers Examples

Example #1

File: strategy.py Project: strycore/django-mithril

    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)