🧪 Add SSE test tools

2024-01-26 13:07:42 +08:00
parent 97df54a315
commit 450250c419
11 changed files with 138 additions and 25 deletions
--- a/test/data/sse/index.html
+++ b/test/data/sse/index.html
@ -0,0 +1,27 @@
+<!DOCTYPE html>
+<html lang="en">
+<head>
+  <meta charset="UTF-8">
+  <meta http-equiv="X-UA-Compatible" content="IE=edge">
+  <meta name="viewport" content="width=device-width, initial-scale=1.0">
+  <title>SSE Example</title>
+</head>
+<body>
+<h1>Server-Sent Events Example</h1>
+<div id="sse-data"></div>
+
+<script>
+  const eventSource = new EventSource("/sse")
+  const sseDataElement = document.getElementById("sse-data")
+
+  eventSource.onmessage = (event) => {
+    sseDataElement.innerText = `Data from server: ${event.data}`
+  }
+
+  eventSource.onerror = (error) => {
+    console.error("EventSource failed:", error)
+    eventSource.close()
+  }
+</script>
+</body>
+</html>
--- a/test/data/sse/server.py
+++ b/test/data/sse/server.py
@ -0,0 +1,28 @@
+import time
+
+from flask import Flask, render_template, Response
+
+app = Flask(__name__, template_folder=".")
+
+
+# Generator function to simulate real-time updates
+def event_stream():
+    count = 0
+    while True:
+        time.sleep(1)
+        count += 1
+        yield f"data: {count}\n\n"
+
+
+@app.route('/')
+def index():
+    return render_template('index.html')
+
+
+@app.route('/sse')
+def sse():
+    return Response(event_stream(), content_type='text/event-stream')
+
+
+if __name__ == '__main__':
+    app.run(debug=True, threaded=True)