使用 Echo 路由器在 golang 中为 WS 创建单元测试

学习知识要善于思考，思考，再思考！今天golang学习网小编就给大家带来《使用 Echo 路由器在 golang 中为 WS 创建单元测试》，以下内容主要包含等知识点，如果你正在学习或准备学习Golang，就都不要错过本文啦~让我们一起来看看吧，能帮助到你就更好了！

我使用 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：客户端未使用 websocket 协议：在这一行的“连接”标头 中找不到“升级”令牌：

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

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

解决方案

这是我的 golang 单元测试 websocket example。

我的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
        }
    }
}

以上就是本文的全部内容了，是否有顺利帮助你解决问题？若是能给你带来学习上的帮助，请大家多多支持golang学习网！更多关于Golang的相关知识，也可关注golang学习网公众号。