diff options
author | Shadab Zafar <dufferzafar0@gmail.com> | 2016-06-05 22:59:57 +0530 |
---|---|---|
committer | Maximilian Hils <git@maximilianhils.com> | 2016-06-06 11:52:45 -0700 |
commit | f5a6b295f7f7e5c4b1dad60e1ec3e7086364506e (patch) | |
tree | 65abac757ce50b9b4922293b8a0c8a354f42d3a0 | |
parent | b86ef9fd0432a0b7774b4566f02721340124a7e9 (diff) | |
download | mitmproxy-f5a6b295f7f7e5c4b1dad60e1ec3e7086364506e.tar.gz mitmproxy-f5a6b295f7f7e5c4b1dad60e1ec3e7086364506e.tar.bz2 mitmproxy-f5a6b295f7f7e5c4b1dad60e1ec3e7086364506e.zip |
Py3: Fix websocket tests
-rw-r--r-- | test/pathod/test_language_http.py | 20 |
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(): |