diff options
author | Maximilian Hils <git@maximilianhils.com> | 2016-03-20 19:40:03 +0100 |
---|---|---|
committer | Maximilian Hils <git@maximilianhils.com> | 2016-03-20 19:40:03 +0100 |
commit | 88d365cfe671bc5c8d96e78334337aa84cc815da (patch) | |
tree | 21a9bfd946cc5b8958d1b3549a277b51a456ba19 /test/pathod | |
parent | 61a657fe56949732d6470862c2ebbe09141f85f2 (diff) | |
download | mitmproxy-88d365cfe671bc5c8d96e78334337aa84cc815da.tar.gz mitmproxy-88d365cfe671bc5c8d96e78334337aa84cc815da.tar.bz2 mitmproxy-88d365cfe671bc5c8d96e78334337aa84cc815da.zip |
py3++: cStringIO
Diffstat (limited to 'test/pathod')
-rw-r--r-- | test/pathod/test_language_actions.py | 4 | ||||
-rw-r--r-- | test/pathod/test_language_http.py | 17 | ||||
-rw-r--r-- | test/pathod/test_language_http2.py | 16 | ||||
-rw-r--r-- | test/pathod/test_language_writer.py | 31 | ||||
-rw-r--r-- | test/pathod/test_pathoc.py | 14 | ||||
-rw-r--r-- | test/pathod/test_pathoc_cmdline.py | 4 | ||||
-rw-r--r-- | test/pathod/test_pathod.py | 6 | ||||
-rw-r--r-- | test/pathod/test_pathod_cmdline.py | 1 | ||||
-rw-r--r-- | test/pathod/tutils.py | 10 |
9 files changed, 48 insertions, 55 deletions
diff --git a/test/pathod/test_language_actions.py b/test/pathod/test_language_actions.py index 1c404714..c2e15189 100644 --- a/test/pathod/test_language_actions.py +++ b/test/pathod/test_language_actions.py @@ -1,4 +1,4 @@ -import cStringIO +from six.moves import cStringIO as StringIO from pathod.language import actions from pathod import language @@ -61,7 +61,7 @@ class TestInject: assert v.offset == "r" def test_serve(self): - s = cStringIO.StringIO() + s = StringIO() r = language.parse_pathod("400:i0,'foo'").next() assert language.serve(r, s, {}) diff --git a/test/pathod/test_language_http.py b/test/pathod/test_language_http.py index fd60c9fd..d1870a63 100644 --- a/test/pathod/test_language_http.py +++ b/test/pathod/test_language_http.py @@ -1,5 +1,4 @@ -import cStringIO - +from six.moves import cStringIO as StringIO from pathod import language from pathod.language import http, base import tutils @@ -10,7 +9,7 @@ def parse_request(s): def test_make_error_response(): - d = cStringIO.StringIO() + d = StringIO() s = http.make_error_response("foo") language.serve(s, d, {}) @@ -76,7 +75,7 @@ class TestRequest: assert r[0].values({}) def test_render(self): - s = cStringIO.StringIO() + s = StringIO() r = parse_request("GET:'/foo'") assert language.serve( r, @@ -163,7 +162,7 @@ class TestResponse: assert "OK" in [i[:] for i in r.preamble({})] def test_render(self): - s = cStringIO.StringIO() + s = StringIO() r = language.parse_pathod("400:m'msg'").next() assert language.serve(r, s, {}) @@ -173,13 +172,13 @@ class TestResponse: assert "p0" not in s.spec() def test_raw(self): - s = cStringIO.StringIO() + s = StringIO() r = language.parse_pathod("400:b'foo'").next() language.serve(r, s, {}) v = s.getvalue() assert "Content-Length" in v - s = cStringIO.StringIO() + s = StringIO() r = language.parse_pathod("400:b'foo':r").next() language.serve(r, s, {}) v = s.getvalue() @@ -187,7 +186,7 @@ class TestResponse: def test_length(self): def testlen(x): - s = cStringIO.StringIO() + s = StringIO() x = x.next() language.serve(x, s, language.Settings()) assert x.length(language.Settings()) == len(s.getvalue()) @@ -198,7 +197,7 @@ class TestResponse: def test_maximum_length(self): def testlen(x): x = x.next() - s = cStringIO.StringIO() + s = StringIO() m = x.maximum_length({}) language.serve(x, s, {}) assert m >= len(s.getvalue()) diff --git a/test/pathod/test_language_http2.py b/test/pathod/test_language_http2.py index 8e9f188c..abfe4606 100644 --- a/test/pathod/test_language_http2.py +++ b/test/pathod/test_language_http2.py @@ -1,4 +1,4 @@ -import cStringIO +from six.moves import cStringIO as StringIO import netlib from netlib import tcp @@ -25,7 +25,7 @@ def default_settings(): def test_make_error_response(): - d = cStringIO.StringIO() + d = StringIO() s = http2.make_error_response("foo", "bar") language.serve(s, d, default_settings()) @@ -84,7 +84,7 @@ class TestRequest: assert r[1].method.string() == "GET" def test_render_simple(self): - s = cStringIO.StringIO() + s = StringIO() r = parse_request("GET:'/foo'") assert language.serve( r, @@ -126,7 +126,7 @@ class TestRequest: assert r.headers[0].values(default_settings()) == ("user-agent", user_agents.get_by_shortcut('a')[2]) def test_render_with_headers(self): - s = cStringIO.StringIO() + s = StringIO() r = parse_request('GET:/foo:h"foo"="bar"') assert language.serve( r, @@ -143,7 +143,7 @@ class TestRequest: def test_render_with_body(self): - s = cStringIO.StringIO() + s = StringIO() r = parse_request("GET:'/foo':bfoobar") assert language.serve( r, @@ -200,7 +200,7 @@ class TestResponse: assert r.body.string() == "foobar" def test_render_simple(self): - s = cStringIO.StringIO() + s = StringIO() r = parse_response('200') assert language.serve( r, @@ -209,7 +209,7 @@ class TestResponse: ) def test_render_with_headers(self): - s = cStringIO.StringIO() + s = StringIO() r = parse_response('200:h"foo"="bar"') assert language.serve( r, @@ -218,7 +218,7 @@ class TestResponse: ) def test_render_with_body(self): - s = cStringIO.StringIO() + s = StringIO() r = parse_response('200:bfoobar') assert language.serve( r, diff --git a/test/pathod/test_language_writer.py b/test/pathod/test_language_writer.py index 1a4ac954..0a85524f 100644 --- a/test/pathod/test_language_writer.py +++ b/test/pathod/test_language_writer.py @@ -1,5 +1,4 @@ -import cStringIO - +from six.moves import cStringIO as StringIO from pathod import language from pathod.language import writer @@ -7,12 +6,12 @@ from pathod.language import writer def test_send_chunk(): v = "foobarfoobar" for bs in range(1, len(v) + 2): - s = cStringIO.StringIO() + s = StringIO() writer.send_chunk(s, v, bs, 0, len(v)) assert s.getvalue() == v for start in range(len(v)): for end in range(len(v)): - s = cStringIO.StringIO() + s = StringIO() writer.send_chunk(s, v, bs, start, end) assert s.getvalue() == v[start:end] @@ -20,21 +19,21 @@ def test_send_chunk(): def test_write_values_inject(): tst = "foo" - s = cStringIO.StringIO() + s = StringIO() writer.write_values(s, [tst], [(0, "inject", "aaa")], blocksize=5) assert s.getvalue() == "aaafoo" - s = cStringIO.StringIO() + s = StringIO() writer.write_values(s, [tst], [(1, "inject", "aaa")], blocksize=5) assert s.getvalue() == "faaaoo" - s = cStringIO.StringIO() + s = StringIO() writer.write_values(s, [tst], [(1, "inject", "aaa")], blocksize=5) assert s.getvalue() == "faaaoo" def test_write_values_disconnects(): - s = cStringIO.StringIO() + s = StringIO() tst = "foo" * 100 writer.write_values(s, [tst], [(0, "disconnect")], blocksize=5) assert not s.getvalue() @@ -42,13 +41,13 @@ def test_write_values_disconnects(): def test_write_values(): tst = "foobarvoing" - s = cStringIO.StringIO() + s = StringIO() writer.write_values(s, [tst], []) assert s.getvalue() == tst for bs in range(1, len(tst) + 2): for off in range(len(tst)): - s = cStringIO.StringIO() + s = StringIO() writer.write_values( s, [tst], [(off, "disconnect")], blocksize=bs ) @@ -58,34 +57,34 @@ def test_write_values(): def test_write_values_pauses(): tst = "".join(str(i) for i in range(10)) for i in range(2, 10): - s = cStringIO.StringIO() + s = StringIO() writer.write_values( s, [tst], [(2, "pause", 0), (1, "pause", 0)], blocksize=i ) assert s.getvalue() == tst for i in range(2, 10): - s = cStringIO.StringIO() + s = StringIO() writer.write_values(s, [tst], [(1, "pause", 0)], blocksize=i) assert s.getvalue() == tst tst = ["".join(str(i) for i in range(10))] * 5 for i in range(2, 10): - s = cStringIO.StringIO() + s = StringIO() writer.write_values(s, tst[:], [(1, "pause", 0)], blocksize=i) assert s.getvalue() == "".join(tst) def test_write_values_after(): - s = cStringIO.StringIO() + s = StringIO() r = language.parse_pathod("400:da").next() language.serve(r, s, {}) - s = cStringIO.StringIO() + s = StringIO() r = language.parse_pathod("400:pa,0").next() language.serve(r, s, {}) - s = cStringIO.StringIO() + s = StringIO() r = language.parse_pathod("400:ia,'xx'").next() language.serve(r, s, {}) assert s.getvalue().endswith('xx') diff --git a/test/pathod/test_pathoc.py b/test/pathod/test_pathoc.py index 280a9141..8d0f92ac 100644 --- a/test/pathod/test_pathoc.py +++ b/test/pathod/test_pathoc.py @@ -1,5 +1,5 @@ import json -import cStringIO +from six.moves import cStringIO as StringIO import re import OpenSSL import pytest @@ -64,7 +64,7 @@ class _TestDaemon: ignoretimeout=None, showsummary=True ): - s = cStringIO.StringIO() + s = StringIO() c = pathoc.Pathoc( ("127.0.0.1", self.d.port), ssl=self.ssl, @@ -129,7 +129,7 @@ class TestDaemonSSL(_TestDaemon): assert d["log"][0]["request"]["clientcert"]["keyinfo"] def test_http2_without_ssl(self): - fp = cStringIO.StringIO() + fp = StringIO() c = pathoc.Pathoc( ("127.0.0.1", self.d.port), use_http2=True, @@ -227,15 +227,15 @@ class TestDaemon(_TestDaemon): def test_connect_fail(self): to = ("foobar", 80) c = pathoc.Pathoc(("127.0.0.1", self.d.port), fp=None) - c.rfile, c.wfile = cStringIO.StringIO(), cStringIO.StringIO() + c.rfile, c.wfile = StringIO(), StringIO() with raises("connect failed"): c.http_connect(to) - c.rfile = cStringIO.StringIO( + c.rfile = StringIO( "HTTP/1.1 500 OK\r\n" ) with raises("connect failed"): c.http_connect(to) - c.rfile = cStringIO.StringIO( + c.rfile = StringIO( "HTTP/1.1 200 OK\r\n" ) c.http_connect(to) @@ -243,7 +243,7 @@ class TestDaemon(_TestDaemon): def test_socks_connect(self): to = ("foobar", 80) c = pathoc.Pathoc(("127.0.0.1", self.d.port), fp=None) - c.rfile, c.wfile = tutils.treader(""), cStringIO.StringIO() + c.rfile, c.wfile = tutils.treader(""), StringIO() tutils.raises(pathoc.PathocError, c.socks_connect, to) c.rfile = tutils.treader( diff --git a/test/pathod/test_pathoc_cmdline.py b/test/pathod/test_pathoc_cmdline.py index ab1438d1..f527e861 100644 --- a/test/pathod/test_pathoc_cmdline.py +++ b/test/pathod/test_pathoc_cmdline.py @@ -1,13 +1,13 @@ from pathod import pathoc_cmdline as cmdline import tutils -import cStringIO +from six.moves import cStringIO as StringIO import mock @mock.patch("argparse.ArgumentParser.error") def test_pathoc(perror): assert cmdline.args_pathoc(["pathoc", "foo.com", "get:/"]) - s = cStringIO.StringIO() + s = StringIO() with tutils.raises(SystemExit): cmdline.args_pathoc(["pathoc", "--show-uas"], s, s) diff --git a/test/pathod/test_pathod.py b/test/pathod/test_pathod.py index f8eb55f9..c0039b73 100644 --- a/test/pathod/test_pathod.py +++ b/test/pathod/test_pathod.py @@ -1,6 +1,4 @@ -import sys -import cStringIO -import OpenSSL +from six.moves import cStringIO as StringIO import pytest from pathod import pathod, version @@ -12,7 +10,7 @@ import tutils class TestPathod(object): def test_logging(self): - s = cStringIO.StringIO() + s = StringIO() p = pathod.Pathod(("127.0.0.1", 0), logfp=s) assert len(p.get_log()) == 0 id = p.add_log(dict(s="foo")) diff --git a/test/pathod/test_pathod_cmdline.py b/test/pathod/test_pathod_cmdline.py index d156bf50..3c0918ef 100644 --- a/test/pathod/test_pathod_cmdline.py +++ b/test/pathod/test_pathod_cmdline.py @@ -1,6 +1,5 @@ from pathod import pathod_cmdline as cmdline import tutils -import cStringIO import mock diff --git a/test/pathod/tutils.py b/test/pathod/tutils.py index abe7a59b..9739afde 100644 --- a/test/pathod/tutils.py +++ b/test/pathod/tutils.py @@ -1,9 +1,7 @@ import tempfile -import os import re import shutil -import cStringIO -from contextlib import contextmanager +from six.moves import cStringIO as StringIO import netlib from pathod import utils, test, pathoc, pathod, language @@ -14,7 +12,7 @@ def treader(bytes): """ Construct a tcp.Read object from bytes. """ - fp = cStringIO.StringIO(bytes) + fp = StringIO(bytes) return tcp.Reader(fp) @@ -93,7 +91,7 @@ class DaemonTests(object): """ if ssl is None: ssl = self.ssl - logfp = cStringIO.StringIO() + logfp = StringIO() c = pathoc.Pathoc( ("localhost", self.d.port), ssl=ssl, @@ -123,6 +121,6 @@ test_data = utils.Data(__name__) def render(r, settings=language.Settings()): r = r.resolve(settings) - s = cStringIO.StringIO() + s = StringIO() assert language.serve(r, s, settings) return s.getvalue() |