mirror of
https://github.com/Grasscutters/mitmproxy.git
synced 2024-11-23 16:17:49 +00:00
104 lines
3.1 KiB
Python
104 lines
3.1 KiB
Python
import cStringIO, sys
|
|
from netlib import wsgi, odict
|
|
|
|
|
|
def tflow():
|
|
h = odict.ODictCaseless()
|
|
h["test"] = ["value"]
|
|
req = wsgi.Request("http", "GET", "/", h, "")
|
|
return wsgi.Flow(("127.0.0.1", 8888), req)
|
|
|
|
|
|
class TestApp:
|
|
def __init__(self):
|
|
self.called = False
|
|
|
|
def __call__(self, environ, start_response):
|
|
self.called = True
|
|
status = '200 OK'
|
|
response_headers = [('Content-type', 'text/plain')]
|
|
start_response(status, response_headers)
|
|
return ['Hello', ' world!\n']
|
|
|
|
|
|
class TestWSGI:
|
|
def test_make_environ(self):
|
|
w = wsgi.WSGIAdaptor(None, "foo", 80, "version")
|
|
tf = tflow()
|
|
assert w.make_environ(tf, None)
|
|
|
|
tf.request.path = "/foo?bar=voing"
|
|
r = w.make_environ(tf, None)
|
|
assert r["QUERY_STRING"] == "bar=voing"
|
|
|
|
def test_serve(self):
|
|
ta = TestApp()
|
|
w = wsgi.WSGIAdaptor(ta, "foo", 80, "version")
|
|
f = tflow()
|
|
f.request.host = "foo"
|
|
f.request.port = 80
|
|
|
|
wfile = cStringIO.StringIO()
|
|
err = w.serve(f, wfile)
|
|
assert ta.called
|
|
assert not err
|
|
|
|
val = wfile.getvalue()
|
|
assert "Hello world" in val
|
|
assert "Server:" in val
|
|
|
|
def _serve(self, app):
|
|
w = wsgi.WSGIAdaptor(app, "foo", 80, "version")
|
|
f = tflow()
|
|
f.request.host = "foo"
|
|
f.request.port = 80
|
|
wfile = cStringIO.StringIO()
|
|
err = w.serve(f, wfile)
|
|
return wfile.getvalue()
|
|
|
|
def test_serve_empty_body(self):
|
|
def app(environ, start_response):
|
|
status = '200 OK'
|
|
response_headers = [('Foo', 'bar')]
|
|
start_response(status, response_headers)
|
|
return []
|
|
assert self._serve(app)
|
|
|
|
def test_serve_double_start(self):
|
|
def app(environ, start_response):
|
|
try:
|
|
raise ValueError("foo")
|
|
except:
|
|
ei = sys.exc_info()
|
|
status = '200 OK'
|
|
response_headers = [('Content-type', 'text/plain')]
|
|
start_response(status, response_headers)
|
|
start_response(status, response_headers)
|
|
assert "Internal Server Error" in self._serve(app)
|
|
|
|
def test_serve_single_err(self):
|
|
def app(environ, start_response):
|
|
try:
|
|
raise ValueError("foo")
|
|
except:
|
|
ei = sys.exc_info()
|
|
status = '200 OK'
|
|
response_headers = [('Content-type', 'text/plain')]
|
|
start_response(status, response_headers, ei)
|
|
assert "Internal Server Error" in self._serve(app)
|
|
|
|
def test_serve_double_err(self):
|
|
def app(environ, start_response):
|
|
try:
|
|
raise ValueError("foo")
|
|
except:
|
|
ei = sys.exc_info()
|
|
status = '200 OK'
|
|
response_headers = [('Content-type', 'text/plain')]
|
|
start_response(status, response_headers)
|
|
yield "aaa"
|
|
start_response(status, response_headers, ei)
|
|
yield "bbb"
|
|
assert "Internal Server Error" in self._serve(app)
|
|
|