From 5b6d3a80bbb57faa1de1cc7730d28b0678b0bdee Mon Sep 17 00:00:00 2001
From: Aldo Cortesi <aldo@nullcube.com>
Date: Sun, 3 May 2015 13:54:52 +1200
Subject: websockets: code specification

---
 test/test_language_websocket.py | 16 +++++++++++++++-
 1 file changed, 15 insertions(+), 1 deletion(-)

(limited to 'test/test_language_websocket.py')

diff --git a/test/test_language_websocket.py b/test/test_language_websocket.py
index 3b1d4852..f2f0b2a8 100644
--- a/test/test_language_websocket.py
+++ b/test/test_language_websocket.py
@@ -1,6 +1,8 @@
 
 from libpathod import language
 from libpathod.language import websockets
+import netlib.websockets
+import tutils
 
 
 def parse_request(s):
@@ -11,7 +13,9 @@ class TestWebsocketFrame:
     def test_values(self):
         specs = [
             "wf",
-            "wf:b'foo'"
+            "wf:b'foo'",
+            "wf:cbinary",
+            "wf:c1"
         ]
         for i in specs:
             wf = parse_request(i)
@@ -23,3 +27,13 @@ class TestWebsocketFrame:
             spec = wf.spec()
             wf2 = parse_request(spec)
             assert wf2.spec() == spec
+
+    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
+
+        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
-- 
cgit v1.2.3