aboutsummaryrefslogtreecommitdiffstats
path: root/test/test_websockets.py
diff options
context:
space:
mode:
Diffstat (limited to 'test/test_websockets.py')
-rw-r--r--test/test_websockets.py16
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)