Question

我有一个WebSocket服务器，我正在尝试正确测试它的一些功能。我有以下情况：

我接受WebSocket连接并在新连接上调用频道registerConn <-以通知type hub struct。这是中心：

type hub struct {
    clients map[client]bool

    registerConn chan client

    // some other fields below...  
}
// This function runs in its own thread forever
func (h *hub) run() {
     for {
           select{
           // A client connects on this channel
           case client := <- h.registerConn:
                  h.clients[client] = true
           }
     }
}

现在，我想在hub_test.go中测试此功能：

func TestRegisterClientWSConnections(t *testing.T){
    for _, cl := range testClients {
        thub.registerConn <- cl
    }

    // TODO: Is this a good way to test?
    time.Sleep(1 * time.Second)

    // I want to know if the testClients have been added to my clients map
    for _, cl := range testClients {
        if thub.clients[cl] == false {
            t.Error("Client ", cl, " not found in the registered clients")
        }
    }
}

由于集线器上的run()函数在后台线程中运行，后台线程中的客户端（第一个for循环）的注册在主要进行检查（第二个for循环）之前没有完成线程并因此失败。

解决方法是添加time.Sleep()以等待注册完成。其他解决方法是添加一个通道以通知测试完成添加。

我不想仅为测试添加新频道，因为这会导致不必要的代码。另一方面，在测试中使用time.Sleep()似乎不是一个好习惯。（或者是吗？）

我可以在哪些方面测试此案例？

Answer 1

与建议的答案一样，Gomega

有一个非常优雅的解决方案

测试现在看起来像这样：

编辑：

func TestRegisterClientsFromWSConnection(t *testing.T){
    g := NewGomegaWithT(t)

    for _, cl := range testClients {
        thub.registerConn <- cl
    }

    g.Eventually(thub.clients).Should(HaveLen(len(thub.clients)), fmt.Sprintf("client map must have len %d", len(testClients)))
}

测试在后台线程中运行的函数的结果

1 个答案: