Django RequestFactory add HTTP_X_FORWARDED_FOR

Question

我有这个代码块，我正在尝试为其编写测试：

def get_client_ip(req):
"""
This is used to get the user's IP from the request object.
"""
x_forwarded_for = req.META.get('HTTP_X_FORWARDED_FOR')
if x_forwarded_for:
ip = x_forwarded_for.split(',')[0]
else:
ip = req.META.get('REMOTE_ADDR', "unknown")
return ip

到目前为止，我已经能够测试 if 语句的 else 部分。 这是我的测试目前的样子：

def test_get_client_ip(self):
"""
Test the get Client IP fuction with a request.
"""
# Create an instance of a GET request.
request = self.factory.get('/home')
ip = get_client_ip(request)
self.assertEqual(ip, '127.0.0.1')

我将如何将"HTTP_X_FORWARDED_FOR"添加到请求对象的 META 中？

Answer 1

事实证明，您可以使用 get 方法中的 extra 关键字为 get 请求提供额外的标头： https://docs.djangoproject.com/en/2.0/topics/testing/tools/#django.test.Client.get

工作代码如下：

def test_get_client_ip(self):
"""
Test the get Client IP fuction with a request.
"""
# Create an instance of a GET request.
request = self.factory.get('/home')
ip = get_client_ip(request)
self.assertEqual(ip, '127.0.0.1')
request = self.factory.get('/home', HTTP_X_FORWARDED_FOR="8.8.8.8")
ip = get_client_ip(request)
self.assertEqual(ip, '8.8.8.8')