From bddf81edfc07aab065c12b245665b613d2f61e13 Mon Sep 17 00:00:00 2001
From: Aldo Cortesi <aldo@nullcube.com>
Date: Sun, 17 May 2015 11:31:02 +1200
Subject: websockets: lINTEGER to specify payload length

---
 test/test_language_websocket.py | 8 ++++++++
 1 file changed, 8 insertions(+)

(limited to 'test/test_language_websocket.py')

diff --git a/test/test_language_websocket.py b/test/test_language_websocket.py
index f9e8310a..0228c350 100644
--- a/test/test_language_websocket.py
+++ b/test/test_language_websocket.py
@@ -15,6 +15,7 @@ class TestWebsocketFrame:
             "wf",
             "wf:dr",
             "wf:b'foo'",
+            "wf:l1024:b'foo'",
             "wf:cbinary",
             "wf:c1",
             "wf:mask:knone",
@@ -106,3 +107,10 @@ class TestWebsocketFrame:
             self.fr,
             "wf:b'foo':mask:knone",
         )
+
+    def test_length(self):
+        assert self.fr("wf:l3:b'foo'").header.payload_length == 3
+        frm = self.fr("wf:l2:b'foo'")
+        assert frm.header.payload_length == 2
+        assert frm.payload == "fo"
+        tutils.raises("expected 1024 bytes", self.fr, "wf:l1024:b'foo'")
-- 
cgit v1.2.3