在进行并行表测试用例时如何推迟测试服务器关闭？

大家好，今天本人给大家带来文章《在进行并行表测试用例时如何推迟测试服务器关闭？》，文中内容主要涉及到，如果你对golang方面的知识点感兴趣，那就请各位朋友继续看下去吧~希望能真正帮到你们，谢谢！

在执行并行表测试函数时，我无法推迟测试 Http 服务器的关闭。我正在尝试使用 sync.waitgroup 进行等待，但 wg.done() 要么关闭得太早，要么从未发生。

ts := httptest.NewServer(http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
    w.WriteHeader(http.StatusOK)
    _, _ = w.Write([]byte(`hello`))
}))
defer ts.Close()

var wg sync.WaitGroup
for _, tc := range testCases {
    wg.Add(1)
    func(tc testCase) {
        // `Done` here doesn't wait for the test to run, so closes the test server early
        // defer wg.Done()
        t.Run(tc.name, func(t *testing.T) {
            // `Done` here is never called, so causes a timeout
            // defer wg.Done()
            t.Parallel()
            _, err := http.Get(ts.URL)
            if err != nil {
                t.Fatal(err.Error())
            }
        })
    }(tc)
}
wg.Wait()

有关可运行的示例，请参阅 https://play.Golang.org/p/1gvwrwlrsja

解决方案

这很可能是因为如果 t.parallel() 在其体内使用，t.run() 不会阻塞。将 t.parallel() 测试用例包装到非并行组中应该可以解决该问题：

...
var wg sync.WaitGroup
for _, tc := range testCases {                                                                                                                                   
    wg.Add(1)                                                                                                                                                                                                         
    func(tc testCase) {                                                                                                                                                                                                   
        defer wg.Done()                                                                                                                                                                                                   
        t.Run("mygroup", func(t *testing.T) {                                                                                                                                                                                   
            t.Run(tc.name, func(t *testing.T) {                                                                                                                                                                                    
                t.Parallel()                                                                                                                                                                                                      
                _, err := http.Get(ts.URL)
                if err != nil {                                                                                                                                                                                                       
                    t.Fatal(err.Error())                                                                                                                                                                                          
                }                                                                                                                                                                                                             
            })                                                                                                                                                                                                            
         })                                                                                                                                                                                                            
    }(tc)
}
wg.Wait()
...

编辑：我在 golang 存储库问题中找到了这个参考，它表明相同：https://GitHub.com/golang/go/issues/17791

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