From 7d56ec9d49b534081d1b8d840c8339b10817b2cd Mon Sep 17 00:00:00 2001 From: Aldo Cortesi Date: Mon, 1 May 2017 18:29:59 +1200 Subject: commands: flow.encode, flow.decode, flow.encode.toggle Use this to bind "z" in flowview. --- test/mitmproxy/addons/test_core.py | 28 ++++++++++++++++++++++++++++ 1 file changed, 28 insertions(+) (limited to 'test') diff --git a/test/mitmproxy/addons/test_core.py b/test/mitmproxy/addons/test_core.py index 302b78ae..eaf0dfa0 100644 --- a/test/mitmproxy/addons/test_core.py +++ b/test/mitmproxy/addons/test_core.py @@ -100,3 +100,31 @@ def test_flow_set(): assert f.response.reason != "foo" sa.flow_set([f], "reason", "foo") assert f.response.reason == "foo" + + +def test_encoding(): + sa = core.Core() + with taddons.context(): + f = tflow.tflow() + assert sa.encode_options() + sa.encode([f], "request", "deflate") + assert f.request.headers["content-encoding"] == "deflate" + + sa.encode([f], "request", "br") + assert f.request.headers["content-encoding"] == "deflate" + + sa.decode([f], "request") + assert "content-encoding" not in f.request.headers + + sa.encode([f], "request", "br") + assert f.request.headers["content-encoding"] == "br" + + sa.encode_toggle([f], "request") + assert "content-encoding" not in f.request.headers + sa.encode_toggle([f], "request") + assert f.request.headers["content-encoding"] == "deflate" + sa.encode_toggle([f], "request") + assert "content-encoding" not in f.request.headers + + with pytest.raises(exceptions.CommandError): + sa.encode([f], "request", "invalid") -- cgit v1.2.3