diff options
| author | Aldo Cortesi <aldo@nullcube.com> | 2015-05-01 10:09:35 +1200 |
|---|---|---|
| committer | Aldo Cortesi <aldo@nullcube.com> | 2015-05-01 10:09:35 +1200 |
| commit | 7d9e38ffb10e92b5127f203c2d8a524da8698b00 (patch) | |
| tree | 7a3f48e50154c7d830c0e06f5e485e2cbe25d07e /test/test_websockets.py | |
| parent | 4dce7ee074c242f5b6530ff64879875d98c1d255 (diff) | |
| download | mitmproxy-7d9e38ffb10e92b5127f203c2d8a524da8698b00.tar.gz mitmproxy-7d9e38ffb10e92b5127f203c2d8a524da8698b00.tar.bz2 mitmproxy-7d9e38ffb10e92b5127f203c2d8a524da8698b00.zip | |
websockets: A progressive masker.
Diffstat (limited to 'test/test_websockets.py')
| -rw-r--r-- | test/test_websockets.py | 16 |
1 files changed, 16 insertions, 0 deletions
diff --git a/test/test_websockets.py b/test/test_websockets.py index d8e56a8f..428f7c61 100644 --- a/test/test_websockets.py +++ b/test/test_websockets.py @@ -232,3 +232,19 @@ class TestFrame: def test_human_readable(self): f = websockets.Frame() assert f.human_readable() + + +def test_masker(): + tests = [ + ["a"], + ["four"], + ["fourf"], + ["fourfive"], + ["a", "aasdfasdfa", "asdf"], + ["a"*50, "aasdfasdfa", "asdf"], + ] + for i in tests: + m = websockets.Masker("abcd") + data = "".join([m(t) for t in i]) + data2 = websockets.Masker("abcd")(data) + assert data2 == "".join(i) |