From 2a5164351f0f5e32106ed3c18d0bf1711c4bb147 Mon Sep 17 00:00:00 2001 From: naivekun Date: Sat, 29 Feb 2020 00:27:06 +0800 Subject: extend tests for headers.replace() --- test/mitmproxy/net/http/test_headers.py | 6 ++++++ 1 file changed, 6 insertions(+) (limited to 'test') diff --git a/test/mitmproxy/net/http/test_headers.py b/test/mitmproxy/net/http/test_headers.py index 8fc8b027..eb065fec 100644 --- a/test/mitmproxy/net/http/test_headers.py +++ b/test/mitmproxy/net/http/test_headers.py @@ -88,6 +88,12 @@ class TestHeaders: headers = Headers(Host="foobarfoo.com", Accept="foo/bar") replacements = headers.replace("foo", "bar", count=1) assert replacements == 1 + assert headers["Host"] == "barbarfoo.com" + assert headers["Accept"] == "foo/bar" + replacements = headers.replace("foo", "bar", count=2) + assert replacements == 2 + assert headers["Host"] == "barbarbar.com" + assert headers["Accept"] == "bar/bar" def test_parse_content_type(): -- cgit v1.2.3