通过nosetests测试烧瓶应用程序时获取IP地址(Get IP Address when tes

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

我怎么能模拟一个IP（127.0.0.1正常工作），当我运行测试？ 我试过设置环境变量，在头部信息与post（）方法，我已经通过nosetests，werkzeugs和烧瓶文档，但没有什么我已经试过了工作凿成。

Answer 1:

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

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

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

文章来源: Get IP Address when testing flask application through nosetests