1 files changed, 27 insertions, 0 deletions
diff --git a/dom/websocket/tests/websocket_hybi/file_check-binary-messages_wsh.py b/dom/websocket/tests/websocket_hybi/file_check-binary-messages_wsh.py
new file mode 100644
index 0000000000..d72ec1f462
--- /dev/null
+++ b/dom/websocket/tests/websocket_hybi/file_check-binary-messages_wsh.py
@@ -0,0 +1,27 @@
+import six
+from mod_pywebsocket import msgutil
+
+
+def web_socket_do_extra_handshake(request):
+    pass  # Always accept.
+
+
+def web_socket_transfer_data(request):
+    expected_messages = ["Hello, world!", "", all_distinct_bytes()]
+
+    for test_number, expected_message in enumerate(expected_messages):
+        expected_message = six.b(expected_message)
+        message = msgutil.receive_message(request)
+        if message == expected_message:
+            msgutil.send_message(request, "PASS: Message #{:d}.".format(test_number))
+        else:
+            msgutil.send_message(
+                request,
+                "FAIL: Message #{:d}: Received unexpected message: {!r}".format(
+                    test_number, message
+                ),
+            )
+
+
+def all_distinct_bytes():
+    return "".join([chr(i) for i in range(256)])