test-version22
/
MeterSphere
mirror of https://gitee.com/fit2cloud-feizhiyun/MeterSphere.git
1
0
Fork 0
Code Issues Projects Releases Wiki Activity

fix(接口测试): 场景接口模块导入逻辑重构 (#15368) 

--user=郭雨琦

Co-authored-by: guoyuqi <xiaomeinvG@126.com>
This commit is contained in:
MeterSphere Bot 2022-06-30 11:43:02 +08:00 committed by GitHub
parent f21a4a5c5d
commit f3e2d99fc3
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
3 changed files with 85 additions and 45 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
backend/src/main/java/io/metersphere/api/service/ApiAutomationService.java
View File

@ -1341,6 +1341,9 @@ public class ApiAutomationService {
List<ApiScenarioWithBLOBs> initData = apiImport.getData(); List<ApiScenarioWithBLOBs> initData = apiImport.getData();
currentScenarioOrder.remove(); currentScenarioOrder.remove();
String defaultVersion = extProjectVersionMapper.getDefaultVersion(request.getProjectId());
request.setDefaultVersion(defaultVersion);
UpdateScenarioModuleDTO updateScenarioModuleDTO = apiScenarioModuleService.checkScenarioModule(request, initData, StringUtils.equals("fullCoverage", request.getModeId()), request.getCoverModule()); UpdateScenarioModuleDTO updateScenarioModuleDTO = apiScenarioModuleService.checkScenarioModule(request, initData, StringUtils.equals("fullCoverage", request.getModeId()), request.getCoverModule());
List<ApiScenarioModule> moduleList = updateScenarioModuleDTO.getModuleList(); List<ApiScenarioModule> moduleList = updateScenarioModuleDTO.getModuleList();
List<ApiScenarioWithBLOBs> data = updateScenarioModuleDTO.getApiScenarioWithBLOBsList(); List<ApiScenarioWithBLOBs> data = updateScenarioModuleDTO.getApiScenarioWithBLOBsList();
@ -1359,8 +1362,7 @@ public class ApiAutomationService {
num = getNextNum(data.get(0).getProjectId()); num = getNextNum(data.get(0).getProjectId());
request.setOpenCustomNum(config.getScenarioCustomNum()); request.setOpenCustomNum(config.getScenarioCustomNum());
} }
String defaultVersion = extProjectVersionMapper.getDefaultVersion(request.getProjectId());
request.setDefaultVersion(defaultVersion);
for (int i = 0; i < data.size(); i++) { for (int i = 0; i < data.size(); i++) {
ApiScenarioWithBLOBs item = data.get(i); ApiScenarioWithBLOBs item = data.get(i);

55
backend/src/main/java/io/metersphere/api/service/ApiModuleService.java
View File

@ -917,23 +917,23 @@ public class ApiModuleService extends NodeTreeService<ApiModuleDTO> {
} }
} }
private ApiModule getMinModule(String[] tagTree, List<ApiModule> moduleList, ApiModule parentModule, Map<String, List<ApiModule>> pidChildrenMap, Map<String, ApiModule> moduleMap private ApiModule getMinModule(String[] tagTree, List<ApiModule> parentModuleList, ApiModule parentModule, Map<String, List<ApiModule>> pidChildrenMap, Map<String, ApiModule> moduleMap
, Map<String, String> idPathMap, Map<String, ApiModuleDTO> idModuleMap) { , Map<String, String> idPathMap, Map<String, ApiModuleDTO> idModuleMap) {
//如果parentModule==null 则证明需要创建根目录同级的模块 //如果parentModule==null 则证明需要创建根目录同级的模块
ApiModule returnModule = null; ApiModule returnModule = null;
for (int i = 0; i < tagTree.length; i++) { for (int i = 0; i < tagTree.length; i++) {
int finalI = i; int finalI = i;
List<ApiModule> collect = moduleList.stream().filter(t -> t.getName().equals(tagTree[finalI])).collect(Collectors.toList()); List<ApiModule> collect = parentModuleList.stream().filter(t -> t.getName().equals(tagTree[finalI])).collect(Collectors.toList());
if (collect.isEmpty()) { if (collect.isEmpty()) {
if (parentModule == null) { if (i == 0) {
List<ApiModule> moduleList1 = pidChildrenMap.get("root"); //证明需要在根目录创建，
ApiModule apiModule = moduleList1.get(0); parentModule = new ApiModule();
apiModule.setId("root"); parentModule.setProjectId(pidChildrenMap.get("root").get(0).getProjectId());
apiModule.setLevel(0); parentModule.setId("root");
parentModule = apiModule; parentModule.setLevel(0);
} else if (i > 0) { } else {
if (!moduleList.isEmpty()) { if (!parentModuleList.isEmpty() && parentModule == null) {
String parentId = moduleList.get(0).getParentId(); String parentId = parentModuleList.get(0).getParentId();
ApiModuleDTO apiModuleDTO = idModuleMap.get(parentId); ApiModuleDTO apiModuleDTO = idModuleMap.get(parentId);
parentModule = JSON.parseObject(JSON.toJSONString(apiModuleDTO), ApiModule.class); parentModule = JSON.parseObject(JSON.toJSONString(apiModuleDTO), ApiModule.class);
} }
@ -941,7 +941,8 @@ public class ApiModuleService extends NodeTreeService<ApiModuleDTO> {
return createModule(tagTree, i, parentModule, moduleMap, pidChildrenMap, idPathMap); return createModule(tagTree, i, parentModule, moduleMap, pidChildrenMap, idPathMap);
} else { } else {
returnModule = collect.get(0); returnModule = collect.get(0);
moduleList = pidChildrenMap.get(collect.get(0).getId()); parentModule = collect.get(0);
parentModuleList = pidChildrenMap.get(collect.get(0).getId());
} }
} }
return returnModule; return returnModule;
@ -988,8 +989,9 @@ public class ApiModuleService extends NodeTreeService<ApiModuleDTO> {
private void buildProcessData(List<ApiModuleDTO> nodeTreeByProjectId, Map<String, List<ApiModule>> pidChildrenMap, Map<String, String> idPathMap, Map<String, String> parentModulePathMap) { private void buildProcessData(List<ApiModuleDTO> nodeTreeByProjectId, Map<String, List<ApiModule>> pidChildrenMap, Map<String, String> idPathMap, Map<String, String> parentModulePathMap) {
//当前层级的模块的所有子模块的集合 //当前层级的模块的所有子模块的集合
List<ApiModuleDTO> childrenList = new ArrayList<>(); Map<String, List<ApiModuleDTO>> idChildrenMap = new HashMap<>();
int i = 0; int i = 0;
Map<String, List<ApiModule>> idModuleMap = new HashMap<>();
List<ApiModule> moduleList = new ArrayList<>(); List<ApiModule> moduleList = new ArrayList<>();
for (ApiModuleDTO apiModuleDTO : nodeTreeByProjectId) { for (ApiModuleDTO apiModuleDTO : nodeTreeByProjectId) {
if (StringUtils.isBlank(apiModuleDTO.getParentId())) { if (StringUtils.isBlank(apiModuleDTO.getParentId())) {
@ -997,19 +999,24 @@ public class ApiModuleService extends NodeTreeService<ApiModuleDTO> {
} }
String parentModulePath = parentModulePathMap.get(apiModuleDTO.getParentId()); String parentModulePath = parentModulePathMap.get(apiModuleDTO.getParentId());
if (parentModulePath != null) { if (parentModulePath != null) {
apiModuleDTO.setPath(parentModulePath + "/" + apiModuleDTO.getName()); if (parentModulePath.equals("/root")) {
apiModuleDTO.setPath("/" + apiModuleDTO.getName());
} else {
apiModuleDTO.setPath(parentModulePath + "/" + apiModuleDTO.getName());
}
} else { } else {
apiModuleDTO.setPath("/" + apiModuleDTO.getName()); apiModuleDTO.setPath("/" + apiModuleDTO.getName());
} }
idPathMap.put(apiModuleDTO.getId(), apiModuleDTO.getPath()); idPathMap.put(apiModuleDTO.getId(), apiModuleDTO.getPath());
ApiModule apiModule = buildModule(moduleList, apiModuleDTO); ApiModule apiModule = buildModule(idModuleMap, apiModuleDTO);
if (pidChildrenMap.get(apiModuleDTO.getParentId()) != null) { if (pidChildrenMap.get(apiModuleDTO.getParentId()) != null) {
pidChildrenMap.get(apiModuleDTO.getParentId()).add(apiModule); pidChildrenMap.get(apiModuleDTO.getParentId()).add(apiModule);
} else { } else {
pidChildrenMap.put(apiModuleDTO.getParentId(), moduleList); pidChildrenMap.put(apiModuleDTO.getParentId(), idModuleMap.get(apiModuleDTO.getId()));
} }
i = i + 1; i = i + 1;
List<ApiModuleDTO> childrenList = idChildrenMap.get(apiModuleDTO.getId());
if (apiModuleDTO.getChildren() != null) { if (apiModuleDTO.getChildren() != null) {
childrenList.addAll(apiModuleDTO.getChildren()); childrenList.addAll(apiModuleDTO.getChildren());
} else { } else {
@ -1021,11 +1028,16 @@ public class ApiModuleService extends NodeTreeService<ApiModuleDTO> {
parentModulePathMap.put(apiModuleDTO.getId(), apiModuleDTO.getPath()); parentModulePathMap.put(apiModuleDTO.getId(), apiModuleDTO.getPath());
} }
if (i == nodeTreeByProjectId.size() && nodeTreeByProjectId.size() > 0) { if (i == nodeTreeByProjectId.size() && nodeTreeByProjectId.size() > 0) {
Collection<List<ApiModuleDTO>> values = idChildrenMap.values();
List<ApiModuleDTO> childrenList = new ArrayList<>();
for (List<ApiModuleDTO> value : values) {
childrenList.addAll(value);
}
buildProcessData(childrenList, pidChildrenMap, idPathMap, parentModulePathMap); buildProcessData(childrenList, pidChildrenMap, idPathMap, parentModulePathMap);
} }
} }
private ApiModule buildModule(List<ApiModule> moduleList, ApiModuleDTO apiModuleDTO) { private ApiModule buildModule(Map<String, List<ApiModule>> idModuleMap, ApiModuleDTO apiModuleDTO) {
ApiModule apiModule = new ApiModule(); ApiModule apiModule = new ApiModule();
apiModule.setId(apiModuleDTO.getId()); apiModule.setId(apiModuleDTO.getId());
apiModule.setName(apiModuleDTO.getName()); apiModule.setName(apiModuleDTO.getName());
@ -1033,7 +1045,14 @@ public class ApiModuleService extends NodeTreeService<ApiModuleDTO> {
apiModule.setProjectId(apiModuleDTO.getProjectId()); apiModule.setProjectId(apiModuleDTO.getProjectId());
apiModule.setProtocol(apiModuleDTO.getProtocol()); apiModule.setProtocol(apiModuleDTO.getProtocol());
apiModule.setLevel(apiModuleDTO.getLevel()); apiModule.setLevel(apiModuleDTO.getLevel());
moduleList.add(apiModule); List<ApiModule> moduleList = idModuleMap.get(apiModuleDTO.getId());
if (moduleList != null) {
moduleList.add(apiModule);
} else {
moduleList = new ArrayList<>();
moduleList.add(apiModule);
idModuleMap.put(apiModuleDTO.getId(), moduleList);
}
return apiModule; return apiModule;
} }

69
backend/src/main/java/io/metersphere/api/service/ApiScenarioModuleService.java
View File

@ -669,9 +669,8 @@ public class ApiScenarioModuleService extends NodeTreeService<ApiScenarioModuleD
datum.setApiScenarioModuleId(scenarioModule.getId()); datum.setApiScenarioModuleId(scenarioModule.getId());
datum.setModulePath(modulePath); datum.setModulePath(modulePath);
} else { } else {
//父级同级的模块list //父级同级的模块list
List<ApiScenarioModule> parentModuleList = pidChildrenMap.get("root"); ApiScenarioModule minModule = getMinModule(tagTree, pidChildrenMap.get("root"), null, pidChildrenMap, map, idPathMap, idModuleMap);
ApiScenarioModule minModule = getMinModule(tagTree, parentModuleList, null, pidChildrenMap, map, idPathMap, idModuleMap);
String id = minModule.getId(); String id = minModule.getId();
datum.setApiScenarioModuleId(id); datum.setApiScenarioModuleId(id);
datum.setModulePath(idPathMap.get(id)); datum.setModulePath(idPathMap.get(id));
@ -680,7 +679,7 @@ public class ApiScenarioModuleService extends NodeTreeService<ApiScenarioModuleD
//导入时即没选中模块，接口自身也没模块的，直接返会当前项目，当前协议下的默认模块 //导入时即没选中模块，接口自身也没模块的，直接返会当前项目，当前协议下的默认模块
List<ApiScenarioModule> moduleList = pidChildrenMap.get("root"); List<ApiScenarioModule> moduleList = pidChildrenMap.get("root");
for (ApiScenarioModule module : moduleList) { for (ApiScenarioModule module : moduleList) {
if (module.getName().equals("未规划接口")) { if (module.getName().equals("未规划场景")) {
datum.setApiScenarioModuleId(module.getId()); datum.setApiScenarioModuleId(module.getId());
datum.setModulePath("/" + module.getName()); datum.setModulePath("/" + module.getName());
} }
@ -706,30 +705,30 @@ public class ApiScenarioModuleService extends NodeTreeService<ApiScenarioModuleD
private ApiScenarioModule getMinModule(String[] tagTree, List<ApiScenarioModule> parentModuleList, ApiScenarioModule parentModule, Map<String, List<ApiScenarioModule>> pidChildrenMap, Map<String, ApiScenarioModule> map, Map<String, String> idPathMap, Map<String, ApiScenarioModuleDTO> idModuleMap) { private ApiScenarioModule getMinModule(String[] tagTree, List<ApiScenarioModule> parentModuleList, ApiScenarioModule parentModule, Map<String, List<ApiScenarioModule>> pidChildrenMap, Map<String, ApiScenarioModule> map, Map<String, String> idPathMap, Map<String, ApiScenarioModuleDTO> idModuleMap) {
//如果parentModule==null 则证明需要创建根目录同级的模块 //如果parentModule==null 则证明需要创建根目录同级的模块
ApiScenarioModule returnModule = null; ApiScenarioModule returnModule = null;
for (int i = 0; i < tagTree.length; i++) { for (int i = 0; i < tagTree.length; i++) {
int finalI = i; int finalI = i;
//查找上一级里面是否有当前全路径的第一级，没有则需要创建 //查找上一级里面是否有当前全路径的第一级，没有则需要创建
List<ApiScenarioModule> collect = parentModuleList.stream().filter(t -> t.getName().equals(tagTree[finalI])).collect(Collectors.toList()); List<ApiScenarioModule> collect = parentModuleList.stream().filter(t -> t.getName().equals(tagTree[finalI])).collect(Collectors.toList());
if (collect.isEmpty()) { if (collect.isEmpty()) {
//如果找不到，而且父级模块也为空，证明当前的父级模块应为root if (i == 0) {
if (parentModule == null) { //证明需要在根目录创建，
List<ApiScenarioModule> moduleList1 = pidChildrenMap.get("root"); parentModule = new ApiScenarioModule();
ApiScenarioModule apiModule = moduleList1.get(0); parentModule.setProjectId(pidChildrenMap.get("root").get(0).getProjectId());
apiModule.setId("root"); parentModule.setId("root");
apiModule.setLevel(0); parentModule.setLevel(0);
parentModule = apiModule; } else {
} else if (i > 0) { if (!parentModuleList.isEmpty() && parentModule == null) {
//如果已经循环第二次级以上，如果父模块不为空， parentModuleList不为空，tagTree[finalI] 找不到，那就从上一级开始创建
if (!parentModuleList.isEmpty()) {
String parentId = parentModuleList.get(0).getParentId(); String parentId = parentModuleList.get(0).getParentId();
ApiScenarioModuleDTO apiScenarioModuleDTO = idModuleMap.get(parentId); ApiScenarioModuleDTO apiScenarioModuleDTO = idModuleMap.get(parentId);
parentModule = JSON.parseObject(JSON.toJSONString(apiScenarioModuleDTO), ApiScenarioModule.class); parentModule = JSON.parseObject(JSON.toJSONString(apiScenarioModuleDTO), ApiScenarioModule.class);
} }
} }
//开始创建tagTree[finalI]这个模块
return createModule(tagTree, i, parentModule, map, pidChildrenMap, idPathMap); return createModule(tagTree, i, parentModule, map, pidChildrenMap, idPathMap);
} else { } else {
returnModule = collect.get(0); returnModule = collect.get(0);
parentModule = collect.get(0);
parentModuleList = pidChildrenMap.get(collect.get(0).getId()); parentModuleList = pidChildrenMap.get(collect.get(0).getId());
} }
} }
@ -779,51 +778,71 @@ public class ApiScenarioModuleService extends NodeTreeService<ApiScenarioModuleD
} }
private void buildProcessData(List<ApiScenarioModuleDTO> nodeTreeByProjectId, Map<String, List<ApiScenarioModule>> pidChildrenMap, Map<String, String> idPathMap, Map<String, String> parentModulePathMap) { private void buildProcessData(List<ApiScenarioModuleDTO> nodeTreeByProjectId, Map<String, List<ApiScenarioModule>> pidChildrenMap, Map<String, String> idPathMap, Map<String, String> parentModulePathMap) {
List<ApiScenarioModuleDTO> childrenList = new ArrayList<>(); Map<String, List<ApiScenarioModuleDTO>> idChildrenMap = new HashMap<>();
int i = 0; int i = 0;
List<ApiScenarioModule> moduleList = new ArrayList<>(); Map<String, List<ApiScenarioModule>> idModuleMap = new HashMap<>();
for (ApiScenarioModuleDTO scenarioModuleDTO : nodeTreeByProjectId) { for (ApiScenarioModuleDTO scenarioModuleDTO : nodeTreeByProjectId) {
if (StringUtils.isBlank(scenarioModuleDTO.getParentId())) { if (StringUtils.isBlank(scenarioModuleDTO.getParentId())) {
scenarioModuleDTO.setParentId("root"); scenarioModuleDTO.setParentId("root");
} }
String parentModulePath = parentModulePathMap.get(scenarioModuleDTO.getParentId()); String parentModulePath = parentModulePathMap.get(scenarioModuleDTO.getParentId());
if (parentModulePath != null) { if (parentModulePath != null) {
scenarioModuleDTO.setPath(parentModulePath + "/" + scenarioModuleDTO.getName()); if (parentModulePath.equals("/root")) {
scenarioModuleDTO.setPath("/" + scenarioModuleDTO.getName());
} else {
scenarioModuleDTO.setPath(parentModulePath + "/" + scenarioModuleDTO.getName());
}
} else { } else {
scenarioModuleDTO.setPath("/" + scenarioModuleDTO.getName()); scenarioModuleDTO.setPath("/" + scenarioModuleDTO.getName());
} }
idPathMap.put(scenarioModuleDTO.getId(), scenarioModuleDTO.getPath()); idPathMap.put(scenarioModuleDTO.getId(), scenarioModuleDTO.getPath());
ApiScenarioModule scenarioModule = buildModule(moduleList, scenarioModuleDTO); ApiScenarioModule scenarioModule = buildModule(idModuleMap, scenarioModuleDTO);
if (pidChildrenMap.get(scenarioModuleDTO.getParentId()) != null) { if (pidChildrenMap.get(scenarioModuleDTO.getParentId()) != null) {
pidChildrenMap.get(scenarioModuleDTO.getParentId()).add(scenarioModule); pidChildrenMap.get(scenarioModuleDTO.getParentId()).add(scenarioModule);
} else { } else {
pidChildrenMap.put(scenarioModuleDTO.getParentId(), moduleList); pidChildrenMap.put(scenarioModuleDTO.getParentId(), idModuleMap.get(scenarioModuleDTO.getId()));
} }
i = i + 1; i = i + 1;
List<ApiScenarioModuleDTO> childrenList = idChildrenMap.get(scenarioModuleDTO.getId());
if (scenarioModuleDTO.getChildren() != null) { if (scenarioModuleDTO.getChildren() != null) {
childrenList.addAll(scenarioModuleDTO.getChildren()); if (childrenList != null) {
childrenList.addAll(scenarioModuleDTO.getChildren());
} else {
idChildrenMap.put(scenarioModuleDTO.getId(), scenarioModuleDTO.getChildren());
}
} else { } else {
childrenList.addAll(new ArrayList<>()); if (i == nodeTreeByProjectId.size() && idChildrenMap.size() == 0) {
if (i == nodeTreeByProjectId.size() && childrenList.size() == 0) {
pidChildrenMap.put(scenarioModuleDTO.getId(), new ArrayList<>()); pidChildrenMap.put(scenarioModuleDTO.getId(), new ArrayList<>());
} }
} }
parentModulePathMap.put(scenarioModuleDTO.getId(), scenarioModuleDTO.getPath()); parentModulePathMap.put(scenarioModuleDTO.getId(), scenarioModuleDTO.getPath());
} }
if (i == nodeTreeByProjectId.size() && nodeTreeByProjectId.size() > 0) { if (i == nodeTreeByProjectId.size() && nodeTreeByProjectId.size() > 0) {
Collection<List<ApiScenarioModuleDTO>> values = idChildrenMap.values();
List<ApiScenarioModuleDTO> childrenList = new ArrayList<>();
for (List<ApiScenarioModuleDTO> value : values) {
childrenList.addAll(value);
}
buildProcessData(childrenList, pidChildrenMap, idPathMap, parentModulePathMap); buildProcessData(childrenList, pidChildrenMap, idPathMap, parentModulePathMap);
} }
} }
private ApiScenarioModule buildModule(List<ApiScenarioModule> moduleList, ApiScenarioModuleDTO scenarioModuleDTO) { private ApiScenarioModule buildModule(Map<String, List<ApiScenarioModule>> IdModuleMap, ApiScenarioModuleDTO scenarioModuleDTO) {
ApiScenarioModule scenarioModule = new ApiScenarioModule(); ApiScenarioModule scenarioModule = new ApiScenarioModule();
scenarioModule.setId(scenarioModuleDTO.getId()); scenarioModule.setId(scenarioModuleDTO.getId());
scenarioModule.setName(scenarioModuleDTO.getName()); scenarioModule.setName(scenarioModuleDTO.getName());
scenarioModule.setParentId(scenarioModuleDTO.getParentId()); scenarioModule.setParentId(scenarioModuleDTO.getParentId());
scenarioModule.setProjectId(scenarioModuleDTO.getProjectId()); scenarioModule.setProjectId(scenarioModuleDTO.getProjectId());
scenarioModule.setLevel(scenarioModuleDTO.getLevel()); scenarioModule.setLevel(scenarioModuleDTO.getLevel());
moduleList.add(scenarioModule); List<ApiScenarioModule> moduleList = IdModuleMap.get(scenarioModuleDTO.getId());
if (moduleList != null) {
moduleList.add(scenarioModule);
} else {
moduleList = new ArrayList<>();
moduleList.add(scenarioModule);
IdModuleMap.put(scenarioModuleDTO.getId(), moduleList);
}
return scenarioModule; return scenarioModule;
} }
} }