Мое приложение зависит от request.remote_addr, которое None, когда я запускаю тесты через nosetests, который использует app.test_client().post('/users/login', ....).
Как я могу эмулировать IP (127.0.0.1 работает отлично), когда я запускаю тесты? Я пробовал устанавливать переменные окружения, отправленные в заголовках с методом post(), и я выкопал документацию по носететам, werkzeugs и флягам, но ничего, что я пробовал, не работал.
Вы можете установить параметры для базовой среды Werkzeug с помощью environment_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()
Друг дал мне это решение, которое работает по всем запросам:
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(...)
Вы также можете передать параметр 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.