1 files changed, 106 insertions, 0 deletions
diff --git a/test/netlib/test_wsgi.py b/test/netlib/test_wsgi.py
new file mode 100644
index 00000000..8c782b27
--- /dev/null
+++ b/test/netlib/test_wsgi.py
@@ -0,0 +1,106 @@
+from io import BytesIO
+import sys
+from netlib import wsgi
+from netlib.http import Headers
+
+
+def tflow():
+    headers = Headers(test=b"value")
+    req = wsgi.Request("http", "GET", "/", "HTTP/1.1", headers, "")
+    return wsgi.Flow(("127.0.0.1", 8888), req)
+
+
+class ExampleApp:
+    
+    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 [b'Hello', b' 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 = ExampleApp()
+        w = wsgi.WSGIAdaptor(ta, "foo", 80, "version")
+        f = tflow()
+        f.request.host = "foo"
+        f.request.port = 80
+
+        wfile = BytesIO()
+        err = w.serve(f, wfile)
+        assert ta.called
+        assert not err
+
+        val = wfile.getvalue()
+        assert b"Hello world" in val
+        assert b"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 = BytesIO()
+        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:
+                sys.exc_info()
+            status = '200 OK'
+            response_headers = [('Content-type', 'text/plain')]
+            start_response(status, response_headers)
+            start_response(status, response_headers)
+        assert b"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)
+            yield b""
+        assert b"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 b"aaa"
+            start_response(status, response_headers, ei)
+            yield b"bbb"
+        assert b"Internal Server Error" in self._serve(app)