在golang中为ws创建单元测试

发布时间：2020-12-16 09:28:14 所属栏目：大数据来源：网络整理

导读：我使用gorilla Web套接字框架并使用以下客户端在本地运行Web套接字并进行调试 https://github.com/gorilla/websocket ws = new WebSocket("ws://localhost:8080/mypath")ws.onmessage = function(ev) { console.log(ev.data) }ws.send("hello") 当我在chrome

我使用gorilla Web套接字框架并使用以下客户端在本地运行Web套接字并进行调试

https://github.com/gorilla/websocket

ws = new WebSocket("ws://localhost:8080/mypath")
ws.onmessage = function(ev) { console.log(ev.data) }
ws.send("hello")

当我在chrome控制台中使用它时,这是有效的,但我的问题是,有没有办法在go中进行一些单元测试并避免使用chrome控制台？

解决方法

使用 net/http/httptest包创建 test server.使用Gorilla客户端连接到该服务器.读取和写入消息以测试连接.

package main

import (
    "net/http"
    "net/http/httptest"
    "strings"
    "testing"

    "github.com/gorilla/websocket"
)

var upgrader = websocket.Upgrader{}

func echo(w http.ResponseWriter,r *http.Request) {
    c,err := upgrader.Upgrade(w,r,nil)
    if err != nil {
        return
    }
    defer c.Close()
    for {
        mt,message,err := c.ReadMessage()
        if err != nil {
            break
        }
        err = c.WriteMessage(mt,message)
        if err != nil {
            break
        }
    }
}

func TestExample(t *testing.T) {
    // Create test server with the echo handler.
    s := httptest.NewServer(http.HandlerFunc(echo))
    defer s.Close()

    // Convert http://127.0.0.1 to ws://127.0.0.
    u := "ws" + strings.TrimPrefix(s.URL,"http")

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

    // Send message to server,read response and check to see if it's what we expect.
    for i := 0; i < 10; i++ {
        if err := ws.WriteMessage(websocket.TextMessage,[]byte("hello")); err != nil {
            t.Fatalf("%v",err)
        }
        _,p,err := ws.ReadMessage()
        if err != nil {
            t.Fatalf("%v",err)
        }
        if string(p) != "hello" {
            t.Fatalf("bad message")
        }
    }
}

（编辑：李大同）

【声明】本站内容均来自网络，其相关言论仅代表作者个人观点，不代表本站立场。若无意侵犯到您的权利，请及时与联系站长删除相关内容!