From d51b8cab0c0d1352865155865dfd258f66103ffe Mon Sep 17 00:00:00 2001 From: Aldo Cortesi Date: Fri, 16 Mar 2012 11:12:56 +1300 Subject: Add a decoded context manager. This simplifies a common chore when modifying traffic - decoding the object, modifying it, then re-encoding it with the same encoding afterwards. You can now simply say: with flow.decoded(request): request.content = "bar" --- test/test_flow.py | 24 ++++++++++++++++++++++++ 1 file changed, 24 insertions(+) (limited to 'test') diff --git a/test/test_flow.py b/test/test_flow.py index b6818960..56303881 100644 --- a/test/test_flow.py +++ b/test/test_flow.py @@ -996,6 +996,29 @@ class uODictCaseless(libpry.AutoTree): assert len(self.od) == 1 +class udecoded(libpry.AutoTree): + def test_del(self): + r = tutils.treq() + assert r.content == "content" + assert not r.headers["content-encoding"] + r.encode("gzip") + assert r.headers["content-encoding"] + assert r.content != "content" + with flow.decoded(r): + assert not r.headers["content-encoding"] + assert r.content == "content" + assert r.headers["content-encoding"] + assert r.content != "content" + + with flow.decoded(r): + r.content = "foo" + + assert r.content != "foo" + r.decode() + assert r.content == "foo" + + + tests = [ uStickyCookieState(), @@ -1012,4 +1035,5 @@ tests = [ uClientConnect(), uODict(), uODictCaseless(), + udecoded() ] -- cgit v1.2.3