mirror of
https://github.com/jackc/pgx.git
synced 2025-05-01 21:19:54 +00:00
c3e41872a8
This is in preparation for a Begin / Tx interface that will similate nested transactions with savepoints. In addition, this passes the TxOptions struct by value and thereby removes an allocation.
81 lines
1.7 KiB
Go
81 lines
1.7 KiB
Go
package pgxpool_test
|
|
|
|
import (
|
|
"context"
|
|
"os"
|
|
"testing"
|
|
|
|
"github.com/jackc/pgx/v4/pgxpool"
|
|
"github.com/stretchr/testify/require"
|
|
)
|
|
|
|
func TestTxExec(t *testing.T) {
|
|
t.Parallel()
|
|
|
|
pool, err := pgxpool.Connect(context.Background(), os.Getenv("PGX_TEST_DATABASE"))
|
|
require.NoError(t, err)
|
|
defer pool.Close()
|
|
|
|
tx, err := pool.Begin(context.Background())
|
|
require.NoError(t, err)
|
|
defer tx.Rollback(context.Background())
|
|
|
|
testExec(t, tx)
|
|
}
|
|
|
|
func TestTxQuery(t *testing.T) {
|
|
t.Parallel()
|
|
|
|
pool, err := pgxpool.Connect(context.Background(), os.Getenv("PGX_TEST_DATABASE"))
|
|
require.NoError(t, err)
|
|
defer pool.Close()
|
|
|
|
tx, err := pool.Begin(context.Background())
|
|
require.NoError(t, err)
|
|
defer tx.Rollback(context.Background())
|
|
|
|
testQuery(t, tx)
|
|
}
|
|
|
|
func TestTxQueryRow(t *testing.T) {
|
|
t.Parallel()
|
|
|
|
pool, err := pgxpool.Connect(context.Background(), os.Getenv("PGX_TEST_DATABASE"))
|
|
require.NoError(t, err)
|
|
defer pool.Close()
|
|
|
|
tx, err := pool.Begin(context.Background())
|
|
require.NoError(t, err)
|
|
defer tx.Rollback(context.Background())
|
|
|
|
testQueryRow(t, tx)
|
|
}
|
|
|
|
func TestTxSendBatch(t *testing.T) {
|
|
t.Parallel()
|
|
|
|
pool, err := pgxpool.Connect(context.Background(), os.Getenv("PGX_TEST_DATABASE"))
|
|
require.NoError(t, err)
|
|
defer pool.Close()
|
|
|
|
tx, err := pool.Begin(context.Background())
|
|
require.NoError(t, err)
|
|
defer tx.Rollback(context.Background())
|
|
|
|
testSendBatch(t, tx)
|
|
}
|
|
|
|
func TestTxCopyFrom(t *testing.T) {
|
|
t.Parallel()
|
|
|
|
pool, err := pgxpool.Connect(context.Background(), os.Getenv("PGX_TEST_DATABASE"))
|
|
require.NoError(t, err)
|
|
defer pool.Close()
|
|
|
|
tx, err := pool.Begin(context.Background())
|
|
require.NoError(t, err)
|
|
defer tx.Rollback(context.Background())
|
|
|
|
testCopyFrom(t, tx)
|
|
}
|