From f6ef23ba610ffe47d08d9684ba6bd554c78862fb Mon Sep 17 00:00:00 2001
From: Marc Abramowitz <marc@marc-abramowitz.com>
Date: Mon, 7 Mar 2016 15:19:32 -0800
Subject: Make gzip work with app that is generator

```
$ tox
...
  py26: commands succeeded
  py27: commands succeeded
  py34: commands succeeded
  py35: commands succeeded
  pypy: commands succeeded
  congratulations :)
```
---
 tests/test_gzipper.py | 53 +++++++++++++++++++++++++++++++++++++++++++++++++--
 1 file changed, 51 insertions(+), 2 deletions(-)

(limited to 'tests')

diff --git a/tests/test_gzipper.py b/tests/test_gzipper.py
index 54b7901..8ad5ec3 100644
--- a/tests/test_gzipper.py
+++ b/tests/test_gzipper.py
@@ -3,17 +3,66 @@ from paste.gzipper import middleware
 import gzip
 import six
 
+
 def simple_app(environ, start_response):
     start_response('200 OK', [('content-type', 'text/plain')])
     return [b'this is a test']
 
-wsgi_app = middleware(simple_app)
-app = TestApp(wsgi_app)
+
+def simple_app_nothing(environ, start_response):
+    start_response('200 OK', [('content-type', 'text/plain')])
+    return [b'']
+
+
+def simple_app_generator(environ, start_response):
+    start_response('200 OK', [('content-type', 'text/plain')])
+    yield b'this is a test'
+
+
+def simple_app_generator_nothing(environ, start_response):
+    start_response('200 OK', [('content-type', 'text/plain')])
+    yield b''
+
 
 def test_gzip():
+    wsgi_app = middleware(simple_app)
+    app = TestApp(wsgi_app)
+    res = app.get(
+        '/', extra_environ=dict(HTTP_ACCEPT_ENCODING='gzip'))
+    assert int(res.header('content-length')) == len(res.body)
+    assert res.body != b'this is a test'
+    actual = gzip.GzipFile(fileobj=six.BytesIO(res.body)).read()
+    assert actual == b'this is a test'
+
+
+def test_gzip_nothing():
+    wsgi_app = middleware(simple_app_nothing)
+    app = TestApp(wsgi_app)
+    res = app.get(
+        '/', extra_environ=dict(HTTP_ACCEPT_ENCODING='gzip'))
+    assert int(res.header('content-length')) == len(res.body)
+    assert res.body != b'this is a test'
+    actual = gzip.GzipFile(fileobj=six.BytesIO(res.body)).read()
+    assert actual == b''
+
+
+def test_gzip_generator():
+    wsgi_app = middleware(simple_app_generator)
+    app = TestApp(wsgi_app)
     res = app.get(
         '/', extra_environ=dict(HTTP_ACCEPT_ENCODING='gzip'))
     assert int(res.header('content-length')) == len(res.body)
     assert res.body != b'this is a test'
     actual = gzip.GzipFile(fileobj=six.BytesIO(res.body)).read()
     assert actual == b'this is a test'
+
+
+def test_gzip_generator_nothing():
+    wsgi_app = middleware(simple_app_generator_nothing)
+    app = TestApp(wsgi_app)
+    res = app.get(
+        '/', extra_environ=dict(HTTP_ACCEPT_ENCODING='gzip'))
+    assert int(res.header('content-length')) == len(res.body)
+    assert res.body != b'this is a test'
+    actual = gzip.GzipFile(fileobj=six.BytesIO(res.body)).read()
+    assert actual == b''
-- 
cgit v1.2.1