大家好,今天本人给大家带来文章《在进行并行表测试用例时如何推迟测试服务器关闭?》,文中内容主要涉及到,如果你对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()
...