Uzyskaj adres IP podczas testowania aplikacji kolby przez testy nosem

Question

Moja aplikacja zależy od request.remote_addr, która jest None, gdy przeprowadzam testy przez testy nosa, które używają app.test_client().post('/users/login', ....).

Jak mogę emulować adres IP (127.0.0.1 działa poprawnie) po uruchomieniu testów? Próbowałem ustawić zmienne środowiskowe, wysłane w nagłówkach za pomocą metody post(), a ja przekopałem dokumentację nosa, dokumentację werkzeugs i kolb, ale nic, co próbowałem, nie zadziałało.

Answer 1

Można ustawić opcje dla bazowego środowiska Werkzeug użyciu 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

Znajomy dał mi to rozwiązanie, które działa we wszystkich żądań:

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