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