aboutsummaryrefslogtreecommitdiffstats
diff options
context:
space:
mode:
authorShadab Zafar <dufferzafar0@gmail.com>2016-06-06 00:53:03 +0530
committerMaximilian Hils <git@maximilianhils.com>2016-06-06 11:48:58 -0700
commitced6fb4cf9ae404d389bc6adad29f4f814fecd71 (patch)
treeddde59c3bd4024717155cac797902231b984b544
parentefb0f7133cea4e99ea514a93d291f7dc10ba7edc (diff)
downloadmitmproxy-ced6fb4cf9ae404d389bc6adad29f4f814fecd71.tar.gz
mitmproxy-ced6fb4cf9ae404d389bc6adad29f4f814fecd71.tar.bz2
mitmproxy-ced6fb4cf9ae404d389bc6adad29f4f814fecd71.zip
Py3: Fix tests by using byte literals
-rw-r--r--test/pathod/test_language_websocket.py12
1 files changed, 6 insertions, 6 deletions
diff --git a/test/pathod/test_language_websocket.py b/test/pathod/test_language_websocket.py
index 57ba9600..58297141 100644
--- a/test/pathod/test_language_websocket.py
+++ b/test/pathod/test_language_websocket.py
@@ -93,9 +93,9 @@ class TestWebsocketFrame:
def test_rawbody(self):
frm = self.fr("wf:mask:r'foo'")
assert len(frm.payload) == 3
- assert frm.payload != "foo"
+ assert frm.payload != b"foo"
- assert self.fr("wf:r'foo'").payload == "foo"
+ assert self.fr("wf:r'foo'").payload == b"foo"
def test_construction_2(self):
# Simple server frame
@@ -109,7 +109,7 @@ class TestWebsocketFrame:
assert frm.header.masking_key
frm = self.fr("wf:b'foo':k'abcd'", is_client=True)
assert frm.header.mask
- assert frm.header.masking_key == 'abcd'
+ assert frm.header.masking_key == b'abcd'
# Server frame, mask explicitly set
frm = self.fr("wf:b'foo':mask")
@@ -117,7 +117,7 @@ class TestWebsocketFrame:
assert frm.header.masking_key
frm = self.fr("wf:b'foo':k'abcd'")
assert frm.header.mask
- assert frm.header.masking_key == 'abcd'
+ assert frm.header.masking_key == b'abcd'
# Client frame, mask explicitly unset
frm = self.fr("wf:b'foo':-mask", is_client=True)
@@ -128,7 +128,7 @@ class TestWebsocketFrame:
assert not frm.header.mask
# We're reading back a corrupted frame - the first 3 characters of the
# mask is mis-interpreted as the payload
- assert frm.payload == "abc"
+ assert frm.payload == b"abc"
def test_knone(self):
with tutils.raises("expected 4 bytes"):
@@ -138,5 +138,5 @@ class TestWebsocketFrame:
assert self.fr("wf:l3:b'foo'").header.payload_length == 3
frm = self.fr("wf:l2:b'foo'")
assert frm.header.payload_length == 2
- assert frm.payload == "fo"
+ assert frm.payload == b"fo"
tutils.raises("expected 1024 bytes", self.fr, "wf:l1024:b'foo'")