diff --git a/xm-core/src/main/java/com/xm/core/ctrl/XmTestCasedbController.java b/xm-core/src/main/java/com/xm/core/ctrl/XmTestCasedbController.java index d0a062fe..197f0639 100644 --- a/xm-core/src/main/java/com/xm/core/ctrl/XmTestCasedbController.java +++ b/xm-core/src/main/java/com/xm/core/ctrl/XmTestCasedbController.java @@ -60,6 +60,8 @@ public class XmTestCasedbController { Map m = new HashMap<>(); Tips tips=new Tips("查询成功"); RequestUtils.transformArray(xmTestCasedb, "ids"); + User user=LoginUtils.getCurrentUserInfo(); + xmTestCasedb.put("cbranchId",user.getBranchId()); PageUtils.startPage(xmTestCasedb); List> xmTestCasedbList = xmTestCasedbService.selectListMapByWhere(xmTestCasedb); //列出XmTestCasedb列表 PageUtils.responePage(m, xmTestCasedbList); @@ -90,6 +92,14 @@ public class XmTestCasedbController { return failed("pk-exists","编号重复,请修改编号再提交"); } } + if(!StringUtils.hasText(xmTestCasedb.getProductId())){ + return failed("productId-0","产品编号不能为空"); + } + User user=LoginUtils.getCurrentUserInfo(); + xmTestCasedb.setCtime(new Date()); + xmTestCasedb.setCuserid(user.getUserid()); + xmTestCasedb.setCusername(user.getUsername()); + xmTestCasedb.setCbranchId(user.getBranchId()); xmTestCasedbService.insert(xmTestCasedb); m.put("data",xmTestCasedb); }catch (BizException e) { @@ -119,6 +129,10 @@ public class XmTestCasedbController { if( xmTestCasedbDb == null ){ return failed("data-not-exists","数据不存在,无法删除"); } + User user=LoginUtils.getCurrentUserInfo(); + if(!user.getBranchId().equals(xmTestCasedbDb.getCbranchId())){ + return failed("cbranchId-err","该测试库不属于您企业,不能删除"); + } xmTestCasedbService.deleteByPk(xmTestCasedb); }catch (BizException e) { tips=e.getTips(); @@ -147,6 +161,11 @@ public class XmTestCasedbController { if( xmTestCasedbDb == null ){ return failed("data-not-exists","数据不存在,无法修改"); } + + User user=LoginUtils.getCurrentUserInfo(); + if(!user.getBranchId().equals(xmTestCasedbDb.getCbranchId())){ + return failed("cbranchId-err","该测试库不属于您企业,不能修改"); + } xmTestCasedbService.updateSomeFieldByPk(xmTestCasedb); m.put("data",xmTestCasedb); }catch (BizException e) { @@ -197,7 +216,10 @@ public class XmTestCasedbController { List no=new ArrayList<>(); User user = LoginUtils.getCurrentUserInfo(); for (XmTestCasedb xmTestCasedbDb : xmTestCasedbsDb) { - Tips tips2 = new Tips("检查通过"); + Tips tips2 = new Tips("检查通过"); + if(!user.getBranchId().equals(xmTestCasedbDb.getCbranchId())){ + return failed("cbranchId-err","该测试库不属于您企业,不能修改"); + } if(!tips2.isOk()){ no.add(xmTestCasedbDb); }else{ @@ -248,7 +270,12 @@ public class XmTestCasedbController { List can=new ArrayList<>(); List no=new ArrayList<>(); + + User user=LoginUtils.getCurrentUserInfo(); for (XmTestCasedb data : datasDb) { + if(!user.getBranchId().equals(data.getCbranchId())){ + return failed("cbranchId-err","该测试库不属于您企业,不能删除"); + } if(true){ can.add(data); }else{ diff --git a/xm-core/src/main/java/com/xm/core/ctrl/XmTestPlanController.java b/xm-core/src/main/java/com/xm/core/ctrl/XmTestPlanController.java index db9c200a..67122654 100644 --- a/xm-core/src/main/java/com/xm/core/ctrl/XmTestPlanController.java +++ b/xm-core/src/main/java/com/xm/core/ctrl/XmTestPlanController.java @@ -60,6 +60,10 @@ public class XmTestPlanController { Map m = new HashMap<>(); Tips tips=new Tips("查询成功"); RequestUtils.transformArray(xmTestPlan, "ids"); + User user=LoginUtils.getCurrentUserInfo(); + if(!user.getBranchId().equals(xmTestCasedbDb.getCbranchId())){ + return failed("cbranchId-err","该测试库不属于您企业,不能修改"); + } PageUtils.startPage(xmTestPlan); List> xmTestPlanList = xmTestPlanService.selectListMapByWhere(xmTestPlan); //列出XmTestPlan列表 PageUtils.responePage(m, xmTestPlanList); @@ -90,6 +94,21 @@ public class XmTestPlanController { return failed("pk-exists","编号重复,请修改编号再提交"); } } + if(!StringUtils.hasText(xmTestPlan.getProjectId())){ + return failed("projectId-0","项目编号不能为空"); + } + + if(!StringUtils.hasText(xmTestPlan.getProductId())){ + return failed("productId-0","产品编号不能为空"); + } + + if(!StringUtils.hasText(xmTestPlan.getCasedbId())){ + return failed("casedbId-0","测试用例库编号不能为空"); + } + User user=LoginUtils.getCurrentUserInfo(); + xmTestPlan.setCuserid(user.getUserid()); + xmTestPlan.setCusername(user.getUsername()); + xmTestPlan.setCtime(new Date()); xmTestPlanService.insert(xmTestPlan); m.put("data",xmTestPlan); }catch (BizException e) {