1 files changed, 24 insertions, 0 deletions
diff --git a/dom/websocket/tests/file_websocket_permessage_deflate_rejected_wsh.py b/dom/websocket/tests/file_websocket_permessage_deflate_rejected_wsh.py
new file mode 100644
index 0000000000..b67627d67c
--- /dev/null
+++ b/dom/websocket/tests/file_websocket_permessage_deflate_rejected_wsh.py
@@ -0,0 +1,24 @@
+from mod_pywebsocket import common, msgutil
+
+
+def web_socket_do_extra_handshake(request):
+    deflate_removed = False
+
+    if request.ws_extension_processors is not None:
+        for extension_processor in request.ws_extension_processors:
+            if extension_processor.name() == "deflate":
+                request.ws_extension_processors.remove(extension_processor)
+                deflate_removed = True
+
+    if deflate_removed is False:
+        raise ValueError("deflate extension processor not found")
+
+
+def web_socket_transfer_data(request):
+    while True:
+        rcvd = msgutil.receive_message(request)
+        opcode = request.ws_stream.get_last_received_opcode()
+        if opcode == common.OPCODE_BINARY:
+            msgutil.send_message(request, rcvd, binary=True)
+        elif opcode == common.OPCODE_TEXT:
+            msgutil.send_message(request, rcvd)