83 lines
2.6 KiB
Go
83 lines
2.6 KiB
Go
package test
|
|
|
|
import (
|
|
"net/http"
|
|
"testing"
|
|
|
|
"github.com/stretchr/testify/assert"
|
|
"github.guxuan/haibei/internal/mods/rbac/schema"
|
|
"github.guxuan/haibei/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,
|
|
Status: schema.RoleStatusEnabled,
|
|
}
|
|
|
|
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(roleFormItem.Status, role.Status)
|
|
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
|
|
role.Status = newStatus
|
|
e.PUT(baseAPI + "/roles/" + role.ID).WithJSON(role).Expect().Status(http.StatusOK)
|
|
|
|
var getRole schema.Role
|
|
e.GET(baseAPI + "/roles/" + 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/" + role.ID).Expect().Status(http.StatusOK)
|
|
e.GET(baseAPI + "/roles/" + 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)
|
|
}
|