bakape
54 lines
1.3 KiB
Go
54 lines
1.3 KiB
Go
package pgconn_test
|
|
|
|
import (
|
|
"context"
|
|
"testing"
|
|
"time"
|
|
|
|
"github.com/jackc/pgconn"
|
|
|
|
"github.com/stretchr/testify/assert"
|
|
"github.com/stretchr/testify/require"
|
|
)
|
|
|
|
func closeConn(t testing.TB, conn *pgconn.PgConn) {
|
|
ctx, cancel := context.WithTimeout(context.Background(), 5*time.Second)
|
|
defer cancel()
|
|
require.NoError(t, conn.Close(ctx))
|
|
}
|
|
|
|
// Do a simple query to ensure the connection is still usable
|
|
func ensureConnValid(t *testing.T, pgConn *pgconn.PgConn) {
|
|
ctx, cancel := context.WithTimeout(context.Background(), time.Second)
|
|
result := pgConn.ExecParams(ctx, "select generate_series(1,$1)", [][]byte{[]byte("3")}, nil, nil, nil).Read()
|
|
cancel()
|
|
|
|
require.Nil(t, result.Err)
|
|
assert.Equal(t, 3, len(result.Rows))
|
|
assert.Equal(t, "1", string(result.Rows[0][0]))
|
|
assert.Equal(t, "2", string(result.Rows[1][0]))
|
|
assert.Equal(t, "3", string(result.Rows[2][0]))
|
|
}
|
|
|
|
// Run subtest both with a context.Background() and nil context
|
|
func splitOnContext(t *testing.T, test func(t *testing.T, ctx context.Context)) {
|
|
t.Helper()
|
|
|
|
cases := [...]struct {
|
|
name string
|
|
ctx context.Context
|
|
}{
|
|
{"background context", context.Background()},
|
|
{"nil context", nil},
|
|
}
|
|
|
|
for i := range cases {
|
|
c := cases[i]
|
|
t.Run(c.name, func(t *testing.T) {
|
|
t.Helper()
|
|
t.Parallel()
|
|
test(t, c.ctx)
|
|
})
|
|
}
|
|
}
|