通过nosetests测试flask应用程序时获取IP地址答案

【问题标题】：Get IP Address when testing flask application through nosetests通过nosetests测试flask应用程序时获取IP地址
【发布时间】：2013-01-30 03:17:16
【问题描述】：

当我通过使用app.test_client().post('/users/login', ....) 的nosetests 运行测试时，我的应用程序依赖于request.remote_addr，即None。

我如何在运行测试时模拟 IP（127.0.0.1 可以正常工作）？ 我已经尝试设置环境变量，使用 post() 方法在标头中发送，并且我已经深入研究了nosetests、werkzeugs 和 flasks 文档，但我尝试过的都没有奏效。

【问题讨论】：

Setting (mocking) request headers for Flask app unit test的可能重复
@Styvane：嗯，这个问题早一个月了，怎么重复了？
也许我没有选对问题，而是yes it is possible

标签： python testing flask nosetests werkzeug

【解决方案1】：

您可以使用environ_base 为底层 Werkzeug 环境设置选项：

from flask import Flask, request
import unittest

app = Flask(__name__)
app.debug = True
app.testing = True

@app.route('/')
def index():
    return str(request.remote_addr)

class TestApp(unittest.TestCase):

    def test_remote_addr(self):
        c = app.test_client()
        resp = c.get('/', environ_base={'REMOTE_ADDR': '127.0.0.1'})
        self.assertEqual('127.0.0.1', resp.data)


if __name__ == '__main__':
    unittest.main()

【讨论】：

谢谢，它工作得很好，但每次通话都必须附加它会变得乏味。一位朋友给了我一个解决方案，可以改变每次通话。也将其作为答案发布。
+1 这似乎是 werkzeug 打算让你这样做的方式，这正是我所寻找的
现在我的测试在需要有效 IP 的实例中通过。谢谢。

【解决方案2】：

一位朋友给了我这个解决方案，它适用于所有请求：

class myProxyHack(object):

    def __init__(self, app):
        self.app = app

    def __call__(self, environ, start_response):
        environ['REMOTE_ADDR'] = environ.get('REMOTE_ADDR', '127.0.0.1')
        return self.app(environ, start_response)

app.wsgi_app = myProxyHack(app.wsgi_app)

app.test_client().post(...)

【讨论】：

【解决方案3】：

如果您愿意，也可以将 header 参数传递给 test_request_context。

例子：

from flask import Flask, request
import unittest

app = Flask(__name__)
app.debug = True
app.testing = True

@app.route('/')
def index():
    return str(request.remote_addr)

class TestApp(unittest.TestCase):

    def test_headers(self):
        user_agent = "Mozilla/5.0 (Windows NT 10.0; Win64; x64; rv:52.0) Gecko/20100101 Firefox/52.0"
        ip_address = 127.0.0.1
        headers = {
            'Remote_Addr': ip_address,
            'User_Agent': user_agent
        }

        with self.test_request_context(headers=headers):
            # Do something
            pass

当您需要使用其他模块中的request 对象执行多个单元测试时，这很有用。

请参阅test_request_context documentation。

【讨论】：