Add Pool.Reset()
This commit is contained in:
@@ -349,6 +349,31 @@ func TestPoolAcquireAllIdle(t *testing.T) {
|
||||
}
|
||||
}
|
||||
|
||||
func TestPoolReset(t *testing.T) {
|
||||
t.Parallel()
|
||||
|
||||
db, err := pgxpool.New(context.Background(), os.Getenv("PGX_TEST_DATABASE"))
|
||||
require.NoError(t, err)
|
||||
defer db.Close()
|
||||
|
||||
conns := make([]*pgxpool.Conn, 3)
|
||||
for i := range conns {
|
||||
conns[i], err = db.Acquire(context.Background())
|
||||
assert.NoError(t, err)
|
||||
}
|
||||
|
||||
db.Reset()
|
||||
|
||||
for _, c := range conns {
|
||||
if c != nil {
|
||||
c.Release()
|
||||
}
|
||||
}
|
||||
waitForReleaseToComplete()
|
||||
|
||||
require.EqualValues(t, 0, db.Stat().TotalConns())
|
||||
}
|
||||
|
||||
func TestConnReleaseChecksMaxConnLifetime(t *testing.T) {
|
||||
t.Parallel()
|
||||
|
||||
|
||||
Reference in New Issue
Block a user