...

Source file src/github.com/jackc/pgx/v4/pgxpool/conn_test.go

Documentation: github.com/jackc/pgx/v4/pgxpool

     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 TestConnExec(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  	c, err := pool.Acquire(context.Background())
    20  	require.NoError(t, err)
    21  	defer c.Release()
    22  
    23  	testExec(t, c)
    24  }
    25  
    26  func TestConnQuery(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  	c, err := pool.Acquire(context.Background())
    34  	require.NoError(t, err)
    35  	defer c.Release()
    36  
    37  	testQuery(t, c)
    38  }
    39  
    40  func TestConnQueryRow(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  	c, err := pool.Acquire(context.Background())
    48  	require.NoError(t, err)
    49  	defer c.Release()
    50  
    51  	testQueryRow(t, c)
    52  }
    53  
    54  func TestConnSendBatch(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  	c, err := pool.Acquire(context.Background())
    62  	require.NoError(t, err)
    63  	defer c.Release()
    64  
    65  	testSendBatch(t, c)
    66  }
    67  
    68  func TestConnCopyFrom(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  	c, err := pool.Acquire(context.Background())
    76  	require.NoError(t, err)
    77  	defer c.Release()
    78  
    79  	testCopyFrom(t, c)
    80  }
    81  

View as plain text