doukuang8166 2017-12-04 16:02
浏览 263
已采纳

在golang中为ws创建单元测试

I use the gorilla web socket framework and use the following client to run the web socket locally and debug it

https://github.com/gorilla/websocket

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

This is working when I use it in the chrome console but my question if there is a way to do some unit test in go and avoid using the chrome console?

  • 写回答

1条回答 默认 最新

  • douzi2785 2017-12-04 16:21
    关注

    Create a test server using the net/http/httptest package. Connect to that server using the Gorilla client. Read and write messages to test the connection.

    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")
            }
        }
    }
    
    本回答被题主选为最佳回答 , 对您是否有帮助呢?
    评论

报告相同问题?

悬赏问题

  • ¥15 彩灯控制电路,会的加我QQ1482956179
  • ¥200 相机拍直接转存到电脑上 立拍立穿无线局域网传
  • ¥15 (关键词-电路设计)
  • ¥15 如何解决MIPS计算是否溢出
  • ¥15 vue中我代理了iframe,iframe却走的是路由,没有显示该显示的网站,这个该如何处理
  • ¥15 操作系统相关算法中while();的含义
  • ¥15 CNVcaller安装后无法找到文件
  • ¥15 visual studio2022中文乱码无法解决
  • ¥15 关于华为5g模块mh5000-31接线问题
  • ¥15 keil L6007U报错