fix: avoid company writes for employee serials

This commit is contained in:
Frudrax Cheng
2026-06-06 14:07:17 +08:00
parent a9aa4a1318
commit f80ab4a8ce
3 changed files with 37 additions and 32 deletions
+8 -6
View File
@@ -248,7 +248,7 @@ func TestEmployeeSerialsService_Generate_Success(t *testing.T) {
database.DB.Unscoped().Delete(&user)
}
func TestEmployeeSerialsService_Generate_CreateNewCompany(t *testing.T) {
func TestEmployeeSerialsService_Generate_DoesNotCreateManagedCompany(t *testing.T) {
var user models.User
password, _ := bcrypt.GenerateFromPassword([]byte("password123"), bcrypt.DefaultCost)
user = models.User{
@@ -266,11 +266,9 @@ func TestEmployeeSerialsService_Generate_CreateNewCompany(t *testing.T) {
assert.NoError(t, err)
assert.Len(t, serials, 3)
var company models.Company
result := database.DB.Where("company_name = ?", "NewEmpCompany").First(&company)
assert.NoError(t, result.Error)
assert.Equal(t, "NewEmpCompany", company.CompanyName)
assert.True(t, company.IsActive)
var count int64
database.DB.Model(&models.Company{}).Where("company_name = ?", "NewEmpCompany").Count(&count)
assert.Equal(t, int64(0), count)
for _, serial := range serials {
database.DB.Unscoped().Delete(&serial)
@@ -419,6 +417,10 @@ func TestEmployeeSerialsService_Update_Success(t *testing.T) {
assert.Equal(t, "新名字", result.EmployeeName)
assert.False(t, result.IsActive)
var count int64
database.DB.Model(&models.Company{}).Where("company_name = ?", "UpdatedEmpCompany").Count(&count)
assert.Equal(t, int64(0), count)
for _, serial := range serials {
database.DB.Unscoped().Delete(&serial)
}