Don't stop test on MySQLWarnings

methane · methane · commit 42a1efd12acf · 2015-02-08T17:18:00.000+09:00
"DROP TABLE IF EXISTS ..." query fails on fresh database.
diff --git a/benchmark_test.go b/benchmark_test.go
@@ -45,7 +45,11 @@ func initDB(b *testing.B, queries ...string) *sql.DB {
 	db := tb.checkDB(sql.Open("mysql", dsn))
 	for _, query := range queries {
 		if _, err := db.Exec(query); err != nil {
-			b.Fatalf("Error on %q: %v", query, err)
+			if w, ok := err.(MySQLWarnings); ok {
+				b.Logf("Warning on %q: %v", query, w)
+			} else {
+				b.Fatalf("Error on %q: %v", query, err)
+			}
 		}
 	}
 	return db

Original file line number	Diff line number	Diff line change
`@@ -45,7 +45,11 @@ func initDB(b testing.B, queries ...string) sql.DB {`
`45`	`45`	`db := tb.checkDB(sql.Open("mysql", dsn))`
`46`	`46`	`for _, query := range queries {`
`47`	`47`	`if _, err := db.Exec(query); err != nil {`
`48`		`- b.Fatalf("Error on %q: %v", query, err)`
	`48`	`+ if w, ok := err.(MySQLWarnings); ok {`
	`49`	`+ b.Logf("Warning on %q: %v", query, w)`
	`50`	`+ } else {`
	`51`	`+ b.Fatalf("Error on %q: %v", query, err)`
	`52`	`+ }`
`49`	`53`	`}`
`50`	`54`	`}`
`51`	`55`	`return db`