नोसेटेट्स

Question

के माध्यम से फ्लास्क एप्लिकेशन का परीक्षण करते समय आईपी पता प्राप्त करें मेरा आवेदन request.remote_addr पर निर्भर करता है जो None है जब मैं app.test_client().post('/users/login', ....) का उपयोग करने वाले नोसेटेट्स के माध्यम से परीक्षण चलाता हूं।

जब मैं परीक्षण चलाता हूं तो मैं एक आईपी (127.0.0.1 ठीक काम करता है) का अनुकरण कैसे कर सकता हूं? मैंने पर्यावरण चर सेट करने की कोशिश की है, पोस्ट() विधि के साथ हेडर में भेजा गया है और मैंने नोसेटेट्स, वेर्कजेज और फ्लास्क दस्तावेज के माध्यम से खोला है लेकिन मैंने कोशिश की है कि कुछ भी काम नहीं किया है।

Answer 1

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

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