通过nosetests测试Flask应用时获取IP地址

Question

我的应用程序依赖于 request.remote_addr，但是当我通过 nosetests 运行测试时，它的值是 None。这是因为我使用了 app.test_client().post('/users/login', ....)。

我该如何在运行测试时模拟一个 IP 地址（127.0.0.1 可以正常工作）呢？我尝试过设置环境变量，使用 post() 方法发送头信息，还查阅了 nosetests、werkzeugs 和 flask 的文档，但我尝试的都没有成功。

Answer 1

如果你愿意的话，还可以给 test_request_context 传一个 header 参数。

举个例子：

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 的文档，了解更多信息。

Answer 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(...)

Answer 3

你可以通过 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()