refactor: shorten model and table prefixes
This commit is contained in:
@@ -15,7 +15,7 @@ func TestInitUserCreatesRootUserWhenTableIsEmpty(t *testing.T) {
|
||||
|
||||
require.NoError(t, InitUser(database))
|
||||
|
||||
var users []models.SenlinAgentUser
|
||||
var users []models.SaUser
|
||||
require.NoError(t, database.Find(&users).Error)
|
||||
require.Len(t, users, 1)
|
||||
require.Equal(t, rootUsername, users[0].Email)
|
||||
@@ -26,7 +26,7 @@ func TestInitUserCreatesRootUserWhenTableIsEmpty(t *testing.T) {
|
||||
|
||||
func TestInitUserDoesNothingWhenTableIsNotEmpty(t *testing.T) {
|
||||
database := newUserTestDatabase(t)
|
||||
existing := models.SenlinAgentUser{
|
||||
existing := models.SaUser{
|
||||
Email: "existing@example.com",
|
||||
DisplayName: "Existing User",
|
||||
PasswordHash: "existing-hash",
|
||||
@@ -36,7 +36,7 @@ func TestInitUserDoesNothingWhenTableIsNotEmpty(t *testing.T) {
|
||||
|
||||
require.NoError(t, InitUser(database))
|
||||
|
||||
var users []models.SenlinAgentUser
|
||||
var users []models.SaUser
|
||||
require.NoError(t, database.Find(&users).Error)
|
||||
require.Len(t, users, 1)
|
||||
require.Equal(t, existing.Email, users[0].Email)
|
||||
@@ -47,6 +47,6 @@ func newUserTestDatabase(t *testing.T) *gorm.DB {
|
||||
|
||||
database, err := gorm.Open(sqlite.Open(":memory:"), &gorm.Config{})
|
||||
require.NoError(t, err)
|
||||
require.NoError(t, database.AutoMigrate(&models.SenlinAgentUser{}))
|
||||
require.NoError(t, database.AutoMigrate(&models.SaUser{}))
|
||||
return database
|
||||
}
|
||||
|
||||
Reference in New Issue
Block a user