refactor: add database initialization entrypoint
This commit is contained in:
@@ -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)
|
||||
}
|
||||
|
||||
|
||||
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) {
|
||||
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)
|
||||
|
||||
Reference in New Issue
Block a user