update: test_utils connection string

cudoventures · Aug 25, 2023 · df42f1f · df42f1f
1 parent 5e4334b
commit df42f1f
Showing 1 changed file with 8 additions and 1 deletion.
diff --git a/utils/test_utils.go b/utils/test_utils.go
@@ -18,8 +18,9 @@ import (
 )
 
 func NewTestDb(schema string) (*database.Db, error) {
+	connStr := fmt.Sprintf("postgresql://bdjuno:password@localhost:6433/bdjuno?sslmode=disable&search_path=%s", schema)
 	dbCfg := dbconfig.NewDatabaseConfig(
-		"postgresql://bdjuno:password@localhost:6433/bdjuno?sslmode=disable&search_path=public",
+		connStr,
 		"",
 		"",
 		"",
@@ -61,6 +62,12 @@ func NewTestDb(schema string) (*database.Db, error) {
 			return err
 		}
 
+		// Create a default partition for the transaction table
+		_, err = bigDipperDb.SQL.Exec(fmt.Sprintf(`CREATE TABLE IF NOT EXISTS %s.transaction_default PARTITION OF transaction DEFAULT;`, schema))
+		if err != nil {
+			return err
+		}
+
 		return nil
 	})