diff options
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) |