Get database name from config

hamidfzm · Aug 16, 2018 · 7b77923 · 7b77923
1 parent 0608610
commit 7b77923
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/models/database.go b/models/database.go
@@ -10,7 +10,7 @@ var DB *gorm.DB
 var tables = []interface{}{&User{}, &Time{}}
 
 func SetupDatabase() {
-	if db, err := gorm.Open("sqlite3", "develop.db"); err != nil {
+	if db, err := gorm.Open("sqlite3", helpers.Config.DBName); err != nil {
 		panic("failed to connect database")
 	} else {
 		DB = db
@@ -21,7 +21,7 @@ func SetupDatabase() {
 }
 
 func SetupTestDatabase() {
-	if db, err := gorm.Open("sqlite3", "../test.db"); err != nil {
+	if db, err := gorm.Open("sqlite3", "../data/test.db"); err != nil {
 		panic("failed to connect database")
 	} else {
 		DB = db