50 lines
1.0 KiB
Go
50 lines
1.0 KiB
Go
package service
|
|
|
|
import (
|
|
"github.com/gorilla/websocket"
|
|
"net/http/httptest"
|
|
"strings"
|
|
"testing"
|
|
)
|
|
|
|
func TestWebSocketService(t *testing.T) {
|
|
// 注册 WebSocket 服务
|
|
RegisterWebsocket(0, "/ws", func(conn *websocket.Conn) {
|
|
for {
|
|
var msg Map
|
|
if err := conn.ReadJSON(&msg); err != nil {
|
|
break
|
|
}
|
|
_ = conn.WriteJSON(Map{"reply": msg["msg"]})
|
|
}
|
|
}, "test websocket")
|
|
|
|
// 启动测试服务器
|
|
server := httptest.NewServer(&routeHandler{})
|
|
defer server.Close()
|
|
|
|
// 建立连接
|
|
wsURL := "ws" + strings.TrimPrefix(server.URL, "http") + "/ws"
|
|
conn, _, err := websocket.DefaultDialer.Dial(wsURL, nil)
|
|
if err != nil {
|
|
t.Fatalf("Dial failed: %v", err)
|
|
}
|
|
defer conn.Close()
|
|
|
|
// 发送消息
|
|
msg := Map{"action": "echo", "msg": "hello"}
|
|
if err := conn.WriteJSON(msg); err != nil {
|
|
t.Fatalf("WriteJSON failed: %v", err)
|
|
}
|
|
|
|
// 接收响应
|
|
var reply Map
|
|
if err := conn.ReadJSON(&reply); err != nil {
|
|
t.Fatalf("ReadJSON failed: %v", err)
|
|
}
|
|
|
|
if reply["reply"] != "hello" {
|
|
t.Errorf("Reply mismatch: %v", reply)
|
|
}
|
|
}
|