测试 websocket 连接在 golang 中可通过 gorilla/websocket 库实现。1. 引入 gorilla/websocket 包并确保服务运行；2. 使用 dialer 建立连接并发送消息；3. 读取响应验证是否符合预期；4. 可通过协程模拟多客户端并发测试；常见问题包括服务未启动、协议错误、缓冲区不足等，可通过调整配置解决。
测试 WebSocket 连接在 Golang 中其实并不复杂，尤其是使用 gorilla/websocket 这个库之后，很多底层细节都被封装好了。我们可以通过编写一个简单的客户端模拟连接，并与服务端进行交互，来验证连接是否正常。
下面是一个基本的测试流程，适合刚接触 WebSocket 测试的朋友。
准备工作：引入 gorilla/websocket
首先确保你已经安装了 gorilla/websocket 包：
立即学习“go语言免费学习笔记（深入）”；
go get github.com/gorilla/websocket
然后就可以开始写测试代码了。
编写测试代码的基本结构
测试 WebSocket 的核心在于建立连接、发送消息、接收响应，并判断是否符合预期。
以下是一个简单的测试模板：
package main
import (
"fmt"
"testing"
"github.com/gorilla/websocket"
)
var upgrader = websocket.Upgrader{
ReadBufferSize: 1024,
WriteBufferSize: 1024,
}
func TestWebSocketConnection(t *testing.T) {
// 模拟服务器地址（假设运行在本地）
url := "ws://localhost:8080/ws"
// 建立连接
conn, _, err := websocket.DefaultDialer.Dial(url, nil)
if err != nil {
t.Fatalf("连接失败: %v", err)
}
defer conn.Close()
// 发送一条测试消息
msg := []byte("hello")
err = conn.WriteMessage(websocket.TextMessage, msg)
if err != nil {
t.Errorf("发送消息失败: %v", err)
}
// 接收返回的消息
_, resp, err := conn.ReadMessage()
if err != nil {
t.Errorf("读取消息失败: %v", err)
}
// 验证返回内容
expected := []byte("echo: hello")
if string(resp) != string(expected) {
t.Errorf("期望得到 %s，实际得到 %s", expected, resp)
}
}这段代码会尝试连接到指定的 WebSocket 地址，发送一条消息，并等待返回结果。
常见问题与调试技巧
有时候测试不通过，可能是因为以下几个原因：
- 服务未启动：确保你的 WebSocket 服务已经在目标地址运行。
- 跨域限制：如果你是用浏览器测试，可能会遇到 CORS 问题；而用 Go 直接连接则不受影响。
-
协议错误：确认 URL 是以
ws://或wss://开头。 -
缓冲区大小不够：如果消息较大，可以适当调高
ReadBufferSize和WriteBufferSize。 - 连接超时：默认的 Dialer 超时时间较短，你可以自定义设置：
dialer := websocket.Dialer{
HandshakeTimeout: 5 * time.Second,
}如何模拟多个客户端并发测试？
如果你需要测试多个客户端同时连接的情况，可以用 Go 协程实现：
func simulateClient(t *testing.T) {
conn, _, err := websocket.DefaultDialer.Dial("ws://localhost:8080/ws", nil)
if err != nil {
t.Error(err)
return
}
defer conn.Close()
conn.WriteMessage(websocket.TextMessage, []byte("test"))
_, resp, _ := conn.ReadMessage()
fmt.Println(string(resp))
}
func TestMultipleClients(t *testing.T) {
for i := 0; i < 10; i++ {
go simulateClient(t)
}
time.Sleep(3 * time.Second) // 等待执行完成
}这样可以快速模拟多用户接入场景，看看服务端是否能处理并发连接和消息。
基本上就这些。WebSocket 测试的核心就是“连得上、发得出、收得到”，剩下的就是根据业务逻辑添加更多断言和测试用例了。
