websockets: specification of fin, mask, rsv1-3 bits.

1 files changed, 28 insertions, 4 deletions

diff --git a/test/test_language_websocket.py b/test/test_language_websocket.py
index f2f0b2a8..e313c591 100644
--- a/test/test_language_websocket.py
+++ b/test/test_language_websocket.py
@@ -15,7 +15,11 @@ class TestWebsocketFrame:
             "wf",
             "wf:b'foo'",
             "wf:cbinary",
-            "wf:c1"
+            "wf:c1",
+            "wf:r",
+            "wf:fin",
+            "wf:fin:rsv1:rsv2:rsv3:mask",
+            "wf:-fin:-rsv1:-rsv2:-rsv3:-mask",
         ]
         for i in specs:
             wf = parse_request(i)
@@ -28,12 +32,32 @@ class TestWebsocketFrame:
             wf2 = parse_request(spec)
             assert wf2.spec() == spec
 
+    def test_raw(self):
+        pass
+
+    def test_flags(self):
+        wf = parse_request("wf:fin:mask:rsv1:rsv2:rsv3")
+        frm = netlib.websockets.Frame.from_bytes(tutils.render(wf))
+        assert frm.header.fin
+        assert frm.header.mask
+        assert frm.header.rsv1
+        assert frm.header.rsv2
+        assert frm.header.rsv3
+
+        wf = parse_request("wf:-fin:-mask:-rsv1:-rsv2:-rsv3")
+        frm = netlib.websockets.Frame.from_bytes(tutils.render(wf))
+        assert not frm.header.fin
+        assert not frm.header.mask
+        assert not frm.header.rsv1
+        assert not frm.header.rsv2
+        assert not frm.header.rsv3
+
     def test_construction(self):
         wf = parse_request("wf:c1")
         frm = netlib.websockets.Frame.from_bytes(tutils.render(wf))
-        assert wf.code.value == 1 == frm.header.opcode
+        assert wf.opcode.value == 1 == frm.header.opcode
 
         wf = parse_request("wf:cbinary")
         frm = netlib.websockets.Frame.from_bytes(tutils.render(wf))
-        assert wf.code.value == frm.header.opcode
-        assert wf.code.value == netlib.websockets.OPCODE.BINARY
+        assert wf.opcode.value == frm.header.opcode
+        assert wf.opcode.value == netlib.websockets.OPCODE.BINARY