1 files changed, 49 insertions, 0 deletions
diff --git a/testing/web-platform/tests/tools/third_party/websockets/example/deployment/kubernetes/app.py b/testing/web-platform/tests/tools/third_party/websockets/example/deployment/kubernetes/app.py
new file mode 100644
index 0000000000..a8bcef6881
--- /dev/null
+++ b/testing/web-platform/tests/tools/third_party/websockets/example/deployment/kubernetes/app.py
@@ -0,0 +1,49 @@
+#!/usr/bin/env python
+
+import asyncio
+import http
+import signal
+import sys
+import time
+
+import websockets
+
+
+async def slow_echo(websocket):
+    async for message in websocket:
+        # Block the event loop! This allows saturating a single asyncio
+        # process without opening an impractical number of connections.
+        time.sleep(0.1)  # 100ms
+        await websocket.send(message)
+
+
+async def health_check(path, request_headers):
+    if path == "/healthz":
+        return http.HTTPStatus.OK, [], b"OK\n"
+    if path == "/inemuri":
+        loop = asyncio.get_running_loop()
+        loop.call_later(1, time.sleep, 10)
+        return http.HTTPStatus.OK, [], b"Sleeping for 10s\n"
+    if path == "/seppuku":
+        loop = asyncio.get_running_loop()
+        loop.call_later(1, sys.exit, 69)
+        return http.HTTPStatus.OK, [], b"Terminating\n"
+
+
+async def main():
+    # Set the stop condition when receiving SIGTERM.
+    loop = asyncio.get_running_loop()
+    stop = loop.create_future()
+    loop.add_signal_handler(signal.SIGTERM, stop.set_result, None)
+
+    async with websockets.serve(
+        slow_echo,
+        host="",
+        port=80,
+        process_request=health_check,
+    ):
+        await stop
+
+
+if __name__ == "__main__":
+    asyncio.run(main())