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 TestConnExec(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 c, err := pool.Acquire(ctx)
24 require.NoError(t, err)
25 defer c.Release()
26
27 testExec(t, ctx, c)
28 }
29
30 func TestConnQuery(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 c, err := pool.Acquire(ctx)
41 require.NoError(t, err)
42 defer c.Release()
43
44 testQuery(t, ctx, c)
45 }
46
47 func TestConnQueryRow(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 c, err := pool.Acquire(ctx)
58 require.NoError(t, err)
59 defer c.Release()
60
61 testQueryRow(t, ctx, c)
62 }
63
64 func TestConnSendBatch(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 c, err := pool.Acquire(ctx)
75 require.NoError(t, err)
76 defer c.Release()
77
78 testSendBatch(t, ctx, c)
79 }
80
81 func TestConnCopyFrom(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 c, err := pool.Acquire(ctx)
92 require.NoError(t, err)
93 defer c.Release()
94
95 testCopyFrom(t, ctx, c)
96 }
97
View as plain text