Add more tests

Singletoned · Singletoned · commit 97c9c5bc298b · 2020-03-28T20:52:37.000Z
diff --git a/httpx/_dispatch/wsgi.py b/httpx/_dispatch/wsgi.py
@@ -18,6 +18,7 @@ def _get_non_empty_chunk(body: typing.Iterable) -> typing.Iterable:
     for chunk in body:
         if chunk:
             return itertools.chain([chunk], body)
+    return []
 
 
 class WSGIDispatch(SyncDispatcher):
diff --git a/tests/client/test_client.py b/tests/client/test_client.py
@@ -155,9 +155,7 @@ def test_elapsed_delay(server):
     assert response.elapsed.total_seconds() > 0.0
 
 
-def test_wsgi_app_generator():
-    def application(env, start_response):
-        body = b"""
+_wsgi_body = b"""
 
 
 <!DOCTYPE html>
@@ -170,10 +168,39 @@ def application(env, start_response):
   </body>
 </html>
 """
+
+
+def test_wsgi_app():
+    def application(env, start_response):
+        start_response("200 OK", [("Content-Type", "text/html")])
+        return [_wsgi_body]
+
+    with httpx.Client(app=application) as client:
+        response = client.get("http://example.com/")
+        assert response.status_code == 200
+        assert b"This page deliberately left blank" in response.content
+
+
+def test_wsgi_app_generator():
+    def application(env, start_response):
         start_response("200 OK", [("Content-Type", "text/html")])
-        for line in body.split(b"\n"):
+        for line in _wsgi_body.split(b"\n"):
             yield line
 
     with httpx.Client(app=application) as client:
         response = client.get("http://example.com/")
+        assert response.status_code == 200
         assert b"This page deliberately left blank" in response.content
+
+
+def test_wsgi_app_generator_empty():
+    def application(env, start_response):
+        body = [b"", b"", b""]
+        start_response("200 OK", [("Content-Type", "text/html")])
+        for line in body:
+            yield line
+
+    with httpx.Client(app=application) as client:
+        response = client.get("http://example.com/")
+        assert response.status_code == 200
+        assert response.content == b""
