diff --git a/backend/cmd/api/main.go b/backend/cmd/api/main.go index ae79ed6..2d79636 100644 --- a/backend/cmd/api/main.go +++ b/backend/cmd/api/main.go @@ -23,7 +23,7 @@ func main() { if err != nil { log.Fatal(err) } - if err := initdb.InitUser(models.DBService); err != nil { + if err := initdb.New(models.DBService); err != nil { log.Fatal(err) } diff --git a/backend/internal/initdb/new.go b/backend/internal/initdb/new.go new file mode 100644 index 0000000..cd59ef1 --- /dev/null +++ b/backend/internal/initdb/new.go @@ -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) +} diff --git a/backend/internal/initdb/user_test.go b/backend/internal/initdb/user_test.go index 5a2b12a..191f2b1 100644 --- a/backend/internal/initdb/user_test.go +++ b/backend/internal/initdb/user_test.go @@ -13,7 +13,7 @@ import ( func TestInitUserCreatesRootUserWhenTableIsEmpty(t *testing.T) { database := newUserTestDatabase(t) - require.NoError(t, InitUser(database)) + require.NoError(t, New(database)) var users []models.SenlinAgentUser 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, InitUser(database)) + require.NoError(t, New(database)) var users []models.SenlinAgentUser require.NoError(t, database.Find(&users).Error)