committing an example that worked before the recent change to httpexceptions

1 files changed, 19 insertions, 1 deletions

diff --git a/tests/test_exceptions/test_httpexceptions.py b/tests/test_exceptions/test_httpexceptions.py
index 5b0101b..4e958ce 100644
--- a/tests/test_exceptions/test_httpexceptions.py
+++ b/tests/test_exceptions/test_httpexceptions.py
@@ -7,7 +7,7 @@ WSGI Exception Middleware
 Regression Test Suite
 """
 from paste.httpexceptions import *
-from paste.wsgilib import header_value
+from paste.wsgilib import header_value, raw_interactive
 import py
 
 def test_HTTPMove():
@@ -42,6 +42,24 @@ def test_template():
     assert '<p>A fun and <b>happy</b> message.</p>' in \
            e.html({'ping': 'fun', 'pong': 'happy'})
 
+def test_iterator_application():
+    """ 
+    This tests to see that an iterator's exceptions are caught by 
+    HTTPExceptionHandler
+    """
+    def basic_found(environ, start_response):
+        raise HTTPFound("/bing/foo")
+        return ['result']
+    app = HTTPExceptionHandler(basic_found)
+    (status, headers, content, errors) = raw_interactive(app)
+    assert '302 Found' == status
+    def iterate_found(environ, start_response):
+        raise HTTPFound("/bing/foo")
+        yield 'result'
+    app = HTTPExceptionHandler(iterate_found)
+    (status, headers, content, errors) = raw_interactive(app)
+    assert '302 Found' == status
+
 def test_redapp():
     """ check that redirect returns the correct, expected results """
     saved = []