forked from mirror/redis
36 lines
524 B
Go
36 lines
524 B
Go
|
package pool_test
|
||
|
|
|
||
|
|
import (
|
||
|
|
"net"
|
||
|
|
"sync"
|
||
|
|
"testing"
|
||
|
|
|
||
|
|
. "github.com/onsi/ginkgo"
|
||
|
|
. "github.com/onsi/gomega"
|
||
|
|
)
|
||
|
|
|
||
|
|
func TestGinkgoSuite(t *testing.T) {
|
||
|
|
RegisterFailHandler(Fail)
|
||
|
|
RunSpecs(t, "pool")
|
||
|
|
}
|
||
|
|
|
||
|
|
func perform(n int, cbs ...func(int)) {
|
||
|
|
var wg sync.WaitGroup
|
||
|
|
for _, cb := range cbs {
|
||
|
|
for i := 0; i < n; i++ {
|
||
|
|
wg.Add(1)
|
||
|
|
go func(cb func(int), i int) {
|
||
|
|
defer GinkgoRecover()
|
||
|
|
defer wg.Done()
|
||
|
|
|
||
|
|
cb(i)
|
||
|
|
}(cb, i)
|
||
|
|
}
|
||
|
|
}
|
||
|
|
wg.Wait()
|
||
|
|
}
|
||
|
|
|
||
|
|
func dummyDialer() (net.Conn, error) {
|
||
|
|
return &net.TCPConn{}, nil
|
||
|
|
}
|