diff options
-rwxr-xr-x | test.py | 12 | ||||
-rw-r--r-- | wsgitools/middlewares.py | 2 |
2 files changed, 12 insertions, 2 deletions
@@ -276,14 +276,15 @@ class StupidIO(object): return self.content[oldpos:self.position] class ContentLengthMiddlewareTest(unittest.TestCase): - def setUp(self): + def customSetUp(self, maxstore=10): self.staticapp = applications.StaticFile(StupidIO(b"success"), "200 Found", [("Content-Type", "text/plain")]) self.app = middlewares.ContentLengthMiddleware(self.staticapp, - maxstore=10) + maxstore=maxstore) self.req = Request(self) def testWithout(self): + self.customSetUp() res = self.req(self.staticapp) res.status("200 Found") try: @@ -293,6 +294,13 @@ class ContentLengthMiddlewareTest(unittest.TestCase): pass def testGet(self): + self.customSetUp() + res = self.req(self.app) + res.status("200 Found") + res.header("Content-length", "7") + + def testInfiniteMaxstore(self): + self.customSetUp(maxstore=()) res = self.req(self.app) res.status("200 Found") res.header("Content-length", "7") diff --git a/wsgitools/middlewares.py b/wsgitools/middlewares.py index b37b130..97b8092 100644 --- a/wsgitools/middlewares.py +++ b/wsgitools/middlewares.py @@ -137,6 +137,8 @@ class ContentLengthMiddleware(object): wsgi standard """ self.app = app + if maxstore == (): + maxstore = float("inf") self.maxstore = maxstore def __call__(self, environ, start_response): """wsgi interface""" |