python - unit - Obtenga la dirección IP al probar la aplicación del matraz a través de las pruebas de nos

test unitarios flask

Question

Mi aplicación depende de request.remote_addr que es None cuando ejecuto pruebas a través de pruebas de detección que usan app.test_client().post(''/users/login'', ....) .

¿Cómo puedo emular una IP (127.0.0.1 funciona bien) cuando ejecuto pruebas? Intenté establecer variables de entorno, envié encabezados con el método post () y revisé las pruebas de detección de narices, werkzeugs y matraces, pero nada de lo que probé ha funcionado.

Answer 1

Puede establecer opciones para el entorno subyacente de Werkzeug utilizando 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

Un amigo me dio esta solución, que funciona en todas las solicitudes:

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