From 492606108a2a5565efc39aed3b56f2e4a6f3c945 Mon Sep 17 00:00:00 2001 From: Helmut Grohne Date: Thu, 15 Mar 2012 15:57:53 +0100 Subject: added a test for GzipWSGIFilter --- test.py | 18 ++++++++++++++++++ 1 file changed, 18 insertions(+) diff --git a/test.py b/test.py index 99204de..1acf5aa 100755 --- a/test.py +++ b/test.py @@ -130,6 +130,9 @@ class Result: if not found: self.testcase.fail("header %s not found" % name) + def get_data(self): + return "".join(self.writtendata) + "".join(self.returneddata) + from wsgitools import applications class StaticContentTest(unittest.TestCase): @@ -340,6 +343,20 @@ class BasicAuthMiddlewareTest(unittest.TestCase): def test401authfail(self): self.doauth(password="spam", status=401) +from wsgitools import filters +import gzip + +class GzipWSGIFilterTest(unittest.TestCase): + def testSimple(self): + app = applications.StaticContent("200 Found", + [("Content-Type", "text/plain")], "nothing") + app = filters.WSGIFilterMiddleware(app, filters.GzipWSGIFilter) + req = Request(self) + req.environ["HTTP_ACCEPT_ENCODING"] = "gzip" + res = req(app) + data = gzip.GzipFile(fileobj=io.StringIO(res.get_data())).read() + self.assertEqual(data, "nothing") + def alltests(case): return unittest.TestLoader().loadTestsFromTestCase(case) @@ -351,6 +368,7 @@ fullsuite.addTest(alltests(AuthDigestMiddlewareTest)) fullsuite.addTest(alltests(ContentLengthMiddlewareTest)) fullsuite.addTest(alltests(BasicAuthMiddlewareTest)) fullsuite.addTest(alltests(NoWriteCallableMiddlewareTest)) +fullsuite.addTest(alltests(GzipWSGIFilterTest)) if __name__ == "__main__": runner = unittest.TextTestRunner(verbosity=2) -- cgit v1.2.3