main_test.go 550 B

123456789101112131415161718192021222324252627282930313233343536
  1. package pool_test
  2. import (
  3. "context"
  4. "net"
  5. "sync"
  6. "testing"
  7. . "github.com/onsi/ginkgo"
  8. . "github.com/onsi/gomega"
  9. )
  10. func TestGinkgoSuite(t *testing.T) {
  11. RegisterFailHandler(Fail)
  12. RunSpecs(t, "pool")
  13. }
  14. func perform(n int, cbs ...func(int)) {
  15. var wg sync.WaitGroup
  16. for _, cb := range cbs {
  17. for i := 0; i < n; i++ {
  18. wg.Add(1)
  19. go func(cb func(int), i int) {
  20. defer GinkgoRecover()
  21. defer wg.Done()
  22. cb(i)
  23. }(cb, i)
  24. }
  25. }
  26. wg.Wait()
  27. }
  28. func dummyDialer(context.Context) (net.Conn, error) {
  29. return &net.TCPConn{}, nil
  30. }