gopkg/pgx
2
0
Fork 0
Code Activity

Add context timeouts for more pgxpool tests

Browse Source
This commit is contained in:
Jack Christensen
2023-05-29 11:15:40 -05:00
parent a1a97a7ca8
commit 5f6636d028
4 changed files with 85 additions and 53 deletions
Download Patch File Download Diff File Expand all files Collapse all files
pgxpool/pool_test.go
+6 -6
View File
@@ -630,7 +630,7 @@ func TestPoolExec(t *testing.T) {
require.NoError(t, err)
defer pool.Close()
testExec(t, pool)
testExec(t, ctx, pool)
}
func TestPoolQuery(t *testing.T) {
@@ -644,7 +644,7 @@ func TestPoolQuery(t *testing.T) {
defer pool.Close()
// Test common usage
testQuery(t, pool)
testQuery(t, ctx, pool)
waitForReleaseToComplete()
// Test expected pool behavior
@@ -675,7 +675,7 @@ func TestPoolQueryRow(t *testing.T) {
require.NoError(t, err)
defer pool.Close()
testQueryRow(t, pool)
testQueryRow(t, ctx, pool)
waitForReleaseToComplete()
stats := pool.Stat()
@@ -708,7 +708,7 @@ func TestPoolSendBatch(t *testing.T) {
require.NoError(t, err)
defer pool.Close()
testSendBatch(t, pool)
testSendBatch(t, ctx, pool)
waitForReleaseToComplete()
stats := pool.Stat()
@@ -883,8 +883,8 @@ func TestConnPoolQueryConcurrentLoad(t *testing.T) {
for i := 0; i < n; i++ {
go func() {
defer func() { done <- true }()
testQuery(t, pool)
testQueryRow(t, pool)
testQuery(t, ctx, pool)
testQueryRow(t, ctx, pool)
}()
}