fix tests

georgysavva · Apr 7, 2024 · 345c2ee · 345c2ee
1 parent 870ab55
commit 345c2ee
Showing 1 changed file with 6 additions and 3 deletions.
diff --git a/dbscan/dbscan_test.go b/dbscan/dbscan_test.go
@@ -406,7 +406,10 @@ func TestMain(m *testing.M) {
 			panic(err)
 		}
 		defer testDB.Close()
-		prepareTestDB(testDB)
+		err = prepareTestDB(testDB)
+		if err != nil {
+			panic(err)
+		}
 		testAPI, err = getAPI()
 		if err != nil {
 			panic(err)
@@ -417,9 +420,9 @@ func TestMain(m *testing.M) {
 }
 
 func prepareTestDB(testDB *pgxpool.Pool) (err error) {
-	_, err = testDB.Query(ctx, `
+	_, err = testDB.Exec(ctx, `
 		CREATE TYPE test_enum_type AS ENUM ('test_val_1', 'test_val_2');
 	`)
 
-	return
+	return err
 }