diff options
author | Thomas Kriechbaumer <Kriechi@users.noreply.github.com> | 2016-09-01 10:39:57 +0200 |
---|---|---|
committer | GitHub <noreply@github.com> | 2016-09-01 10:39:57 +0200 |
commit | 55d938b880fd861a22ac66da0da9a741bdd9abd5 (patch) | |
tree | d469bbd0dd5b1966591a332bf2094d4389100219 /test/netlib/websockets/test_masker.py | |
parent | 281d779fa3eb6b81ec76d046337275c0a82eff46 (diff) | |
parent | 0d0c2c788df4b60e951e6fcc13b479de8cec22c1 (diff) | |
download | mitmproxy-55d938b880fd861a22ac66da0da9a741bdd9abd5.tar.gz mitmproxy-55d938b880fd861a22ac66da0da9a741bdd9abd5.tar.bz2 mitmproxy-55d938b880fd861a22ac66da0da9a741bdd9abd5.zip |
Merge pull request #1488 from mitmproxy/websockets
add WebSockets support
Diffstat (limited to 'test/netlib/websockets/test_masker.py')
-rw-r--r-- | test/netlib/websockets/test_masker.py | 23 |
1 files changed, 23 insertions, 0 deletions
diff --git a/test/netlib/websockets/test_masker.py b/test/netlib/websockets/test_masker.py new file mode 100644 index 00000000..528fce71 --- /dev/null +++ b/test/netlib/websockets/test_masker.py @@ -0,0 +1,23 @@ +import codecs +import pytest + +from netlib import websockets + + +class TestMasker(object): + + @pytest.mark.parametrize("input,expected", [ + ([b"a"], '00'), + ([b"four"], '070d1616'), + ([b"fourf"], '070d161607'), + ([b"fourfive"], '070d1616070b1501'), + ([b"a", b"aasdfasdfa", b"asdf"], '000302170504021705040205120605'), + ([b"a" * 50, b"aasdfasdfa", b"asdf"], '00030205000302050003020500030205000302050003020500030205000302050003020500030205000302050003020500030205120605051206050500110702'), # noqa + ]) + def test_masker(self, input, expected): + m = websockets.Masker(b"abcd") + data = b"".join([m(t) for t in input]) + assert data == codecs.decode(expected, 'hex') + + data = websockets.Masker(b"abcd")(data) + assert data == b"".join(input) |