fix ut: pool with badconn

This commit is contained in:
hidu 2021-07-16 17:02:39 +08:00
parent 78875ecee7
commit 6e0ed33250
1 changed files with 21 additions and 7 deletions

View File

@ -12,10 +12,10 @@ import (
"testing"
"time"
"github.com/go-redis/redis/v8"
. "github.com/onsi/ginkgo"
. "github.com/onsi/gomega"
"github.com/go-redis/redis/v8"
)
const (
@ -373,14 +373,28 @@ func (e badConnError) Timeout() bool { return true }
func (e badConnError) Temporary() bool { return false }
type badConn struct {
net.TCPConn
readDelay, writeDelay time.Duration
readErr, writeErr error
}
var _ net.Conn = &badConn{}
func (cn *badConn) Close() error {
return nil
}
func (cn *badConn) LocalAddr() net.Addr {
return &net.TCPAddr{}
}
func (cn *badConn) RemoteAddr() net.Addr {
return &net.TCPAddr{}
}
func (cn *badConn) SetDeadline(t time.Time) error {
return nil
}
func (cn *badConn) SetReadDeadline(t time.Time) error {
return nil
}