2
0

Add Pool.Reset()

This commit is contained in:
Jack Christensen
2022-07-30 12:22:29 -05:00
parent 83670d675d
commit 0eda0109ca
2 changed files with 34 additions and 0 deletions
+25
View File
@@ -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()