0

我使用 gorilla/websocket 作为 ws 和 labstack/echo 作为路由器。我需要为处理程序创建单元测试。我找到了使用默认 go 路由器解决此问题的主题，但我不明白如何将它与 echo 一起使用。我有这个：

func TestWS(t *testing.T){
    provider := handler.New(coordinateservice.New())

    e := echo.New()
    rec := httptest.NewRecorder()

    req := httptest.NewRequest(http.MethodGet, "/admin/orders/:id/details", nil)

    c := e.NewContext(req, rec)

    c.SetPath("/admin/orders/:id/details")
    c.SetParamNames("id")
    c.SetParamValues("9999")

    if assert.NoError(t, provider.OrderHandler.OpenWs(c)) {
        assert.Equal(t, http.StatusOK, rec.Code)
    }

    u := url.URL{Scheme: "ws", Host: "127.0.0.1", Path: "/admin/orders/9999/details"}

    fmt.Println(u.String())

    // Connect to the server
    ws, _, err := websocket.DefaultDialer.Dial(u.String(), nil)
    if err != nil {
        t.Fatalf("%v", err)
    }
    defer ws.Close()

    _, p, err := ws.ReadMessage()
    if err != nil {
        t.Fatalf("%v", err)
    }

    fmt.Println(string(p))
}

websocket: the client is not using the websocket protocol: 'upgrade' token not found in 'Connection' header此行中的错误：

    ws, _, err := websocket.DefaultDialer.Dial(u.String(), nil)
    if err != nil {
        t.Fatalf("%v", err)
    }

我需要做什么才能将 ws 连接到回声处理程序？

4

1 回答 1

0

这是我的 golang 单元测试 websocket示例。

我的httptest库的稳定性不保证，但是可以参考使用net.Pipe创建ws的连接。

使用回声：

func main() {
    app := echo.New()
    app.GET("/", hello)

    client := httptest.NewClient(app)
    go func() {
        // use http.Handler, if not has host.
        client.NewRequest("GET", "/example/wsio").WithWebsocket(handlerGobwasWebsocket).Do().Out()
        // use network
        client.NewRequest("GET", "http://localhost:8088/example/wsio").WithWebsocket(handlerGobwasWebsocket).Do().Out()
    }()

    app.Start(":8088")
}

func hello(c echo.Context) error {
    return c.String(http.StatusOK, "Hello, World!")
}

func handlerGobwasWebsocket(conn net.Conn) {
    go func() {
        wsutil.WriteClientBinary(conn, []byte("aaaaaa"))
        wsutil.WriteClientBinary(conn, []byte("bbbbbb"))
        wsutil.WriteClientBinary(conn, []byte("ccccc"))
    }()

    defer conn.Close()
    for {
        b, err := wsutil.ReadServerBinary(conn)
        fmt.Println("ws io client read: ", string(b), err)
        if err != nil {
            fmt.Println("gobwas client err:", err)
            return
        }
    }
}
于 2020-12-02T02:09:58.573 回答