diff options
| author | Aldo Cortesi <aldo@nullcube.com> | 2011-03-12 16:00:01 +1300 |
|---|---|---|
| committer | Aldo Cortesi <aldo@nullcube.com> | 2011-03-12 16:00:01 +1300 |
| commit | 9a9a2d9125e6b1fbbfff0020ad0f29df74249f44 (patch) | |
| tree | fa12eef4ea83dbe602e4ad00295f3e8993eb2e40 | |
| parent | 4ce393cc41269d119a01b852d7b486a7858cc57d (diff) | |
| download | mitmproxy-9a9a2d9125e6b1fbbfff0020ad0f29df74249f44.tar.gz mitmproxy-9a9a2d9125e6b1fbbfff0020ad0f29df74249f44.tar.bz2 mitmproxy-9a9a2d9125e6b1fbbfff0020ad0f29df74249f44.zip | |
Unit tests for proxy.read_chunked
| -rw-r--r-- | test/test_proxy.py | 17 |
1 files changed, 17 insertions, 0 deletions
diff --git a/test/test_proxy.py b/test/test_proxy.py index 6bae46fc..82d2dbec 100644 --- a/test/test_proxy.py +++ b/test/test_proxy.py @@ -5,6 +5,22 @@ import email.utils import tutils +class u_read_chunked(libpry.AutoTree): + def test_all(self): + s = cStringIO.StringIO("1\r\na\r\n0\r\n") + libpry.raises(IOError, proxy.read_chunked, s) + + s = cStringIO.StringIO("1\r\na\r\n0\r\n\r\n") + assert proxy.read_chunked(s) == "a" + + s = cStringIO.StringIO("\r\n") + libpry.raises(IOError, proxy.read_chunked, s) + + s = cStringIO.StringIO("1\r\nfoo") + libpry.raises(IOError, proxy.read_chunked, s) + + + class u_parse_request_line(libpry.AutoTree): def test_simple(self): libpry.raises(proxy.ProxyError, proxy.parse_request_line, "") @@ -201,4 +217,5 @@ tests = [ u_parse_url(), uError(), uClientConnect(), + u_read_chunked(), ] |