4
1
Fork 0
backend-core/testutils_test.go

26 lines
562 B
Go
Raw Normal View History

2019-08-31 23:03:56 +08:00
package main
import (
"database/sql"
"net/http/httptest"
"testing"
)
func assertCode(t *testing.T, w *httptest.ResponseRecorder, code int) {
if got, want := w.Code, code; want != got {
t.Errorf("Want response code %d, got %d", want, got)
}
}
func assertDB(t *testing.T, db *sql.DB, query string, args ...interface{}) {
rows, err := db.Query(query, args...)
2019-08-31 23:03:56 +08:00
if err != nil {
t.Errorf("Error during query %s: %s", query, err)
return
2019-08-31 23:03:56 +08:00
}
2019-08-31 23:32:42 +08:00
defer rows.Close()
2019-08-31 23:03:56 +08:00
if rows.Next() != true {
t.Errorf("Want one result, found none for query %s", query)
}
}