aboutsummaryrefslogtreecommitdiffstats
path: root/test/pathod
diff options
context:
space:
mode:
authorShadab Zafar <dufferzafar0@gmail.com>2016-06-05 22:59:57 +0530
committerMaximilian Hils <git@maximilianhils.com>2016-06-06 11:52:45 -0700
commitf5a6b295f7f7e5c4b1dad60e1ec3e7086364506e (patch)
tree65abac757ce50b9b4922293b8a0c8a354f42d3a0 /test/pathod
parentb86ef9fd0432a0b7774b4566f02721340124a7e9 (diff)
downloadmitmproxy-f5a6b295f7f7e5c4b1dad60e1ec3e7086364506e.tar.gz
mitmproxy-f5a6b295f7f7e5c4b1dad60e1ec3e7086364506e.tar.bz2
mitmproxy-f5a6b295f7f7e5c4b1dad60e1ec3e7086364506e.zip
Py3: Fix websocket tests
Diffstat (limited to 'test/pathod')
-rw-r--r--test/pathod/test_language_http.py20
1 files changed, 10 insertions, 10 deletions
diff --git a/test/pathod/test_language_http.py b/test/pathod/test_language_http.py
index dc05be5d..4825bc33 100644
--- a/test/pathod/test_language_http.py
+++ b/test/pathod/test_language_http.py
@@ -128,17 +128,17 @@ class TestRequest:
def test_websocket(self):
r = parse_request('ws:/path/')
res = r.resolve(language.Settings())
- assert res.method.string().lower() == "get"
- assert res.tok(http.Path).value.val == "/path/"
- assert res.tok(http.Method).value.val.lower() == "get"
- assert http.get_header("Upgrade", res.headers).value.val == "websocket"
+ assert res.method.string().lower() == b"get"
+ assert res.tok(http.Path).value.val == b"/path/"
+ assert res.tok(http.Method).value.val.lower() == b"get"
+ assert http.get_header(b"Upgrade", res.headers).value.val == b"websocket"
r = parse_request('ws:put:/path/')
res = r.resolve(language.Settings())
- assert r.method.string().lower() == "put"
- assert res.tok(http.Path).value.val == "/path/"
- assert res.tok(http.Method).value.val.lower() == "put"
- assert http.get_header("Upgrade", res.headers).value.val == "websocket"
+ assert r.method.string().lower() == b"put"
+ assert res.tok(http.Path).value.val == b"/path/"
+ assert res.tok(http.Method).value.val.lower() == b"put"
+ assert http.get_header(b"Upgrade", res.headers).value.val == b"websocket"
class TestResponse:
@@ -259,8 +259,8 @@ class TestResponse:
def test_websockets(self):
r = next(language.parse_pathod("ws"))
tutils.raises("no websocket key", r.resolve, language.Settings())
- res = r.resolve(language.Settings(websocket_key="foo"))
- assert res.status_code.string() == "101"
+ res = r.resolve(language.Settings(websocket_key=b"foo"))
+ assert res.status_code.string() == b"101"
def test_ctype_shortcut():