Grasscutters/mitmproxy
3
0
Fork 0
mirror of https://github.com/Grasscutters/mitmproxy.git synced 2024-11-23 08:11:00 +00:00
Code Issues Projects Releases Wiki Activity

Py3: Use BytesIO instead of StringIO for tests

Browse Source
This commit is contained in:
Shadab Zafar 2016-06-04 20:56:07 +05:30 committed by Maximilian Hils
parent 12dd6f6707
commit 9a19540f8b
1 changed files with 8 additions and 8 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

16
test/pathod/test_language_http.py
View File

@ -1,4 +1,4 @@
from six.moves import cStringIO as StringIO from six import BytesIO
from pathod import language from pathod import language
from pathod.language import http, base from pathod.language import http, base
import tutils import tutils
@ -9,7 +9,7 @@ def parse_request(s):
def test_make_error_response(): def test_make_error_response():
d = StringIO() d = BytesIO()
s = http.make_error_response("foo") s = http.make_error_response("foo")
language.serve(s, d, {}) language.serve(s, d, {})
@ -75,7 +75,7 @@ class TestRequest:
assert r[0].values({}) assert r[0].values({})
def test_render(self): def test_render(self):
s = StringIO() s = BytesIO()
r = parse_request("GET:'/foo'") r = parse_request("GET:'/foo'")
assert language.serve( assert language.serve(
r, r,
@ -162,7 +162,7 @@ class TestResponse:
assert "OK" in [i[:] for i in r.preamble({})] assert "OK" in [i[:] for i in r.preamble({})]
def test_render(self): def test_render(self):
s = StringIO() s = BytesIO()
r = next(language.parse_pathod("400:m'msg'")) r = next(language.parse_pathod("400:m'msg'"))
assert language.serve(r, s, {}) assert language.serve(r, s, {})
@ -172,13 +172,13 @@ class TestResponse:
assert "p0" not in s.spec() assert "p0" not in s.spec()
def test_raw(self): def test_raw(self):
s = StringIO() s = BytesIO()
r = next(language.parse_pathod("400:b'foo'")) r = next(language.parse_pathod("400:b'foo'"))
language.serve(r, s, {}) language.serve(r, s, {})
v = s.getvalue() v = s.getvalue()
assert "Content-Length" in v assert "Content-Length" in v
s = StringIO() s = BytesIO()
r = next(language.parse_pathod("400:b'foo':r")) r = next(language.parse_pathod("400:b'foo':r"))
language.serve(r, s, {}) language.serve(r, s, {})
v = s.getvalue() v = s.getvalue()
@ -186,7 +186,7 @@ class TestResponse:
def test_length(self): def test_length(self):
def testlen(x): def testlen(x):
s = StringIO() s = BytesIO()
x = next(x) x = next(x)
language.serve(x, s, language.Settings()) language.serve(x, s, language.Settings())
assert x.length(language.Settings()) == len(s.getvalue()) assert x.length(language.Settings()) == len(s.getvalue())
@ -196,8 +196,8 @@ class TestResponse:
def test_maximum_length(self): def test_maximum_length(self):
def testlen(x): def testlen(x):
s = StringIO()
x = next(x) x = next(x)
s = BytesIO()
m = x.maximum_length({}) m = x.maximum_length({})
language.serve(x, s, {}) language.serve(x, s, {})
assert m >= len(s.getvalue()) assert m >= len(s.getvalue())