wangbing
5 years ago
parent
8c7151f64a
commit
def805685f
@ -0,0 +1,173 @@
|
||||
package ${basePackage}.datainit;
|
||||
|
||||
import org.junit.Test;
|
||||
import org.junit.runner.RunWith;
|
||||
import org.springframework.beans.factory.annotation.Autowired;
|
||||
import org.springframework.boot.test.context.SpringBootTest;
|
||||
import org.springframework.test.annotation.Rollback;
|
||||
import org.springframework.test.context.junit4.SpringRunner;
|
||||
import org.springframework.transaction.annotation.Transactional;
|
||||
import ${basePackage}.frame.auth.Token;
|
||||
import ${basePackage}.frame.utils.MapperUtil;
|
||||
import ${basePackage}.module.system.mgr.ResManager;
|
||||
import ${basePackage}.module.system.mgr.RoleManager;
|
||||
import ${basePackage}.module.system.mgr.RoleResManager;
|
||||
import ${basePackage}.module.system.req.ResCreateRequest;
|
||||
import ${basePackage}.module.system.req.RoleCreateRequest;
|
||||
import ${basePackage}.module.system.req.RoleResCheckRequest;
|
||||
import ${basePackage}.module.system.rsp.ResCreateResponse;
|
||||
import ${basePackage}.module.system.rsp.RoleCreateResponse;
|
||||
import ${basePackage}.module.system.rsp.RoleResCheckResponse;
|
||||
|
||||
import static org.junit.Assert.assertTrue;
|
||||
|
||||
@RunWith(SpringRunner.class)
|
||||
@SpringBootTest
|
||||
@Transactional
|
||||
public class RoleInit {
|
||||
|
||||
@Autowired
|
||||
private ResManager resManager;
|
||||
@Autowired
|
||||
private RoleManager roleManager;
|
||||
@Autowired
|
||||
private RoleResManager roleCreateRequest;
|
||||
@Autowired
|
||||
private Token token;
|
||||
|
||||
/**
|
||||
* 字典数据初始化
|
||||
*/
|
||||
@Test
|
||||
@Rollback(false)
|
||||
public void init() {
|
||||
String roleCode = "management";
|
||||
String roleName = "管理员";
|
||||
String roleComment = "管理员";
|
||||
Long roleId = null;
|
||||
|
||||
{//初始化角色
|
||||
roleId = createRole(roleCode, roleName, roleComment);
|
||||
}
|
||||
|
||||
{// 初始化核心管理资源，涉及系统运行，特殊需要再放开
|
||||
// createRes(roleId, roleCode, "WEB_DICT", "字典管理", "网页地址", "/admin/system/dict.htm");
|
||||
// createRes(roleId, roleCode, "WEB_DICT_CREATE", "字典新建", "异步请求", "/ajax/system/dict/create");
|
||||
// createRes(roleId, roleCode, "WEB_DICT_DELETE", "字典删除", "异步请求", "/ajax/system/dict/delete");
|
||||
// createRes(roleId, roleCode, "WEB_DICT_UPDATE", "字典更新", "异步请求", "/ajax/system/dict/update");
|
||||
// createRes(roleId, roleCode, "WEB_DICT_FIND", "字典查询", "异步请求", "/ajax/system/dict/find");
|
||||
// createRes(roleId, roleCode, "WEB_DICT_GET", "字典获取", "异步请求", "/ajax/system/dict/get");
|
||||
// createRes(roleId, roleCode, "WEB_DICT_LOAD", "字典详情", "异步请求", "/ajax/system/dict/load");
|
||||
// createRes(roleId, roleCode, "WEB_DICT_IMPORT", "字典导入", "异步请求", "/ajax/system/dict/imports");
|
||||
// createRes(roleId, roleCode, "WEB_DICT_EXPORT", "字典导出", "异步请求", "/ajax/system/dict/exports");
|
||||
// createRes(roleId, roleCode, "WEB_DICT_ITEM", "字典项管理", "网页地址", "/admin/system/dictItem.htm");
|
||||
// createRes(roleId, roleCode, "WEB_DICT_ITEM_CREATE", "字典项新增", "异步请求", "/ajax/system/dictItem/create");
|
||||
// createRes(roleId, roleCode, "WEB_DICT_ITEM_DELETE", "字典项删除", "异步请求", "/ajax/system/dictItem/delete");
|
||||
// createRes(roleId, roleCode, "WEB_DICT_ITEM_UPDATE", "字典项更新", "异步请求", "/ajax/system/dictItem/update");
|
||||
// createRes(roleId, roleCode, "WEB_DICT_ITEM_FIND", "字典项查找", "异步请求", "/ajax/system/dictItem/find");
|
||||
// createRes(roleId, roleCode, "WEB_DICT_ITEM_GET", "字典项获取", "异步请求", "/ajax/system/dictItem/get");
|
||||
// createRes(roleId, roleCode, "WEB_RES", "资源管理", "网页地址", "/admin/system/res.htm");
|
||||
// createRes(roleId, roleCode, "WEB_RES_CREATE", "资源新增", "异步请求", "/ajax/system/res/create");
|
||||
// createRes(roleId, roleCode, "WEB_RES_DELETE", "资源删除", "异步请求", "/ajax/system/res/delete");
|
||||
// createRes(roleId, roleCode, "WEB_RES_UPDATE", "资源编辑", "异步请求", "/ajax/system/res/update");
|
||||
// createRes(roleId, roleCode, "WEB_RES_FIND", "资源查询", "异步请求", "/ajax/system/res/find");
|
||||
// createRes(roleId, roleCode, "WEB_RES_GET", "资源获取", "异步请求", "/ajax/system/res/get");
|
||||
// createRes(roleId, roleCode, "WEB_RES_TREE", "资源树查询", "异步请求", "/ajax/system/res/tree");
|
||||
// createRes(roleId, roleCode, "WEB_LOGERR", "错误日志", "网页地址", "/admin/system/logErr.htm");
|
||||
// createRes(roleId, roleCode, "WEB_LOGERR_DELETE", "错误日志删除", "异步请求", "/ajax/system/logErr/delete");
|
||||
// createRes(roleId, roleCode, "WEB_LOGERR_UPDATE", "错误日志更新", "异步请求", "/ajax/system/logErr/update");
|
||||
// createRes(roleId, roleCode, "WEB_LOGERR_FIND", "错误日志查询", "异步请求", "/ajax/system/logErr/find");
|
||||
}
|
||||
|
||||
{// 初始化一般管理资源
|
||||
createRes(roleId, roleCode, "WEB", "网页资源", "目录", "");
|
||||
createRes(roleId, roleCode, "WEB_INDEX", "主页", "网页地址", "/admin/index.htm");
|
||||
createRes(roleId, roleCode, "WEB_INDEX_HOME", "首页", "网页地址", "/admin/home.htm");
|
||||
createRes(roleId, roleCode, "WEB_INDEX_ROOT", "根页", "网页地址", "/");
|
||||
|
||||
createRes(roleId, roleCode, "WEB_USER", "用户管理", "网页地址", "/admin/system/user.htm");
|
||||
createRes(roleId, roleCode, "WEB_USER_CREATE", "用户新增", "异步请求", "/ajax/system/user/create");
|
||||
createRes(roleId, roleCode, "WEB_USER_DELETE", "用户删除", "异步请求", "/ajax/system/user/delete");
|
||||
createRes(roleId, roleCode, "WEB_USER_UPDATE", "用户更新", "异步请求", "/ajax/system/user/update");
|
||||
createRes(roleId, roleCode, "WEB_USER_FIND", "用户查询", "异步请求", "/ajax/system/user/find");
|
||||
createRes(roleId, roleCode, "WEB_USER_GET", "用户获取", "异步请求", "/ajax/system/user/get");
|
||||
createRes(roleId, roleCode, "WEB_USER_IMPORT", "用户导入", "异步请求", "/ajax/system/user/imports");
|
||||
createRes(roleId, roleCode, "WEB_USER_EXPORT", "用户导出", "异步请求", "/ajax/system/user/exports");
|
||||
|
||||
createRes(roleId, roleCode, "WEB_DEPT", "机构管理", "网页地址", "/admin/system/dept.htm");
|
||||
createRes(roleId, roleCode, "WEB_DEPT_CREATE", "机构新增", "异步请求", "/ajax/system/dept/create");
|
||||
createRes(roleId, roleCode, "WEB_DEPT_DELETE", "机构删除", "异步请求", "/ajax/system/dept/delete");
|
||||
createRes(roleId, roleCode, "WEB_DEPT_UPDATE", "机构更新", "异步请求", "/ajax/system/dept/update");
|
||||
createRes(roleId, roleCode, "WEB_DEPT_FIND", "机构查询", "异步请求", "/ajax/system/dept/find");
|
||||
createRes(roleId, roleCode, "WEB_DEPT_GET", "机构获取", "异步请求", "/ajax/system/dept/get");
|
||||
createRes(roleId, roleCode, "WEB_DEPT_TREE", "机构树查询", "异步请求", "/ajax/system/dept/tree");
|
||||
createRes(roleId, roleCode, "WEB_DEPT_IMPORT", "机构导入", "异步请求", "/ajax/system/dept/imports");
|
||||
createRes(roleId, roleCode, "WEB_DEPT_EXPORT", "机构导出", "异步请求", "/ajax/system/dept/exports");
|
||||
|
||||
createRes(roleId, roleCode, "WEB_FILE", "文件管理", "网页地址", "/admin/system/file.htm");
|
||||
createRes(roleId, roleCode, "WEB_FILE_DELETE", "文件删除", "异步请求", "/ajax/system/file/delete");
|
||||
createRes(roleId, roleCode, "WEB_FILE_FIND", "文件查询", "异步请求", "/ajax/system/file/find");
|
||||
|
||||
createRes(roleId, roleCode, "WEB_ROLE", "角色管理", "网页地址", "/admin/system/role.htm");
|
||||
createRes(roleId, roleCode, "WEB_ROLE_CREATE", "角色新增", "异步请求", "/ajax/system/role/create");
|
||||
createRes(roleId, roleCode, "WEB_ROLE_DELETE", "角色删除", "异步请求", "/ajax/system/role/delete");
|
||||
createRes(roleId, roleCode, "WEB_ROLE_UPDATE", "角色更新", "异步请求", "/ajax/system/role/update");
|
||||
createRes(roleId, roleCode, "WEB_ROLE_FIND", "角色查询", "异步请求", "/ajax/system/role/find");
|
||||
createRes(roleId, roleCode, "WEB_ROLE_GET", "角色获取", "异步请求", "/ajax/system/role/get");
|
||||
createRes(roleId, roleCode, "WEB_ROLE_RESCHECK", "角色资源分配", "异步请求", "/ajax/system/roleRes/check");
|
||||
createRes(roleId, roleCode, "WEB_ROLE_ROLERES", "角色资源关系查询", "异步请求", "/ajax/system/roleRes/find");
|
||||
|
||||
createRes(roleId, roleCode, "WEB_TASK", "计划任务", "网页地址", "/admin/system/schedule.htm");
|
||||
|
||||
createRes(roleId, roleCode, "WEB_TOKENS", "登录凭证管理", "网页地址", "/admin/system/tokens.htm");
|
||||
createRes(roleId, roleCode, "WEB_TOKENS_FIND", "登录凭证查询", "异步请求", "/ajax/system/tokens/find");
|
||||
createRes(roleId, roleCode, "WEB_TOKENS_EXPORT", "凭证导出", "异步请求", "/ajax/system/tokens/exports");
|
||||
createRes(roleId, roleCode, "WEB_TOKENS_LOGOUT", "凭证注销", "异步请求", "/ajax/system/tokens/logout");
|
||||
|
||||
}
|
||||
}
|
||||
|
||||
private void createRes(Long roleId, String roleCode, String code, String name, String resType, String value) {
|
||||
ResCreateRequest resCreateRequest = new ResCreateRequest();
|
||||
resCreateRequest.setResName(name);
|
||||
resCreateRequest.setResCode(code);
|
||||
resCreateRequest.setResType(resType);
|
||||
resCreateRequest.setResValue(value);
|
||||
resCreateRequest.setValid(true);
|
||||
ResCreateResponse resCreateResponse = resManager.create(resCreateRequest, token);
|
||||
if (resCreateResponse.hasError()) {
|
||||
System.err.println("==========>" + MapperUtil.toJson(resCreateResponse));
|
||||
}
|
||||
assertTrue(!resCreateResponse.hasError());
|
||||
allocateRes(roleId, roleCode, resCreateResponse.getId(), code);
|
||||
}
|
||||
|
||||
private Long createRole(String roleCode, String roleName, String roleComment) {
|
||||
RoleCreateRequest roleCreateRequest = new RoleCreateRequest();
|
||||
roleCreateRequest.setCode(roleCode);
|
||||
roleCreateRequest.setName(roleName);
|
||||
roleCreateRequest.setComment(roleComment);
|
||||
RoleCreateResponse roleCreateResponse = roleManager.create(roleCreateRequest, token);
|
||||
if (roleCreateResponse.hasError()) {
|
||||
System.err.println("==========>" + MapperUtil.toJson(roleCreateResponse));
|
||||
}
|
||||
assertTrue(!roleCreateResponse.hasError());
|
||||
return roleCreateResponse.getId();
|
||||
}
|
||||
|
||||
private void allocateRes(Long roleId, String roleCode, Long resId, String resCode) {
|
||||
RoleResCheckRequest roleResCheckRequest = new RoleResCheckRequest();
|
||||
roleResCheckRequest.setRoleId(roleId);
|
||||
roleResCheckRequest.setRoleCode(roleCode);
|
||||
roleResCheckRequest.setResId(resId);
|
||||
roleResCheckRequest.setResCode(resCode);
|
||||
roleResCheckRequest.setChecked(true);
|
||||
|
||||
RoleResCheckResponse roleResCheckResponse = roleCreateRequest.check(roleResCheckRequest, token);
|
||||
if (roleResCheckResponse.hasError()) {
|
||||
System.err.println("==========>" + MapperUtil.toJson(roleResCheckResponse));
|
||||
}
|
||||
assertTrue(!roleResCheckResponse.hasError());
|
||||
}
|
||||
}
|
||||
|
||||
Loading…
Reference in new issue