如何用Django测试客户端指定IP地址？

Question

I am testing an API with Django test client. The API uses geo blocking so in my test I need to specify an IP address to make sure it works properly. How can I do that?

I am making a request in my test like this:

from django.test.client import Client as HttpClient
.
.
.
client = HttpClient()
response = client.get(uri + query_string)

Answer 1

Client.get()方法有一个extra关键字参数参数，可用于指定标题。

c.get(/my-url/, REMOTE_ADDR="127.0.0.1") 

Answer 2

Pass REMOTE_ADDR in constructor.

client = HttpClient(REMOTE_ADDR='127.0.0.1')

or

client.get('/path/', {'param':'foo'}, **{'HTTP_USER_AGENT':'firefox-22', 'REMOTE_ADDR':'127.0.0.1'})

Answer 3

像这样简单：

client_address = request.META.get('HTTP_X_FORWARDED_FOR') or request.META.get('REMOTE_ADDR')

Answer 4

You can also set it for all future requests:

client.defaults['REMOTE_ADDR'] = '1.2.3.4'

Also with subclassing:

class DecoratedApiClient(Client):
    def set_ip_addr(self, ip_addr):
        self.defaults['REMOTE_ADDR'] = ip_addr

client = DecoratedApiClient()
client.set_ip_addr('1.2.3.4')