python - Django RequestFactory 添加 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 请求提供额外的 header :
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')