Remove mustPrepare from tests

Now that Query/QueryRow always prepares statements when given SQL
text there is no need to test raw SQL and prepared statements of
the same query.
pull/78/merge
Jack Christensen 2015-09-03 13:39:28 -05:00
parent e69d2bba26
commit f672dd2732
3 changed files with 20 additions and 71 deletions

View File

@ -20,15 +20,6 @@ func closeConn(t testing.TB, conn *pgx.Conn) {
} }
} }
func mustPrepare(t testing.TB, conn *pgx.Conn, name, sql string) *pgx.PreparedStatement {
ps, err := conn.Prepare(name, sql)
if err != nil {
t.Fatalf("Could not prepare %v: %v", name, err)
}
return ps
}
func mustExec(t testing.TB, conn *pgx.Conn, sql string, arguments ...interface{}) (commandTag pgx.CommandTag) { func mustExec(t testing.TB, conn *pgx.Conn, sql string, arguments ...interface{}) (commandTag pgx.CommandTag) {
var err error var err error
if commandTag, err = conn.Exec(sql, arguments...); err != nil { if commandTag, err = conn.Exec(sql, arguments...); err != nil {

View File

@ -488,19 +488,13 @@ func TestQueryRowNoResults(t *testing.T) {
conn := mustConnect(t, *defaultConnConfig) conn := mustConnect(t, *defaultConnConfig)
defer closeConn(t, conn) defer closeConn(t, conn)
sql := "select 1 where 1=0"
psName := "selectNothing"
mustPrepare(t, conn, psName, sql)
for _, sql := range []string{sql, psName} {
var n int32 var n int32
err := conn.QueryRow(sql).Scan(&n) err := conn.QueryRow("select 1 where 1=0").Scan(&n)
if err != pgx.ErrNoRows { if err != pgx.ErrNoRows {
t.Errorf("Expected pgx.ErrNoRows, got %v", err) t.Errorf("Expected pgx.ErrNoRows, got %v", err)
} }
ensureConnValid(t, conn) ensureConnValid(t, conn)
}
} }
func TestQueryRowCoreInt16Slice(t *testing.T) { func TestQueryRowCoreInt16Slice(t *testing.T) {

View File

@ -1,7 +1,6 @@
package pgx_test package pgx_test
import ( import (
"fmt"
"github.com/jackc/pgx" "github.com/jackc/pgx"
"net" "net"
"reflect" "reflect"
@ -16,13 +15,6 @@ func TestDateTranscode(t *testing.T) {
conn := mustConnect(t, *defaultConnConfig) conn := mustConnect(t, *defaultConnConfig)
defer closeConn(t, conn) defer closeConn(t, conn)
mustPrepare(t, conn, "testTranscode", "select $1::date")
defer func() {
if err := conn.Deallocate("testTranscode"); err != nil {
t.Fatalf("Unable to deallocate prepared statement: %v", err)
}
}()
dates := []time.Time{ dates := []time.Time{
time.Date(1990, 1, 1, 0, 0, 0, 0, time.Local), time.Date(1990, 1, 1, 0, 0, 0, 0, time.Local),
time.Date(1999, 12, 31, 0, 0, 0, 0, time.Local), time.Date(1999, 12, 31, 0, 0, 0, 0, time.Local),
@ -36,7 +28,6 @@ func TestDateTranscode(t *testing.T) {
for _, actualDate := range dates { for _, actualDate := range dates {
var d time.Time var d time.Time
// Test text format
err := conn.QueryRow("select $1::date", actualDate).Scan(&d) err := conn.QueryRow("select $1::date", actualDate).Scan(&d)
if err != nil { if err != nil {
t.Fatalf("Unexpected failure on QueryRow Scan: %v", err) t.Fatalf("Unexpected failure on QueryRow Scan: %v", err)
@ -44,15 +35,6 @@ func TestDateTranscode(t *testing.T) {
if !actualDate.Equal(d) { if !actualDate.Equal(d) {
t.Errorf("Did not transcode date successfully: %v is not %v", d, actualDate) t.Errorf("Did not transcode date successfully: %v is not %v", d, actualDate)
} }
// Test binary format
err = conn.QueryRow("testTranscode", actualDate).Scan(&d)
if err != nil {
t.Fatalf("Unexpected failure on QueryRow Scan: %v", err)
}
if !actualDate.Equal(d) {
t.Errorf("Did not transcode date successfully: %v is not %v", d, actualDate)
}
} }
} }
@ -74,14 +56,7 @@ func TestTimestampTzTranscode(t *testing.T) {
t.Errorf("Did not transcode time successfully: %v is not %v", outputTime, inputTime) t.Errorf("Did not transcode time successfully: %v is not %v", outputTime, inputTime)
} }
mustPrepare(t, conn, "testTranscode", "select $1::timestamptz") err = conn.QueryRow("select $1::timestamptz", inputTime).Scan(&outputTime)
defer func() {
if err := conn.Deallocate("testTranscode"); err != nil {
t.Fatalf("Unable to deallocate prepared statement: %v", err)
}
}()
err = conn.QueryRow("testTranscode", inputTime).Scan(&outputTime)
if err != nil { if err != nil {
t.Fatalf("QueryRow Scan failed: %v", err) t.Fatalf("QueryRow Scan failed: %v", err)
} }
@ -195,24 +170,19 @@ func TestNullX(t *testing.T) {
} }
for i, tt := range tests { for i, tt := range tests {
psName := fmt.Sprintf("success%d", i)
mustPrepare(t, conn, psName, tt.sql)
for _, sql := range []string{tt.sql, psName} {
actual = zero actual = zero
err := conn.QueryRow(sql, tt.queryArgs...).Scan(tt.scanArgs...) err := conn.QueryRow(tt.sql, tt.queryArgs...).Scan(tt.scanArgs...)
if err != nil { if err != nil {
t.Errorf("%d. Unexpected failure: %v (sql -> %v, queryArgs -> %v)", i, err, sql, tt.queryArgs) t.Errorf("%d. Unexpected failure: %v (sql -> %v, queryArgs -> %v)", i, err, tt.sql, tt.queryArgs)
} }
if actual != tt.expected { if actual != tt.expected {
t.Errorf("%d. Expected %v, got %v (sql -> %v, queryArgs -> %v)", i, tt.expected, actual, sql, tt.queryArgs) t.Errorf("%d. Expected %v, got %v (sql -> %v, queryArgs -> %v)", i, tt.expected, actual, tt.sql, tt.queryArgs)
} }
ensureConnValid(t, conn) ensureConnValid(t, conn)
} }
}
} }
func TestArrayDecoding(t *testing.T) { func TestArrayDecoding(t *testing.T) {
@ -270,12 +240,9 @@ func TestArrayDecoding(t *testing.T) {
} }
for i, tt := range tests { for i, tt := range tests {
psName := fmt.Sprintf("ps%d", i) err := conn.QueryRow(tt.sql, tt.query).Scan(tt.scan)
mustPrepare(t, conn, psName, tt.sql)
err := conn.QueryRow(psName, tt.query).Scan(tt.scan)
if err != nil { if err != nil {
t.Errorf(`error reading array: %v`, err) t.Errorf(`%d. error reading array: %v`, i, err)
} }
tt.assert(t, tt.query, tt.scan) tt.assert(t, tt.query, tt.scan)
ensureConnValid(t, conn) ensureConnValid(t, conn)
@ -396,12 +363,9 @@ func TestNullXMismatch(t *testing.T) {
} }
for i, tt := range tests { for i, tt := range tests {
psName := fmt.Sprintf("ps%d", i)
mustPrepare(t, conn, psName, tt.sql)
actual = zero actual = zero
err := conn.QueryRow(psName, tt.queryArgs...).Scan(tt.scanArgs...) err := conn.QueryRow(tt.sql, tt.queryArgs...).Scan(tt.scanArgs...)
if err == nil || !strings.Contains(err.Error(), tt.err) { if err == nil || !strings.Contains(err.Error(), tt.err) {
t.Errorf(`%d. Expected error to contain "%s", but it didn't: %v`, i, tt.err, err) t.Errorf(`%d. Expected error to contain "%s", but it didn't: %v`, i, tt.err, err)
} }