github.com/jackc/pgx/v5@v5.5.5/pgxpool/tx_test.go (about)

     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  }