diff options
author | Maximilian Hils <git@maximilianhils.com> | 2016-07-21 01:16:35 -0700 |
---|---|---|
committer | Maximilian Hils <git@maximilianhils.com> | 2016-07-21 01:16:35 -0700 |
commit | 6ffeaaebed0ac248b5ba1f60c6add44eb6e98004 (patch) | |
tree | 60900f6ea943125258dfb723b971ffe47726a19d /test/netlib | |
parent | 8a3a21bba1e6706295cc22e1b3a876a7a86cb705 (diff) | |
parent | c090e02848e0b0b34021ca72f0fc0d39be4de0d0 (diff) | |
download | mitmproxy-6ffeaaebed0ac248b5ba1f60c6add44eb6e98004.tar.gz mitmproxy-6ffeaaebed0ac248b5ba1f60c6add44eb6e98004.tar.bz2 mitmproxy-6ffeaaebed0ac248b5ba1f60c6add44eb6e98004.zip |
Merge branch 'master' of https://github.com/mitmproxy/mitmproxy
Diffstat (limited to 'test/netlib')
-rw-r--r-- | test/netlib/http/http1/test_read.py | 9 | ||||
-rw-r--r-- | test/netlib/test_utils.py | 7 |
2 files changed, 16 insertions, 0 deletions
diff --git a/test/netlib/http/http1/test_read.py b/test/netlib/http/http1/test_read.py index c8a40ecb..44eff2ee 100644 --- a/test/netlib/http/http1/test_read.py +++ b/test/netlib/http/http1/test_read.py @@ -13,6 +13,7 @@ from netlib.http.http1.read import ( _read_headers, _read_chunked, get_header_tokens ) from netlib.tutils import treq, tresp, raises +from netlib import exceptions def test_get_header_tokens(): @@ -42,6 +43,14 @@ def test_read_request(input): assert rfile.read() == b"skip" +@pytest.mark.parametrize("input", [ + b"CONNECT :0 0", +]) +def test_read_request_error(input): + rfile = BytesIO(input) + raises(exceptions.HttpException, read_request, rfile) + + def test_read_request_head(): rfile = BytesIO( b"GET / HTTP/1.1\r\n" diff --git a/test/netlib/test_utils.py b/test/netlib/test_utils.py index eaa66f13..9dcbffd8 100644 --- a/test/netlib/test_utils.py +++ b/test/netlib/test_utils.py @@ -3,6 +3,13 @@ from netlib import utils, tutils +def test_is_valid_host(): + assert not utils.is_valid_host(b"") + assert utils.is_valid_host(b"one.two") + assert not utils.is_valid_host(b"one" * 255) + assert utils.is_valid_host(b"one.two.") + + def test_bidi(): b = utils.BiDi(a=1, b=2) assert b.a == 1 |