websockets: nicer frame construction

- Resolve unspecified values on instantiation
- Add a check for masking key length
- Smarter resolution for masking_key and mask values. Do the right thing unless told not to.

1 files changed, 12 insertions, 0 deletions

diff --git a/test/test_websockets.py b/test/test_websockets.py
index 8ae18edd..4b286b6f 100644
--- a/test/test_websockets.py
+++ b/test/test_websockets.py
@@ -192,6 +192,18 @@ class TestFrameHeader:
 
     def test_violations(self):
         tutils.raises("opcode", websockets.FrameHeader, opcode=17)
+        tutils.raises("masking key", websockets.FrameHeader, masking_key="x")
+
+    def test_automask(self):
+        f = websockets.FrameHeader(mask=True)
+        assert f.masking_key
+
+        f = websockets.FrameHeader(masking_key="foob")
+        assert f.mask
+
+        f = websockets.FrameHeader(masking_key="foob", mask=0)
+        assert not f.mask
+        assert f.masking_key
 
 
 class TestFrame: