refactor: add database initialization entrypoint
This commit is contained in:
@@ -23,7 +23,7 @@ func main() {
|
|||||||
if err != nil {
|
if err != nil {
|
||||||
log.Fatal(err)
|
log.Fatal(err)
|
||||||
}
|
}
|
||||||
if err := initdb.InitUser(models.DBService); err != nil {
|
if err := initdb.New(models.DBService); err != nil {
|
||||||
log.Fatal(err)
|
log.Fatal(err)
|
||||||
}
|
}
|
||||||
|
|
||||||
|
|||||||
8
backend/internal/initdb/new.go
Normal file
8
backend/internal/initdb/new.go
Normal file
@@ -0,0 +1,8 @@
|
|||||||
|
package initdb
|
||||||
|
|
||||||
|
import "gorm.io/gorm"
|
||||||
|
|
||||||
|
// New initializes the default database records.
|
||||||
|
func New(database *gorm.DB) error {
|
||||||
|
return InitUser(database)
|
||||||
|
}
|
||||||
@@ -13,7 +13,7 @@ import (
|
|||||||
func TestInitUserCreatesRootUserWhenTableIsEmpty(t *testing.T) {
|
func TestInitUserCreatesRootUserWhenTableIsEmpty(t *testing.T) {
|
||||||
database := newUserTestDatabase(t)
|
database := newUserTestDatabase(t)
|
||||||
|
|
||||||
require.NoError(t, InitUser(database))
|
require.NoError(t, New(database))
|
||||||
|
|
||||||
var users []models.SenlinAgentUser
|
var users []models.SenlinAgentUser
|
||||||
require.NoError(t, database.Find(&users).Error)
|
require.NoError(t, database.Find(&users).Error)
|
||||||
@@ -34,7 +34,7 @@ func TestInitUserDoesNothingWhenTableIsNotEmpty(t *testing.T) {
|
|||||||
}
|
}
|
||||||
require.NoError(t, database.Create(&existing).Error)
|
require.NoError(t, database.Create(&existing).Error)
|
||||||
|
|
||||||
require.NoError(t, InitUser(database))
|
require.NoError(t, New(database))
|
||||||
|
|
||||||
var users []models.SenlinAgentUser
|
var users []models.SenlinAgentUser
|
||||||
require.NoError(t, database.Find(&users).Error)
|
require.NoError(t, database.Find(&users).Error)
|
||||||
|
|||||||
Reference in New Issue
Block a user