testing

integration/go/go_pgx/load_balancer_test.go

@@ -231,10 +231,27 @@ outer:
 }
 
 func prewarm(t *testing.T, pool *pgxpool.Pool) {
+	ctx := context.Background()
 	for range 25 {
-		for _, q := range []string{"BEGIN", "SELECT 1", "COMMIT", "SELECT 1"} {
-			_, err := pool.Exec(context.Background(), q)
-			assert.NoError(t, err)
-		}
+		// pin to one connection via a real Tx
+		tx, err := pool.Begin(ctx)
+		assert.NoError(t, err)
+		_, err = tx.Exec(ctx, "SELECT 1")
+		assert.NoError(t, err)
+		err = tx.Commit(ctx)
+		assert.NoError(t, err)
+
+		// simple query to warm another round
+		_, err = pool.Exec(ctx, "SELECT 1")
+		assert.NoError(t, err)
 	}
 }
+
+// func prewarm(t *testing.T, pool *pgxpool.Pool) {
+// 	for range 25 {
+// 		for _, q := range []string{"BEGIN", "SELECT 1", "COMMIT", "SELECT 1"} {
+// 			_, err := pool.Exec(context.Background(), q)
+// 			assert.NoError(t, err)
+// 		}
+// 	}
+// }