初始版本：事件回调式 WebSocket 客户端与服务端（RFC 6455）

2026-08-22 06:14:48 +08:00
commit 3ca5aff516
11 changed files with 1799 additions and 0 deletions
@@ -0,0 +1,237 @@
+/*
+ * simcu::websocket.tests —— 端到端集成测试（cjpm test）。
+ * 覆盖：文本回声 / 广播 / 关闭握手（客户端主动、服务端主动）/ terminate /
+ *       ping 心跳后连接仍可用 / 路径过滤 / maxPayload 超限(1009)。
+ *
+ * 运行：在 websocket-cj 目录执行 cjpm test
+ */
+package simcu::websocket.tests
+
+import std.time.*
+import std.unittest.*
+import std.unittest.testmacro.Expect
+import std.unittest.testmacro.TestCase
+import std.unittest.testmacro.Test
+
+import simcu::websocket.client.WebSocketClient
+import simcu::websocket.common.MessageType
+import simcu::websocket.common.WebSocketException
+import simcu::websocket.common.WebSocketMessage
+import simcu::websocket.server.WebSocketConnection
+import simcu::websocket.server.WebSocketServer
+
+/// 测试回收集合：把回调中捕获的值存到独立实例字段，避免闭包捕获歧义。
+class WsTestHarness {
+    var serverConn: ?WebSocketConnection = None
+    var connMessage: ?String = None
+    var connErrorCode: Int64 = 0
+    var connCloseCode: Int64 = -1
+    var connCloseReason: String = ""
+    var clientMessage: ?String = None
+    var clientCloseCode: Int64 = -1
+    var clientCloseReason: String = ""
+    var clientErrorCount: Int64 = 0
+}
+
+/// 轮询等待条件成立（每 50ms 检查一次）。
+private func waitUntil(timeoutMs: Int64, condition: () -> Bool): Bool {
+    let total = timeoutMs / 50
+    for (_ in 0..total) {
+        if (condition()) {
+            return true
+        }
+        sleep(Duration.millisecond * 50)
+    }
+    condition()
+}
+
+@Test
+public class WebSocketTest {
+    /// 1. 文本回声：client.sendText → server 收到后回发 → client 收到
+    @TestCase
+    public func echoText(): Unit {
+        let h = WsTestHarness()
+        let server = WebSocketServer(bindAt: 0)
+        server.on("connection", { conn: WebSocketConnection =>
+            h.serverConn = Some(conn)
+            conn.on("message", { m: WebSocketMessage =>
+                h.connMessage = m.text.getOrThrow()
+                conn.sendText("echo: ${m.text.getOrThrow()}")
+            })
+        })
+        server.listen()
+        let client = WebSocketClient("127.0.0.1", server.localPort)
+        client.onMessage = { m: WebSocketMessage =>
+            h.clientMessage = m.text.getOrThrow()
+        }
+        client.connect()
+        client.sendText("hello")
+
+        @Expect(waitUntil(5000) { => h.connMessage == Some("hello") }, true)
+        @Expect(waitUntil(5000) { => h.clientMessage == Some("echo: hello") }, true)
+        client.close()
+        server.close()
+    }
+
+    /// 2. 广播：server.broadcastText 两个客户端都收到
+    @TestCase
+    public func broadcast(): Unit {
+        let h1 = WsTestHarness()
+        let h2 = WsTestHarness()
+        let server = WebSocketServer(bindAt: 0)
+        server.listen()
+
+        let c1 = WebSocketClient("127.0.0.1", server.localPort)
+        c1.onMessage = { m: WebSocketMessage =>
+            h1.clientMessage = m.text.getOrThrow()
+        }
+        c1.connect()
+        let c2 = WebSocketClient("127.0.0.1", server.localPort)
+        c2.onMessage = { m: WebSocketMessage =>
+            h2.clientMessage = m.text.getOrThrow()
+        }
+        c2.connect()
+        @Expect(waitUntil(5000) { => server.connectionCount == 2 }, true)
+
+        server.broadcastText("hi-all")
+        @Expect(waitUntil(5000) { => h1.clientMessage == Some("hi-all") }, true)
+        @Expect(waitUntil(5000) { => h2.clientMessage == Some("hi-all") }, true)
+        c1.close()
+        c2.close()
+        server.close()
+    }
+
+    /// 3. 关闭握手：client.close(1000, "bye") → 双端 onClose 都收到 (1000, "bye")
+    @TestCase
+    public func closeHandshake(): Unit {
+        let h = WsTestHarness()
+        let server = WebSocketServer(bindAt: 0)
+        server.on("connection", { conn: WebSocketConnection =>
+            conn.on("close", { code: Int64, reason: String =>
+                h.connCloseCode = code
+                h.connCloseReason = reason
+            })
+        })
+        server.listen()
+        let client = WebSocketClient("127.0.0.1", server.localPort)
+        client.onClose = { code: Int64, reason: String =>
+            h.clientCloseCode = code
+            h.clientCloseReason = reason
+        }
+        client.connect()
+        client.close(code: 1000, reason: "bye")
+
+        @Expect(waitUntil(5000) { => h.clientCloseCode == 1000 && h.clientCloseReason == "bye" }, true)
+        @Expect(waitUntil(5000) { => h.connCloseCode == 1000 && h.connCloseReason == "bye" }, true)
+        server.close()
+    }
+
+    /// 4. 服务端主动关闭：conn.close(4000, "svr") → client onClose 收 (4000, "svr")
+    @TestCase
+    public func serverClose(): Unit {
+        let h = WsTestHarness()
+        let server = WebSocketServer(bindAt: 0)
+        server.on("connection", { conn: WebSocketConnection =>
+            h.serverConn = Some(conn)
+            conn.close(code: 4000, reason: "svr")
+        })
+        server.listen()
+        let client = WebSocketClient("127.0.0.1", server.localPort)
+        client.onClose = { code: Int64, reason: String =>
+            h.clientCloseCode = code
+            h.clientCloseReason = reason
+        }
+        client.connect()
+
+        @Expect(waitUntil(5000) { => h.clientCloseCode == 4000 && h.clientCloseReason == "svr" }, true)
+        server.close()
+    }
+
+    /// 5. terminate：client.terminate() 立即断开 → server conn onClose 收 (1006, "")
+    @TestCase
+    public func terminate(): Unit {
+        let h = WsTestHarness()
+        let server = WebSocketServer(bindAt: 0)
+        server.on("connection", { conn: WebSocketConnection =>
+            conn.on("close", { code: Int64, reason: String =>
+                h.connCloseCode = code
+                h.connCloseReason = reason
+            })
+        })
+        server.listen()
+        let client = WebSocketClient("127.0.0.1", server.localPort)
+        client.connect()
+        client.terminate()
+
+        @Expect(waitUntil(5000) { => h.connCloseCode == 1006 && h.connCloseReason == "" }, true)
+        server.close()
+    }
+
+    /// 6. ping：心跳后连接仍可用，后续消息正常到达
+    @TestCase
+    public func ping(): Unit {
+        let h = WsTestHarness()
+        let server = WebSocketServer(bindAt: 0)
+        server.on("connection", { conn: WebSocketConnection =>
+            conn.on("message", { m: WebSocketMessage =>
+                h.connMessage = m.text.getOrThrow()
+            })
+        })
+        server.listen()
+        let client = WebSocketClient("127.0.0.1", server.localPort)
+        client.connect()
+
+        client.ping()
+        sleep(Duration.millisecond * 100)
+        client.sendText("after-ping")
+        @Expect(waitUntil(5000) { => h.connMessage == Some("after-ping") }, true)
+        client.close()
+        server.close()
+    }
+
+    /// 7. 路径过滤：path=Some("/ws") 时 /ws 握手成功，/other 被拒（connect 抛异常）
+    @TestCase
+    public func pathFilter(): Unit {
+        let server = WebSocketServer(bindAt: 0, path: Some("/ws"))
+        server.listen()
+
+        let ok = WebSocketClient("127.0.0.1", server.localPort, path: "/ws")
+        ok.connect()
+        ok.close()
+
+        let bad = WebSocketClient("127.0.0.1", server.localPort, path: "/other")
+        var rejected = false
+        try {
+            bad.connect()
+        } catch (e: WebSocketException) {
+            rejected = true
+        }
+        @Expect(rejected, true)
+        server.close()
+    }
+
+    /// 8. maxPayload：server 限制 10 字节，client 发长文本 → server 连接 onError 收 code=1009
+    @TestCase
+    public func maxPayload(): Unit {
+        let h = WsTestHarness()
+        let server = WebSocketServer(bindAt: 0, maxPayload: 10)
+        server.on("connection", { conn: WebSocketConnection =>
+            conn.on("error", { e: Exception =>
+                if (let we: WebSocketException <- e) {
+                    h.connErrorCode = we.code
+                }
+            })
+        })
+        server.listen()
+        let client = WebSocketClient("127.0.0.1", server.localPort)
+        client.onError = { e: Exception =>
+            h.clientErrorCount += 1
+        }
+        client.connect()
+        client.sendText("this message is way too long")
+
+        @Expect(waitUntil(5000) { => h.connErrorCode == 1009 }, true)
+        client.terminate()
+        server.close()
+    }
+}