7718ee6207
Always require context and prepend options to arguments if necessary.
47 lines
950 B
Go
47 lines
950 B
Go
package pool_test
|
|
|
|
import (
|
|
"context"
|
|
"os"
|
|
"testing"
|
|
|
|
"github.com/jackc/pgx/pool"
|
|
"github.com/stretchr/testify/require"
|
|
)
|
|
|
|
func TestConnExec(t *testing.T) {
|
|
pool, err := pool.Connect(context.Background(), os.Getenv("PGX_TEST_DATABASE"))
|
|
require.NoError(t, err)
|
|
defer pool.Close()
|
|
|
|
c, err := pool.Acquire(context.Background())
|
|
require.NoError(t, err)
|
|
defer c.Release()
|
|
|
|
testExec(t, c)
|
|
}
|
|
|
|
func TestConnQuery(t *testing.T) {
|
|
pool, err := pool.Connect(context.Background(), os.Getenv("PGX_TEST_DATABASE"))
|
|
require.NoError(t, err)
|
|
defer pool.Close()
|
|
|
|
c, err := pool.Acquire(context.Background())
|
|
require.NoError(t, err)
|
|
defer c.Release()
|
|
|
|
testQuery(t, c)
|
|
}
|
|
|
|
func TestConnQueryRow(t *testing.T) {
|
|
pool, err := pool.Connect(context.Background(), os.Getenv("PGX_TEST_DATABASE"))
|
|
require.NoError(t, err)
|
|
defer pool.Close()
|
|
|
|
c, err := pool.Acquire(context.Background())
|
|
require.NoError(t, err)
|
|
defer c.Release()
|
|
|
|
testQueryRow(t, c)
|
|
}
|