我一直在寻找,但到目前为止,只有Ariejan de Vroom撰写的类似文章here。
我想知道我是否可以将goroutine引入单元测试,以便它可以精确计算正在运行的goroutines的并发数量,并且可以告诉我它们是否正确地在我所说的数字中生成goroutine。
我有以下代码,例如..
import (
"testing"
"github.com/stretchr/testify/assert"
)
func createList(job int, done chan bool) {
time.Sleep(500)
// do something
time.Sleep(500)
done <- true
return
}
func TestNewList(t *testing.T) {
list := NewList()
if assert.NotNil(t, list) {
const numGoRoutines = 16
jobs := make(chan int, numGoRoutines)
done := make(chan bool, 1)
for j := 1; j <= numGoRoutines; j++ {
jobs <- j
go createList(j, done)
fmt.Println("sent job", j)
}
close(jobs)
fmt.Println("sent all jobs")
<-done
}
答案 0 :(得分:0)
一种可能的方法是使用runtime.Stack()
或分析runtime.debug.PrintStack()
的输出,以便在给定时间查看所有goroutine。
这些选项详见“How to dump goroutine stacktraces?”。
答案 1 :(得分:0)
据我了解,您愿意限制同时运行的例程数量并验证它是否正常运行。我建议写一个函数,它将采用例程和参数,并使用模拟例程来测试它
在以下示例中,spawn
函数同时运行fn
例程count
次但不超过limit
例程。我将它包装到main函数中以在操场上运行它,但是你可以使用相同的方法来测试方法。
package main
import (
"fmt"
"sync"
"time"
)
func spawn(fn func(), count int, limit int) {
limiter := make(chan bool, limit)
spawned := func() {
defer func() { <-limiter }()
fn()
}
for i := 0; i < count; i++ {
limiter <- true
go spawned()
}
}
func main() {
count := 10
limit := 3
var wg sync.WaitGroup
wg.Add(count)
concurrentCount := 0
failed := false
var mock = func() {
defer func() {
wg.Done()
concurrentCount--
}()
concurrentCount++
if concurrentCount > limit {
failed = true // test could be failed here without waiting all routines finish
}
time.Sleep(100)
}
spawn(mock, count, limit)
wg.Wait()
if failed {
fmt.Println("Test failed")
} else {
fmt.Println("Test passed")
}
}