aboutsummaryrefslogtreecommitdiffstats
path: root/test/test_flow.py
diff options
context:
space:
mode:
Diffstat (limited to 'test/test_flow.py')
-rw-r--r--test/test_flow.py7
1 files changed, 7 insertions, 0 deletions
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()