From 32ad26f8bfe573f817da76db227480d3b83904d1 Mon Sep 17 00:00:00 2001 From: Aldo Cortesi Date: Sat, 4 Aug 2012 13:18:05 +1200 Subject: Add a size() method to flow.Request and flow.Response. --- test/test_flow.py | 7 +++++++ 1 file changed, 7 insertions(+) (limited to 'test') diff --git a/test/test_flow.py b/test/test_flow.py index a2051eae..eccd11f4 100644 --- a/test/test_flow.py +++ b/test/test_flow.py @@ -724,12 +724,14 @@ class TestRequest: assert not r.set_url("") assert r.get_url() == u assert r._assemble() + assert r.size() == len(r._assemble()) r2 = r.copy() assert r == r2 r.content = None assert r._assemble() + assert r.size() == len(r._assemble()) r.close = True assert "connection: close" in r._assemble() @@ -739,6 +741,8 @@ class TestRequest: r.content = flow.CONTENT_MISSING assert not r._assemble() + + def test_getset_form_urlencoded(self): h = flow.ODictCaseless() h["content-type"] = [flow.HDR_FORM_URLENCODED] @@ -860,12 +864,15 @@ class TestResponse: req = flow.Request(c, (1, 1), "host", 22, "https", "GET", "/", h, "content") resp = flow.Response(req, (1, 1), 200, "msg", h.copy(), "content", None) assert resp._assemble() + assert resp.size() == len(resp._assemble()) + resp2 = resp.copy() assert resp2 == resp resp.content = None assert resp._assemble() + assert resp.size() == len(resp._assemble()) resp.content = flow.CONTENT_MISSING assert not resp._assemble() -- cgit v1.2.3