95 lines
3.4 KiB
Go
95 lines
3.4 KiB
Go
package repository
|
|
|
|
import (
|
|
"errors"
|
|
"regexp"
|
|
"testing"
|
|
"time"
|
|
|
|
"github.com/DATA-DOG/go-sqlmock"
|
|
"github.com/wmms/wmms/pkg/database"
|
|
)
|
|
|
|
func TestListRolesIncludesChinesePermissionNames(t *testing.T) {
|
|
db, mock, err := sqlmock.New()
|
|
if err != nil {
|
|
t.Fatalf("open sql mock: %v", err)
|
|
}
|
|
t.Cleanup(func() { database.DB = nil; db.Close() })
|
|
database.DB = db
|
|
now := time.Date(2026, 9, 16, 9, 0, 0, 0, time.Local)
|
|
|
|
roleQuery := `SELECT id, code, name, enabled, created_at, updated_at FROM roles ORDER BY id`
|
|
mock.ExpectQuery(regexp.QuoteMeta(roleQuery)).WillReturnRows(
|
|
sqlmock.NewRows([]string{"id", "code", "name", "enabled", "created_at", "updated_at"}).
|
|
AddRow(2, "warehouse_admin", "仓库管理员", true, now, now),
|
|
)
|
|
permissionQuery := `SELECT rp.role_id, p.id, p.code, p.name
|
|
FROM role_permissions rp JOIN permissions p ON p.id = rp.permission_id
|
|
ORDER BY rp.role_id, p.code`
|
|
mock.ExpectQuery(regexp.QuoteMeta(permissionQuery)).WillReturnRows(
|
|
sqlmock.NewRows([]string{"role_id", "id", "code", "name"}).
|
|
AddRow(2, 12, "inventory.stock.read", "查看库存"),
|
|
)
|
|
|
|
roles, err := NewOrganizationRepository().ListRoles()
|
|
if err != nil {
|
|
t.Fatalf("list roles: %v", err)
|
|
}
|
|
if len(roles) != 1 || len(roles[0].Permissions) != 1 {
|
|
t.Fatalf("unexpected roles: %+v", roles)
|
|
}
|
|
if roles[0].Permissions[0].Name != "查看库存" {
|
|
t.Fatalf("permission name not returned: %+v", roles[0].Permissions[0])
|
|
}
|
|
if err := mock.ExpectationsWereMet(); err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
}
|
|
|
|
func TestUpdateRoleRejectsSuperAdminChanges(t *testing.T) {
|
|
db, mock, err := sqlmock.New()
|
|
if err != nil {
|
|
t.Fatalf("open sql mock: %v", err)
|
|
}
|
|
t.Cleanup(func() { database.DB = nil; db.Close() })
|
|
database.DB = db
|
|
|
|
mock.ExpectBegin()
|
|
mock.ExpectQuery(regexp.QuoteMeta(`SELECT code FROM roles WHERE id = ? FOR UPDATE`)).WithArgs(int64(1)).WillReturnRows(sqlmock.NewRows([]string{"code"}).AddRow("super_admin"))
|
|
mock.ExpectRollback()
|
|
|
|
err = NewOrganizationRepository().UpdateRole(1, "超级管理员", false, nil)
|
|
if !errors.Is(err, ErrProtectedRole) {
|
|
t.Fatalf("expected protected role error, got %v", err)
|
|
}
|
|
if err := mock.ExpectationsWereMet(); err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
}
|
|
|
|
func TestUpdateRoleReplacesPermissionsInTransaction(t *testing.T) {
|
|
db, mock, err := sqlmock.New()
|
|
if err != nil {
|
|
t.Fatalf("open sql mock: %v", err)
|
|
}
|
|
t.Cleanup(func() { database.DB = nil; db.Close() })
|
|
database.DB = db
|
|
|
|
mock.ExpectBegin()
|
|
mock.ExpectQuery(regexp.QuoteMeta(`SELECT code FROM roles WHERE id = ? FOR UPDATE`)).WithArgs(int64(3)).WillReturnRows(sqlmock.NewRows([]string{"code"}).AddRow("purchaser"))
|
|
mock.ExpectExec(regexp.QuoteMeta(`UPDATE roles SET name = ?, enabled = ? WHERE id = ?`)).WithArgs("采购人员", true, int64(3)).WillReturnResult(sqlmock.NewResult(0, 1))
|
|
mock.ExpectExec(regexp.QuoteMeta(`DELETE FROM role_permissions WHERE role_id = ?`)).WithArgs(int64(3)).WillReturnResult(sqlmock.NewResult(0, 2))
|
|
mock.ExpectExec(regexp.QuoteMeta(`INSERT INTO role_permissions (role_id, permission_id)
|
|
SELECT ?, id FROM permissions WHERE code IN (?, ?)`)).WithArgs(int64(3), "inventory.stock.read", "partner.supplier.read").WillReturnResult(sqlmock.NewResult(0, 2))
|
|
mock.ExpectCommit()
|
|
|
|
err = NewOrganizationRepository().UpdateRole(3, "采购人员", true, []string{"inventory.stock.read", "partner.supplier.read"})
|
|
if err != nil {
|
|
t.Fatalf("update role: %v", err)
|
|
}
|
|
if err := mock.ExpectationsWereMet(); err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
}
|