diff options
author | Shadab Zafar <dufferzafar0@gmail.com> | 2016-06-06 00:53:03 +0530 |
---|---|---|
committer | Maximilian Hils <git@maximilianhils.com> | 2016-06-06 11:48:58 -0700 |
commit | ced6fb4cf9ae404d389bc6adad29f4f814fecd71 (patch) | |
tree | ddde59c3bd4024717155cac797902231b984b544 /test/pathod | |
parent | efb0f7133cea4e99ea514a93d291f7dc10ba7edc (diff) | |
download | mitmproxy-ced6fb4cf9ae404d389bc6adad29f4f814fecd71.tar.gz mitmproxy-ced6fb4cf9ae404d389bc6adad29f4f814fecd71.tar.bz2 mitmproxy-ced6fb4cf9ae404d389bc6adad29f4f814fecd71.zip |
Py3: Fix tests by using byte literals
Diffstat (limited to 'test/pathod')
-rw-r--r-- | test/pathod/test_language_websocket.py | 12 |
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'") |