Блог

Я использую 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 к обработчику echo?

Это мой пример веб-сокета для модульного тестирования golang.

Стабильность моей библиотеки httptest не гарантируется, но вы можете обратиться к using net.Pipe для создания соединения для ws.

используйте echo:

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