refactor: add database initialization entrypoint

This commit is contained in:
2026-07-23 13:55:49 +08:00
parent 50274184dc
commit 51ee510112
3 changed files with 11 additions and 3 deletions

View File

@@ -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)
} }

View 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)
}

View File

@@ -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)