2
0

Add Hijack from v5

This commit is contained in:
Jack Christensen
2022-04-14 11:50:12 -05:00
committed by Jack Christensen
parent dc0ad04ff5
commit 37c3f157bc
2 changed files with 42 additions and 0 deletions
+26
View File
@@ -115,6 +115,32 @@ func TestPoolAcquireAndConnRelease(t *testing.T) {
c.Release()
}
func TestPoolAcquireAndConnHijack(t *testing.T) {
t.Parallel()
ctx := context.Background()
pool, err := pgxpool.Connect(ctx, os.Getenv("PGX_TEST_DATABASE"))
require.NoError(t, err)
defer pool.Close()
c, err := pool.Acquire(ctx)
require.NoError(t, err)
connsBeforeHijack := pool.Stat().TotalConns()
conn := c.Hijack()
defer conn.Close(ctx)
connsAfterHijack := pool.Stat().TotalConns()
require.Equal(t, connsBeforeHijack-1, connsAfterHijack)
var n int32
err = conn.QueryRow(ctx, `select 1`).Scan(&n)
require.NoError(t, err)
require.Equal(t, int32(1), n)
}
func TestPoolAcquireFunc(t *testing.T) {
t.Parallel()