Add test for websocket response with request larger than read BUFSIZE

circuits · Nov 4, 2016 · 5290d8c · 5290d8c
1 parent 259c7e5
commit 5290d8c
Showing 1 changed file with 7 additions and 1 deletion.
diff --git a/tests/web/test_websockets.py b/tests/web/test_websockets.py
@@ -6,7 +6,7 @@
 
 from circuits import Component
 from circuits.web.controllers import Controller
-from circuits.net.sockets import close, write
+from circuits.net.sockets import close, write, BUFSIZE
 from circuits.web.websockets import WebSocketClient, WebSocketsDispatcher
 
 
@@ -81,6 +81,12 @@ def test(manager, watcher, webapp):
     assert watcher.wait("read", channel="ws")
     assert client.response == "Received: Hello!"
 
+    for size in (BUFSIZE, BUFSIZE + 1, BUFSZIE * 2):
+        data = "A" * (size + 1)
+        client.fire(write(data), "ws")
+        assert watcher.wait("read", channel="ws")
+        assert client.response == "Received: %s" % (data,)
+
     f = urlopen(webapp.server.http.base)
     s = f.read()
     assert s == b"Hello World!"