test/MeterSphere
1
0
Fork 0
mirror of https://gitee.com/fit2cloud-feizhiyun/MeterSphere.git synced 2024-12-02 20:19:16 +08:00
Code Issues Actions 18 Packages Projects Releases Wiki Activity

fix(接口测试): 接口自动化导入导出

Browse Source 
--user=郭雨琦
接口自动化导入导出，新增数据是根据当前用户权限下的项目里找不到才新增
This commit is contained in:
guoyuqi 2022-03-22 16:41:26 +08:00 committed by xiaomeinvG
parent 6ade39a431
commit 654e59d89b
3 changed files with 45 additions and 52 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

68
backend/src/main/java/io/metersphere/api/dto/automation/parse/ApiScenarioImportUtil.java
View File

@ -1,12 +1,9 @@
package io.metersphere.api.dto.automation.parse;
import com.alibaba.fastjson.JSON;
import com.alibaba.fastjson.JSONArray;
import com.alibaba.fastjson.JSONObject;
import io.metersphere.api.dto.automation.ApiScenarioModuleDTO;
import io.metersphere.api.dto.definition.ApiDefinitionRequest;
import io.metersphere.api.dto.definition.ApiDefinitionResult;
import io.metersphere.api.dto.definition.SaveApiTestCaseRequest;
import io.metersphere.api.dto.definition.parse.ms.NodeTree;
import io.metersphere.api.service.ApiDefinitionService;
import io.metersphere.api.service.ApiScenarioModuleService;
@ -18,7 +15,7 @@ import io.metersphere.commons.constants.APITestStatus;
import io.metersphere.commons.utils.BeanUtils;
import io.metersphere.commons.utils.CommonBeanFactory;
import io.metersphere.commons.utils.SessionUtils;
import io.metersphere.service.ProjectService;
import io.metersphere.service.CheckPermissionService;
import org.apache.commons.collections.CollectionUtils;
import org.apache.commons.lang3.StringUtils;
@ -118,63 +115,51 @@ public class ApiScenarioImportUtil {
}
}
public static boolean checkWorkSpace(String projectId, String currentProjectId) {
if(!Objects.equals(projectId, currentProjectId)){
ProjectService projectService = CommonBeanFactory.getBean(ProjectService.class);
Project project = projectService.getProjectById(projectId);
return Objects.equals(project.getWorkspaceId(), SessionUtils.getCurrentWorkspaceId());
}
return true;
private static ApiDefinition getApiDefinitionResult(JSONObject object, ApiDefinitionService apiDefinitionService) {
CheckPermissionService checkPermissionService = CommonBeanFactory.getBean(CheckPermissionService.class);
Set<String> userRelatedProjectIds = checkPermissionService.getUserRelatedProjectIds();
List<String> projectIds = new ArrayList<>(userRelatedProjectIds);
ApiDefinitionExample apiDefinitionExample = new ApiDefinitionExample();
apiDefinitionExample.createCriteria()
.andPathEqualTo(object.getString("path"))
.andMethodEqualTo(object.getString("method"))
.andProtocolEqualTo(object.getString("protocol"))
.andProjectIdIn(projectIds)
.andStatusNotEqualTo("Trash");
return apiDefinitionService.getApiDefinition(apiDefinitionExample);
}
private static ApiDefinitionResult getApiDefinitionResult(JSONObject object, ApiDefinitionService apiDefinitionService) {
ApiDefinitionRequest apiDefinitionRequest = new ApiDefinitionRequest();
apiDefinitionRequest.setPath(object.getString("path"));
apiDefinitionRequest.setMethod(object.getString("method"));
apiDefinitionRequest.setPath(object.getString("protocol"));
return apiDefinitionService.getApiDefinitionResult(apiDefinitionRequest);
}
private static ApiTestCaseWithBLOBs getApiTestCase(JSONObject object, ApiTestCaseService testCaseService, ApiDefinitionResult apiDefinitionResult) {
SaveApiTestCaseRequest request = new SaveApiTestCaseRequest();
request.setName(object.getString("name"));
request.setApiDefinitionId(apiDefinitionResult.getId());
return testCaseService.getSameCaseWithBLOBs(request);
}
public static void checkCase(JSONObject object, String versionId, String projectId, ApiTestCaseMapper apiTestCaseMapper, ApiDefinitionMapper apiDefinitionMapper) {
public static void checkCase(int i,JSONObject object, String versionId, String projectId, ApiTestCaseMapper apiTestCaseMapper, ApiDefinitionMapper apiDefinitionMapper) {
ApiTestCaseService testCaseService = CommonBeanFactory.getBean(ApiTestCaseService.class);
ApiDefinitionService apiDefinitionService = CommonBeanFactory.getBean(ApiDefinitionService.class);
ApiTestCaseWithBLOBs bloBs = testCaseService.get(object.getString("id"));
if (bloBs == null) {
ApiDefinitionResult apiDefinition = getApiDefinitionResult(object,apiDefinitionService);
ApiTestCaseWithBLOBs testCase;
ApiDefinition apiDefinition = getApiDefinitionResult(object,apiDefinitionService);
if(apiDefinition!=null){
testCase= getApiTestCase(object, testCaseService, apiDefinition);
if (testCase != null) {
object.put("id", testCase.getId());
structureCaseByJson(i,object,testCaseService, apiDefinition,apiTestCaseMapper);
}else{
structureCaseByJson(object, apiDefinition,apiTestCaseMapper);
}
}else{
ApiDefinitionResult apiDefinitionResult = structureApiDefinitionByJson(apiDefinitionService, object, versionId, projectId, apiDefinitionMapper);
structureCaseByJson(object, apiDefinitionResult,apiTestCaseMapper);
ApiDefinitionResult apiDefinitionResult = structureApiDefinitionByJson(i,apiDefinitionService, object, versionId, projectId, apiDefinitionMapper);
structureCaseByJson(i,object,testCaseService, apiDefinitionResult,apiTestCaseMapper);
}
}
}
public static ApiDefinitionResult structureApiDefinitionByJson(ApiDefinitionService apiDefinitionService,JSONObject object, String versionId, String projectId,ApiDefinitionMapper apiDefinitionMapper) {
public static ApiDefinitionResult structureApiDefinitionByJson(int i,ApiDefinitionService apiDefinitionService,JSONObject object, String versionId, String projectId,ApiDefinitionMapper apiDefinitionMapper) {
ApiDefinitionResult test = new ApiDefinitionResult();
apiDefinitionService.checkQuota();
String protocal = object.getString("protocal");
if (StringUtils.equals(protocal, "DUBBO")) {
test.setMethod("dubbo://");
}else{
test.setMethod(protocal);
test.setMethod(object.getString("method"));
}
apiDefinitionService.initModulePathAndId(projectId, test);
test.setProtocol(object.getString("protocol"));
test.setModulePath(object.getString("modulePath"));
test.setModuleId(object.getString("moduleId"));
String id = UUID.randomUUID().toString();
test.setId(id);
test.setNum(apiDefinitionService.getNextNum(projectId)+i);
test.setName(object.getString("name"));
test.setPath(object.getString("path"));
test.setCreateUser(SessionUtils.getUserId());
@ -197,7 +182,7 @@ public class ApiScenarioImportUtil {
return test;
}
public static void structureCaseByJson(JSONObject object, ApiDefinitionResult apiDefinition, ApiTestCaseMapper apiTestCaseMapper) {
public static void structureCaseByJson(int i,JSONObject object,ApiTestCaseService testCaseService, ApiDefinition apiDefinition, ApiTestCaseMapper apiTestCaseMapper) {
String projectId = apiDefinition.getProjectId();
ApiDefinitionService apiDefinitionService = CommonBeanFactory.getBean(ApiDefinitionService.class);
ApiTestCaseWithBLOBs apiTestCase = new ApiTestCaseWithBLOBs();
@ -213,6 +198,7 @@ public class ApiScenarioImportUtil {
apiTestCase.setUpdateTime(System.currentTimeMillis());
apiTestCase.setVersionId(apiDefinition.getVersionId());
apiTestCase.setPriority("P0");
apiTestCase.setNum(testCaseService.getNextNum(apiTestCase.getApiDefinitionId(), apiDefinition.getNum()+i));
object.put("id", apiTestCase.getId());
object.put("resourceId", apiTestCase.getId());
object.put("projectId", projectId);

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

@ -404,7 +404,7 @@ public class ApiAutomationService {
JSONObject element = JSON.parseObject(scenario.getScenarioDefinition());
JSONArray hashTree = element.getJSONArray("hashTree");
ApiScenarioImportUtil.formatHashTree(hashTree);
setReferenced(hashTree,scenario.getVersionId(),scenario.getProjectId(),apiTestCaseMapper,apiDefinitionMapper);
setReferenced(hashTree,scenario.getVersionId(),scenario.getProjectId(),apiTestCaseMapper,apiDefinitionMapper,true);
scenario.setScenarioDefinition(JSONObject.toJSONString(element));
}
}
@ -1968,7 +1968,7 @@ public class ApiAutomationService {
return strings;
}
private void setReferenced(JSONArray hashTree,String versionId,String projectId, ApiTestCaseMapper apiTestCaseMapper,ApiDefinitionMapper apiDefinitionMapper) {
private void setReferenced(JSONArray hashTree,String versionId,String projectId, ApiTestCaseMapper apiTestCaseMapper,ApiDefinitionMapper apiDefinitionMapper,boolean isAdd) {
// 将引用转成复制
if (CollectionUtils.isNotEmpty(hashTree)) {
for (int i = 0; i < hashTree.size(); i++) {
@ -1978,8 +1978,8 @@ public class ApiAutomationService {
// 检测引用对象是否存在，若果不存在则改成复制对象
String refType = object.getString("refType");
if (StringUtils.isNotEmpty(refType)) {
if (refType.equals("CASE")) {
ApiScenarioImportUtil.checkCase(object,versionId,projectId,apiTestCaseMapper,apiDefinitionMapper);
if (refType.equals("CASE")&&isAdd) {
ApiScenarioImportUtil.checkCase(i,object,versionId,projectId,apiTestCaseMapper,apiDefinitionMapper);
} else {
checkAutomation(object);
}
@ -1991,9 +1991,16 @@ public class ApiAutomationService {
if (environmentMap != null) {
object.put("environmentMap", new HashMap<>());
}
if(StringUtils.isNotEmpty(object.getString("refType"))&&object.getString("refType").equals("CASE")){
if (CollectionUtils.isNotEmpty(object.getJSONArray("hashTree"))) {
setReferenced(object.getJSONArray("hashTree"),versionId,projectId,apiTestCaseMapper,apiDefinitionMapper);
setReferenced(object.getJSONArray("hashTree"),versionId,projectId,apiTestCaseMapper,apiDefinitionMapper,true);
}
}else {
if (CollectionUtils.isNotEmpty(object.getJSONArray("hashTree"))) {
setReferenced(object.getJSONArray("hashTree"),versionId,projectId,apiTestCaseMapper,apiDefinitionMapper,false);
}
}
}
}
}

10
backend/src/main/java/io/metersphere/api/service/ApiDefinitionService.java
View File

@ -721,7 +721,7 @@ public class ApiDefinitionService {
return test;
}
private int getNextNum(String projectId) {
public int getNextNum(String projectId) {
ApiDefinition apiDefinition = extApiDefinitionMapper.getNextNum(projectId);
if (apiDefinition == null || apiDefinition.getNum() == null) {
return 100001;
@ -2021,12 +2021,12 @@ public class ApiDefinitionService {
}
}
public ApiDefinitionResult getApiDefinitionResult(ApiDefinitionRequest request) {
List<ApiDefinitionResult> resList = extApiDefinitionMapper.list(request);
if(resList==null){
public ApiDefinition getApiDefinition(ApiDefinitionExample apiDefinitionExample) {
List<ApiDefinition> apiDefinitions = apiDefinitionMapper.selectByExample(apiDefinitionExample);
if(apiDefinitions==null||apiDefinitions.size()==0){
return null;
}
return resList.get(0);
return apiDefinitions.get(0);
}
public void initModulePathAndId(String projectId, ApiDefinitionWithBLOBs test) {