1 package pgxpool_test
2
3 import (
4 "context"
5 "os"
6 "testing"
7 "time"
8
9 "github.com/jackc/pgx/v5/pgxpool"
10 "github.com/stretchr/testify/require"
11 )
12
13 func TestTxExec(t *testing.T) {
14 t.Parallel()
15
16 ctx, cancel := context.WithTimeout(context.Background(), 120*time.Second)
17 defer cancel()
18
19 pool, err := pgxpool.New(ctx, os.Getenv("PGX_TEST_DATABASE"))
20 require.NoError(t, err)
21 defer pool.Close()
22
23 tx, err := pool.Begin(ctx)
24 require.NoError(t, err)
25 defer tx.Rollback(ctx)
26
27 testExec(t, ctx, tx)
28 }
29
30 func TestTxQuery(t *testing.T) {
31 t.Parallel()
32
33 ctx, cancel := context.WithTimeout(context.Background(), 120*time.Second)
34 defer cancel()
35
36 pool, err := pgxpool.New(ctx, os.Getenv("PGX_TEST_DATABASE"))
37 require.NoError(t, err)
38 defer pool.Close()
39
40 tx, err := pool.Begin(ctx)
41 require.NoError(t, err)
42 defer tx.Rollback(ctx)
43
44 testQuery(t, ctx, tx)
45 }
46
47 func TestTxQueryRow(t *testing.T) {
48 t.Parallel()
49
50 ctx, cancel := context.WithTimeout(context.Background(), 120*time.Second)
51 defer cancel()
52
53 pool, err := pgxpool.New(ctx, os.Getenv("PGX_TEST_DATABASE"))
54 require.NoError(t, err)
55 defer pool.Close()
56
57 tx, err := pool.Begin(ctx)
58 require.NoError(t, err)
59 defer tx.Rollback(ctx)
60
61 testQueryRow(t, ctx, tx)
62 }
63
64 func TestTxSendBatch(t *testing.T) {
65 t.Parallel()
66
67 ctx, cancel := context.WithTimeout(context.Background(), 120*time.Second)
68 defer cancel()
69
70 pool, err := pgxpool.New(ctx, os.Getenv("PGX_TEST_DATABASE"))
71 require.NoError(t, err)
72 defer pool.Close()
73
74 tx, err := pool.Begin(ctx)
75 require.NoError(t, err)
76 defer tx.Rollback(ctx)
77
78 testSendBatch(t, ctx, tx)
79 }
80
81 func TestTxCopyFrom(t *testing.T) {
82 t.Parallel()
83
84 ctx, cancel := context.WithTimeout(context.Background(), 120*time.Second)
85 defer cancel()
86
87 pool, err := pgxpool.New(ctx, os.Getenv("PGX_TEST_DATABASE"))
88 require.NoError(t, err)
89 defer pool.Close()
90
91 tx, err := pool.Begin(ctx)
92 require.NoError(t, err)
93 defer tx.Rollback(ctx)
94
95 testCopyFrom(t, ctx, tx)
96 }
97
View as plain text