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