ifms_go/test/role_test.go
2025-06-10 17:50:46 +08:00

81 lines
2.6 KiB
Go

package test
import (
"ifms/pkg/convert"
"net/http"
"testing"
"github.com/stretchr/testify/assert"
"ifms/internal/mods/rbac/schema"
"ifms/pkg/util"
)
func TestRole(t *testing.T) {
e := tester(t)
menuFormItem := schema.MenuForm{
Code: "role",
Name: "Role management",
Description: "Role management",
Sequence: 8,
Type: "page",
Path: "/system/role",
Properties: `{"icon":"role"}`,
Status: schema.MenuStatusEnabled,
}
var menu schema.Menu
e.POST(baseAPI + "/menus").WithJSON(menuFormItem).
Expect().Status(http.StatusOK).JSON().Decode(&util.ResponseResult{Data: &menu})
assert := assert.New(t)
assert.NotEmpty(menu.ID)
assert.Equal(menuFormItem.Code, menu.Code)
assert.Equal(menuFormItem.Name, menu.Name)
assert.Equal(menuFormItem.Description, menu.Description)
assert.Equal(menuFormItem.Sequence, menu.Sequence)
assert.Equal(menuFormItem.Type, menu.Type)
assert.Equal(menuFormItem.Path, menu.Path)
assert.Equal(menuFormItem.Properties, menu.Properties)
assert.Equal(menuFormItem.Status, menu.Status)
roleFormItem := schema.RoleForm{
Code: "admin",
Name: "Administrator",
//Menus: schema.RoleMenus{
// {MenuID: menu.ID},
//},
Description: "Administrator",
Sequence: 9,
}
var role schema.Role
e.POST(baseAPI + "/roles").WithJSON(roleFormItem).Expect().Status(http.StatusOK).JSON().Decode(&util.ResponseResult{Data: &role})
assert.NotEmpty(role.ID)
assert.Equal(roleFormItem.Code, role.Code)
assert.Equal(roleFormItem.Name, role.Name)
assert.Equal(roleFormItem.Description, role.Description)
assert.Equal(roleFormItem.Sequence, role.Sequence)
//assert.Equal(len(roleFormItem.Menus), len(role.Menus))
var roles schema.Roles
e.GET(baseAPI + "/roles").Expect().Status(http.StatusOK).JSON().Decode(&util.ResponseResult{Data: &roles})
assert.GreaterOrEqual(len(roles), 1)
newName := "Administrator 1"
newStatus := schema.RoleStatusDisabled
role.Name = newName
e.PUT(baseAPI + "/roles/" + convert.ToString(role.ID)).WithJSON(role).Expect().Status(http.StatusOK)
var getRole schema.Role
e.GET(baseAPI + "/roles/" + convert.ToString(role.ID)).Expect().Status(http.StatusOK).JSON().Decode(&util.ResponseResult{Data: &getRole})
assert.Equal(newName, getRole.Name)
assert.Equal(newStatus, getRole.Status)
e.DELETE(baseAPI + "/roles/" + convert.ToString(role.ID)).Expect().Status(http.StatusOK)
e.GET(baseAPI + "/roles/" + convert.ToString(role.ID)).Expect().Status(http.StatusNotFound)
e.DELETE(baseAPI + "/menus/" + menu.ID).Expect().Status(http.StatusOK)
e.GET(baseAPI + "/menus/" + menu.ID).Expect().Status(http.StatusNotFound)
}