GOMODULES
/
pgx
mirror of https://github.com/jackc/pgx.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

Use testing.TB interface for shared test/benchmark helpers

pgx-vs-pq
Jack Christensen 2013-12-10 08:25:30 -06:00
parent f76489f7d7
commit f0a3d22571
1 changed files with 7 additions and 10 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

17
helper_test.go
View File

@ -2,15 +2,12 @@ package pgx_test
import ( import (
"github.com/JackC/pgx" "github.com/JackC/pgx"
"testing"
) )
type test interface {
Fatalf(format string, args ...interface{})
}
var sharedConnection *pgx.Connection var sharedConnection *pgx.Connection
func getSharedConnection(t test) (c *pgx.Connection) { func getSharedConnection(t testing.TB) (c *pgx.Connection) {
if sharedConnection == nil || !sharedConnection.IsAlive() { if sharedConnection == nil || !sharedConnection.IsAlive() {
var err error var err error
sharedConnection, err = pgx.Connect(*defaultConnectionParameters) sharedConnection, err = pgx.Connect(*defaultConnectionParameters)
@ -22,13 +19,13 @@ func getSharedConnection(t test) (c *pgx.Connection) {
return sharedConnection return sharedConnection
} }
func mustPrepare(t test, conn *pgx.Connection, name, sql string) { func mustPrepare(t testing.TB, conn *pgx.Connection, name, sql string) {
if err := conn.Prepare(name, sql); err != nil { if err := conn.Prepare(name, sql); err != nil {
t.Fatalf("Could not prepare %v: %v", name, err) t.Fatalf("Could not prepare %v: %v", name, err)
} }
} }
func mustExecute(t test, conn *pgx.Connection, sql string, arguments ...interface{}) (commandTag string) { func mustExecute(t testing.TB, conn *pgx.Connection, sql string, arguments ...interface{}) (commandTag string) {
var err error var err error
if commandTag, err = conn.Execute(sql, arguments...); err != nil { if commandTag, err = conn.Execute(sql, arguments...); err != nil {
t.Fatalf("Execute unexpectedly failed with %v: %v", sql, err) t.Fatalf("Execute unexpectedly failed with %v: %v", sql, err)
@ -36,7 +33,7 @@ func mustExecute(t test, conn *pgx.Connection, sql string, arguments ...interfac
return return
} }
func mustSelectRow(t test, conn *pgx.Connection, sql string, arguments ...interface{}) (row map[string]interface{}) { func mustSelectRow(t testing.TB, conn *pgx.Connection, sql string, arguments ...interface{}) (row map[string]interface{}) {
var err error var err error
if row, err = conn.SelectRow(sql, arguments...); err != nil { if row, err = conn.SelectRow(sql, arguments...); err != nil {
t.Fatalf("SelectRow unexpectedly failed with %v: %v", sql, err) t.Fatalf("SelectRow unexpectedly failed with %v: %v", sql, err)
@ -44,7 +41,7 @@ func mustSelectRow(t test, conn *pgx.Connection, sql string, arguments ...interf
return return
} }
func mustSelectRows(t test, conn *pgx.Connection, sql string, arguments ...interface{}) (rows []map[string]interface{}) { func mustSelectRows(t testing.TB, conn *pgx.Connection, sql string, arguments ...interface{}) (rows []map[string]interface{}) {
var err error var err error
if rows, err = conn.SelectRows(sql, arguments...); err != nil { if rows, err = conn.SelectRows(sql, arguments...); err != nil {
t.Fatalf("SelectRows unexpected failed with %v: %v", sql, err) t.Fatalf("SelectRows unexpected failed with %v: %v", sql, err)
@ -52,7 +49,7 @@ func mustSelectRows(t test, conn *pgx.Connection, sql string, arguments ...inter
return return
} }
func mustSelectValue(t test, conn *pgx.Connection, sql string, arguments ...interface{}) (value interface{}) { func mustSelectValue(t testing.TB, conn *pgx.Connection, sql string, arguments ...interface{}) (value interface{}) {
var err error var err error
if value, err = conn.SelectValue(sql, arguments...); err != nil { if value, err = conn.SelectValue(sql, arguments...); err != nil {
t.Fatalf("SelectValue unexpectedly failed with %v: %v", sql, err) t.Fatalf("SelectValue unexpectedly failed with %v: %v", sql, err)