Skip to content

Commit c2af4eb

Browse files
committed
readded fail
1 parent 6e5dc88 commit c2af4eb

File tree

1 file changed

+9
-8
lines changed

1 file changed

+9
-8
lines changed

driver_test.go

Lines changed: 9 additions & 8 deletions
Original file line numberDiff line numberDiff line change
@@ -126,24 +126,25 @@ func runTests(t *testing.T, name string, tests ...func(dbt *DBTest)) {
126126
dbt.mustExec("DROP TABLE IF EXISTS test")
127127
}
128128

129+
func (dbt *DBTest) fail(method, query string, err error) {
130+
if len(query) > 300 {
131+
query = "[query too large to print]"
132+
}
133+
dbt.Fatalf("Error on %s %s: %v", method, query, err)
134+
}
135+
129136
func (dbt *DBTest) mustExec(query string, args ...interface{}) (res sql.Result) {
130137
res, err := dbt.db.Exec(query, args...)
131138
if err != nil {
132-
if len(query) > 300 {
133-
query = "[query too large to print]"
134-
}
135-
dbt.Fatalf("Error on Exec %s: %v", query, err)
139+
dbt.fail("Exec", query, err)
136140
}
137141
return res
138142
}
139143

140144
func (dbt *DBTest) mustQuery(query string, args ...interface{}) (rows *sql.Rows) {
141145
rows, err := dbt.db.Query(query, args...)
142146
if err != nil {
143-
if len(query) > 300 {
144-
query = "[query too large to print]"
145-
}
146-
dbt.Fatalf("Error on Query %s: %v", query, err)
147+
dbt.fail("Query", query, err)
147148
}
148149
return rows
149150
}

0 commit comments

Comments
 (0)