diff options
Diffstat (limited to 'test/pathod/test_language_writer.py')
| -rw-r--r-- | test/pathod/test_language_writer.py | 31 |
1 files changed, 15 insertions, 16 deletions
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') |