diff options
author | Helmut Grohne <helmut@subdivi.de> | 2012-03-15 14:13:22 +0100 |
---|---|---|
committer | Helmut Grohne <helmut@subdivi.de> | 2012-03-15 14:19:38 +0100 |
commit | 7acd2451ca9c988f0e2a30004a69eaa03c7056b5 (patch) | |
tree | 179c77569709c82aea469d4b84abf52bf40eab34 /test.py | |
parent | a62abee18b11c6bfe637657bf481e866474bb44a (diff) | |
download | wsgitools-7acd2451ca9c988f0e2a30004a69eaa03c7056b5.tar.gz |
added failing test for NoWriteCallableMiddleware
Also set the returneddata attribute of the response object instead of
the request object.
Diffstat (limited to 'test.py')
-rwxr-xr-x | test.py | 17 |
1 files changed, 16 insertions, 1 deletions
@@ -78,7 +78,7 @@ class Request: res.headersdata = headers return write iterator = app(self.environ, start_response) - self.returneddata = list(iterator) + res.returneddata = list(iterator) if hasattr(iterator, "close"): iterator.close() return res @@ -235,6 +235,20 @@ class AuthDigestMiddlewareTest(unittest.TestCase): from wsgitools import middlewares +def writing_application(environ, start_response): + write = start_response("404 Not found", [("Content-Type", "text/plain")]) + write = start_response("200 Ok", [("Content-Type", "text/plain")]) + write("first") + yield "" + yield "second" + +class NoWriteCallableMiddlewareTest(unittest.TestCase): + def testWrite(self): + app = middlewares.NoWriteCallableMiddleware(writing_application) + res = Request(self)(app) + self.assertEqual(res.writtendata, []) + self.assertEqual("".join(res.returneddata), "firstsecond") + class StupidIO: """file-like without tell method, so StaticFile is not able to determine the content-length.""" @@ -324,6 +338,7 @@ fullsuite.addTest(alltests(StaticFileTest)) fullsuite.addTest(alltests(AuthDigestMiddlewareTest)) fullsuite.addTest(alltests(ContentLengthMiddlewareTest)) fullsuite.addTest(alltests(BasicAuthMiddlewareTest)) +fullsuite.addTest(alltests(NoWriteCallableMiddlewareTest)) if __name__ == "__main__": runner = unittest.TextTestRunner(verbosity=2) |