20

私のアプリケーションは、request.remote_addrをNone使用するnosetestsを介してテストを実行するタイミングによって異なりますapp.test_client().post('/users/login', ....)。

テストを実行するときにIPをエミュレートするにはどうすればよいですか（127.0.0.1は正常に機能します）？ post（）メソッドを使用してヘッダーで送信される環境変数の設定を試し、nosetests、werkzeugs、flacksのドキュメントを調べましたが、試したことは何も機能しませんでした。

4

3 に答える 3

27

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()
于 2013-02-14T10:46:21.087 に答える
8

友人が私にこのソリューションをくれました。これはすべてのリクエストで機能します。

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(...)
于 2013-02-14T10:54:06.290 に答える
0

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のドキュメントを参照してください。

于 2019-07-22T15:43:19.103 に答える