From d97e3a3296db77f6a33ce010a66d2a0b2d4bac49 Mon Sep 17 00:00:00 2001
From: Max Romanov <max.romanov@nginx.com>
Date: Thu, 1 Oct 2020 23:55:35 +0300
Subject: Tests: added ASGI WebSocket.

---
 test/python/websockets/subprotocol/asgi.py | 25 +++++++++++++++++++++++++
 1 file changed, 25 insertions(+)
 create mode 100644 test/python/websockets/subprotocol/asgi.py

(limited to 'test/python/websockets/subprotocol')

diff --git a/test/python/websockets/subprotocol/asgi.py b/test/python/websockets/subprotocol/asgi.py
new file mode 100644
index 00000000..92263dd7
--- /dev/null
+++ b/test/python/websockets/subprotocol/asgi.py
@@ -0,0 +1,25 @@
+async def application(scope, receive, send):
+    assert scope['type'] == 'websocket'
+
+    while True:
+        m = await receive()
+        if m['type'] == 'websocket.connect':
+            subprotocols = scope['subprotocols']
+
+            await send({
+              'type': 'websocket.accept',
+              'headers': [
+                  (b'x-subprotocols', str(subprotocols).encode()),
+              ],
+              'subprotocol': subprotocols[0],
+            })
+
+        if m['type'] == 'websocket.receive':
+            await send({
+              'type': 'websocket.send',
+              'bytes': m.get('bytes', None),
+              'text': m.get('text', None),
+            })
+
+        if m['type'] == 'websocket.disconnect':
+            break;
-- 
cgit