mirror of https://github.com/go-redis/redis.git
66 lines
1.7 KiB
Go
66 lines
1.7 KiB
Go
//go:build linux || darwin || dragonfly || freebsd || netbsd || openbsd || solaris || illumos
|
|
|
|
package pool
|
|
|
|
import (
|
|
"crypto/tls"
|
|
"net"
|
|
"net/http/httptest"
|
|
"time"
|
|
|
|
. "github.com/bsm/ginkgo/v2"
|
|
. "github.com/bsm/gomega"
|
|
)
|
|
|
|
var _ = Describe("tests conn_check with real conns", func() {
|
|
var ts *httptest.Server
|
|
var conn net.Conn
|
|
var tlsConn *tls.Conn
|
|
var err error
|
|
|
|
BeforeEach(func() {
|
|
ts = httptest.NewServer(nil)
|
|
conn, err = net.DialTimeout(ts.Listener.Addr().Network(), ts.Listener.Addr().String(), time.Second)
|
|
Expect(err).NotTo(HaveOccurred())
|
|
tlsTestServer := httptest.NewUnstartedServer(nil)
|
|
tlsTestServer.StartTLS()
|
|
tlsConn, err = tls.DialWithDialer(&net.Dialer{Timeout: time.Second}, tlsTestServer.Listener.Addr().Network(), tlsTestServer.Listener.Addr().String(), &tls.Config{InsecureSkipVerify: true})
|
|
Expect(err).NotTo(HaveOccurred())
|
|
})
|
|
|
|
AfterEach(func() {
|
|
ts.Close()
|
|
})
|
|
|
|
It("good conn check", func() {
|
|
Expect(connCheck(conn)).NotTo(HaveOccurred())
|
|
|
|
Expect(conn.Close()).NotTo(HaveOccurred())
|
|
Expect(connCheck(conn)).To(HaveOccurred())
|
|
})
|
|
|
|
It("good tls conn check", func() {
|
|
Expect(connCheck(tlsConn)).NotTo(HaveOccurred())
|
|
|
|
Expect(tlsConn.Close()).NotTo(HaveOccurred())
|
|
Expect(connCheck(tlsConn)).To(HaveOccurred())
|
|
})
|
|
|
|
It("bad conn check", func() {
|
|
Expect(conn.Close()).NotTo(HaveOccurred())
|
|
Expect(connCheck(conn)).To(HaveOccurred())
|
|
})
|
|
|
|
It("bad tls conn check", func() {
|
|
Expect(tlsConn.Close()).NotTo(HaveOccurred())
|
|
Expect(connCheck(tlsConn)).To(HaveOccurred())
|
|
})
|
|
|
|
It("check conn deadline", func() {
|
|
Expect(conn.SetDeadline(time.Now())).NotTo(HaveOccurred())
|
|
time.Sleep(time.Millisecond * 10)
|
|
Expect(connCheck(conn)).NotTo(HaveOccurred())
|
|
Expect(conn.Close()).NotTo(HaveOccurred())
|
|
})
|
|
})
|