refactor customer_test.go

core/sqlite-init/sqlite_init.go

@@ -7,6 +7,7 @@ import (
 	_ "github.com/mattn/go-sqlite3" // Import the SQLite3 driver
 	"io/ioutil"
 	"log"
+	"os"
 	"path/filepath"
 	"strings"
 )
@@ -24,7 +25,9 @@ type Root struct {
 }
 
 func main() {
-	dbPaths := []string{"/Users/donov/databases/test.db", "/Users/donov/databases/nkode.db"}
+	testDbPath := os.Getenv("TEST_DB_PATH")
+	dbPath := os.Getenv("DB_PATH")
+	dbPaths := []string{testDbPath, dbPath}
 	outputStr := MakeSvgFiles()
 	for _, path := range dbPaths {
 		MakeTables(path)