Improved error checking in tests

Author: ankane

bun_test.go

@@ -79,10 +79,17 @@ func TestBun(t *testing.T) {
 	sqldb := sql.OpenDB(pgconn)
 	db := bun.NewDB(sqldb, pgdialect.New())
 
-	db.Exec("CREATE EXTENSION IF NOT EXISTS vector")
-	db.Exec("DROP TABLE IF EXISTS bun_items")
+	_, err := db.Exec("CREATE EXTENSION IF NOT EXISTS vector")
+	if err != nil {
+		panic(err)
+	}
+
+	_, err = db.Exec("DROP TABLE IF EXISTS bun_items")
+	if err != nil {
+		panic(err)
+	}
 
-	_, err := db.NewCreateTable().Model((*BunItem)(nil)).Exec(ctx)
+	_, err = db.NewCreateTable().Model((*BunItem)(nil)).Exec(ctx)
 	if err != nil {
 		panic(err)
 	}

gorm_test.go

@@ -57,7 +57,10 @@ func TestGorm(t *testing.T) {
 	db.Exec("CREATE EXTENSION IF NOT EXISTS vector")
 	db.Exec("DROP TABLE IF EXISTS gorm_items")
 
-	db.AutoMigrate(&GormItem{})
+	err = db.AutoMigrate(&GormItem{})
+	if err != nil {
+		panic(err)
+	}
 
 	db.Exec("CREATE INDEX ON gorm_items USING hnsw (embedding vector_l2_ops)")
 

pg_test.go

@@ -62,10 +62,17 @@ func TestPg(t *testing.T) {
 	})
 	defer db.Close()
 
-	db.Exec("CREATE EXTENSION IF NOT EXISTS vector")
-	db.Exec("DROP TABLE IF EXISTS pg_items")
+	_, err := db.Exec("CREATE EXTENSION IF NOT EXISTS vector")
+	if err != nil {
+		panic(err)
+	}
+
+	_, err = db.Exec("DROP TABLE IF EXISTS pg_items")
+	if err != nil {
+		panic(err)
+	}
 
-	err := db.Model((*PgItem)(nil)).CreateTable(&orm.CreateTableOptions{})
+	err = db.Model((*PgItem)(nil)).CreateTable(&orm.CreateTableOptions{})
 	if err != nil {
 		panic(err)
 	}

sqlx_test.go

@@ -58,7 +58,10 @@ func TestSqlx(t *testing.T) {
 	CreateSqlxItems(db)
 
 	var items []SqlxItem
-	db.Select(&items, "SELECT * FROM sqlx_items ORDER BY embedding <-> $1 LIMIT 5", pgvector.NewVector([]float32{1, 1, 1}))
+	err := db.Select(&items, "SELECT * FROM sqlx_items ORDER BY embedding <-> $1 LIMIT 5", pgvector.NewVector([]float32{1, 1, 1}))
+	if err != nil {
+		panic(err)
+	}
 	if items[0].Id != 1 || items[1].Id != 3 || items[2].Id != 2 {
 		t.Error()
 	}