From a5065b7cb4db7cea35a98957911cdeb4cd547b1c Mon Sep 17 00:00:00 2001
From: WangXu10 <xu.wang@fit2cloud.com>
Date: Fri, 7 Jun 2024 10:01:13 +0800
Subject: [PATCH] =?UTF-8?q?feat(=E6=B5=8B=E8=AF=95=E8=AE=A1=E5=88=92):=20?=
 =?UTF-8?q?=E5=A4=84=E7=90=86=E5=85=B3=E8=81=94api&=E5=85=B3=E8=81=94apiCa?=
 =?UTF-8?q?se?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 .../plan/constants/AssociateCaseType.java     |  6 +-
 .../BaseCollectionAssociateRequest.java       | 10 +--
 .../plan/mapper/ExtTestPlanApiCaseMapper.xml  |  2 +-
 .../plan/service/TestPlanApiCaseService.java  | 77 +++++++++++++++++--
 .../service/TestPlanApiScenarioService.java   | 11 +--
 .../plan/service/TestPlanBugService.java      |  2 +-
 .../TestPlanFunctionalCaseService.java        | 11 +--
 .../plan/service/TestPlanResourceService.java |  5 +-
 .../TestPlanApiCaseControllerTests.java       | 33 ++++++++
 9 files changed, 125 insertions(+), 32 deletions(-)

diff --git a/backend/services/test-plan/src/main/java/io/metersphere/plan/constants/AssociateCaseType.java b/backend/services/test-plan/src/main/java/io/metersphere/plan/constants/AssociateCaseType.java
index 7e8a185eb4..7f2b3f6c19 100644
--- a/backend/services/test-plan/src/main/java/io/metersphere/plan/constants/AssociateCaseType.java
+++ b/backend/services/test-plan/src/main/java/io/metersphere/plan/constants/AssociateCaseType.java
@@ -2,9 +2,13 @@ package io.metersphere.plan.constants;
 
 public class AssociateCaseType {
     /**
-     * 接口用例
+     * 接口
      */
     public static final String API = "API";
+    /**
+     * 接口用例
+     */
+    public static final String API_CASE = "API_CASE";
     /**
      * 场景用例
      */
diff --git a/backend/services/test-plan/src/main/java/io/metersphere/plan/dto/request/BaseCollectionAssociateRequest.java b/backend/services/test-plan/src/main/java/io/metersphere/plan/dto/request/BaseCollectionAssociateRequest.java
index 50e5ee7fa9..c3175fddc1 100644
--- a/backend/services/test-plan/src/main/java/io/metersphere/plan/dto/request/BaseCollectionAssociateRequest.java
+++ b/backend/services/test-plan/src/main/java/io/metersphere/plan/dto/request/BaseCollectionAssociateRequest.java
@@ -8,12 +8,10 @@ import java.util.List;
 @Data
 public class BaseCollectionAssociateRequest {
 
-	@Schema(description = "测试集ID")
-	private String collectionId;
+    @Schema(description = "测试集ID")
+    private String collectionId;
 
-	@Schema(description = "关联的用例ID集合")
-	private List<String> ids;
+    @Schema(description = "关联的用例ID集合")
+    private List<String> ids;
 
-	@Schema(description = "测试集类型")
-	private String type;
 }
diff --git a/backend/services/test-plan/src/main/java/io/metersphere/plan/mapper/ExtTestPlanApiCaseMapper.xml b/backend/services/test-plan/src/main/java/io/metersphere/plan/mapper/ExtTestPlanApiCaseMapper.xml
index 9ecd35d1eb..9f1cd3d948 100644
--- a/backend/services/test-plan/src/main/java/io/metersphere/plan/mapper/ExtTestPlanApiCaseMapper.xml
+++ b/backend/services/test-plan/src/main/java/io/metersphere/plan/mapper/ExtTestPlanApiCaseMapper.xml
@@ -21,7 +21,7 @@
     <select id="getMaxPosByTestPlanId" resultType="java.lang.Long">
         SELECT max(pos)
         FROM test_plan_api_case
-        WHERE test_plan_id = #{0}
+        WHERE test_plan_collection_id = #{0}
     </select>
 
     <select id="getIdByParam"
diff --git a/backend/services/test-plan/src/main/java/io/metersphere/plan/service/TestPlanApiCaseService.java b/backend/services/test-plan/src/main/java/io/metersphere/plan/service/TestPlanApiCaseService.java
index cd7a224616..edb586631d 100644
--- a/backend/services/test-plan/src/main/java/io/metersphere/plan/service/TestPlanApiCaseService.java
+++ b/backend/services/test-plan/src/main/java/io/metersphere/plan/service/TestPlanApiCaseService.java
@@ -1,12 +1,16 @@
 package io.metersphere.plan.service;
 
+import io.metersphere.api.domain.ApiTestCase;
+import io.metersphere.api.domain.ApiTestCaseExample;
 import io.metersphere.api.dto.definition.ApiDefinitionDTO;
 import io.metersphere.api.dto.definition.ApiTestCaseDTO;
+import io.metersphere.api.mapper.ApiTestCaseMapper;
 import io.metersphere.api.service.definition.ApiDefinitionModuleService;
 import io.metersphere.api.service.definition.ApiDefinitionService;
 import io.metersphere.api.service.definition.ApiTestCaseService;
 import io.metersphere.functional.dto.FunctionalCaseModuleCountDTO;
 import io.metersphere.functional.dto.ProjectOptionDTO;
+import io.metersphere.plan.constants.AssociateCaseType;
 import io.metersphere.plan.constants.TreeTypeEnums;
 import io.metersphere.plan.domain.TestPlanApiCase;
 import io.metersphere.plan.domain.TestPlanApiCaseExample;
@@ -79,6 +83,8 @@ public class TestPlanApiCaseService extends TestPlanResourceService {
     private SqlSessionFactory sqlSessionFactory;
     @Resource
     private TestPlanCollectionMapper testPlanCollectionMapper;
+    @Resource
+    private ApiTestCaseMapper apiTestCaseMapper;
 
     @Override
     public void deleteBatchByTestPlanId(List<String> testPlanIdList) {
@@ -88,8 +94,13 @@ public class TestPlanApiCaseService extends TestPlanResourceService {
     }
 
     @Override
-    public long getNextOrder(String projectId) {
-        return 0;
+    public long getNextOrder(String collectionId) {
+        Long maxPos = extTestPlanApiCaseMapper.getMaxPosByTestPlanId(collectionId);
+        if (maxPos == null) {
+            return 0;
+        } else {
+            return maxPos + DEFAULT_NODE_INTERVAL_POS;
+        }
     }
 
     @Override
@@ -441,12 +452,64 @@ public class TestPlanApiCaseService extends TestPlanResourceService {
     }
 
     @Override
-    public void associateCollection(String planId, List<BaseCollectionAssociateRequest> collectionAssociates) {
-        List<BaseCollectionAssociateRequest> apiAssociates = collectionAssociates.stream().filter(associate -> StringUtils.equals(associate.getType(), CaseType.API_CASE.getKey())).toList();
-        if (CollectionUtils.isNotEmpty(apiAssociates)) {
-            apiAssociates.forEach(apiAssociate -> {
-                // TODO: 调用具体的关联功能用例入库方法  入参{计划ID, 测试集ID, 关联的用例ID集合}
+    public void associateCollection(String planId, Map<String, List<BaseCollectionAssociateRequest>> collectionAssociates, String userId) {
+        List<TestPlanApiCase> testPlanApiCaseList = new ArrayList<>();
+        //处理数据
+        handleApiData(collectionAssociates.get(AssociateCaseType.API), userId, testPlanApiCaseList, planId);
+        handleApiCaseData(collectionAssociates.get(AssociateCaseType.API_CASE), userId, testPlanApiCaseList, planId);
+        testPlanApiCaseMapper.batchInsert(testPlanApiCaseList);
+    }
+
+    private void handleApiCaseData(List<BaseCollectionAssociateRequest> apiCaseList, String userId, List<TestPlanApiCase> testPlanApiCaseList, String planId) {
+        if (CollectionUtils.isNotEmpty(apiCaseList)) {
+            List<String> ids = apiCaseList.stream().flatMap(item -> item.getIds().stream()).toList();
+            ApiTestCaseExample example = new ApiTestCaseExample();
+            example.createCriteria().andIdIn(ids);
+            List<ApiTestCase> apiTestCaseList = apiTestCaseMapper.selectByExample(example);
+            apiCaseList.forEach(apiCase -> {
+                List<String> apiCaseIds = apiCase.getIds();
+                List<ApiTestCase> apiTestCases = apiTestCaseList.stream().filter(item -> apiCaseIds.contains(item.getId())).collect(Collectors.toList());
+                buildTestPlanApiCase(planId, apiTestCases, apiCase.getCollectionId(), userId, testPlanApiCaseList);
             });
         }
     }
+
+    private void handleApiData(List<BaseCollectionAssociateRequest> apiCaseList, String userId, List<TestPlanApiCase> testPlanApiCaseList, String planId) {
+        if (CollectionUtils.isNotEmpty(apiCaseList)) {
+            List<String> ids = apiCaseList.stream().flatMap(item -> item.getIds().stream()).toList();
+            ApiTestCaseExample example = new ApiTestCaseExample();
+            example.createCriteria().andApiDefinitionIdIn(ids);
+            List<ApiTestCase> apiTestCaseList = apiTestCaseMapper.selectByExample(example);
+            apiCaseList.forEach(apiCase -> {
+                List<String> apiCaseIds = apiCase.getIds();
+                List<ApiTestCase> apiTestCases = apiTestCaseList.stream().filter(item -> apiCaseIds.contains(item.getApiDefinitionId())).collect(Collectors.toList());
+                buildTestPlanApiCase(planId, apiTestCases, apiCase.getCollectionId(), userId, testPlanApiCaseList);
+            });
+        }
+    }
+
+
+    /**
+     * 构建测试计划接口用例对象
+     *
+     * @param planId
+     * @param apiTestCases
+     * @param collectionId
+     * @param userId
+     * @param testPlanApiCaseList
+     */
+    private void buildTestPlanApiCase(String planId, List<ApiTestCase> apiTestCases, String collectionId, String userId, List<TestPlanApiCase> testPlanApiCaseList) {
+        apiTestCases.forEach(apiTestCase -> {
+            TestPlanApiCase testPlanApiCase = new TestPlanApiCase();
+            testPlanApiCase.setId(IDGenerator.nextStr());
+            testPlanApiCase.setTestPlanCollectionId(collectionId);
+            testPlanApiCase.setTestPlanId(planId);
+            testPlanApiCase.setApiCaseId(apiTestCase.getId());
+            testPlanApiCase.setEnvironmentId(apiTestCase.getEnvironmentId());
+            testPlanApiCase.setCreateTime(System.currentTimeMillis());
+            testPlanApiCase.setCreateUser(userId);
+            testPlanApiCase.setPos(getNextOrder(collectionId));
+            testPlanApiCaseList.add(testPlanApiCase);
+        });
+    }
 }
diff --git a/backend/services/test-plan/src/main/java/io/metersphere/plan/service/TestPlanApiScenarioService.java b/backend/services/test-plan/src/main/java/io/metersphere/plan/service/TestPlanApiScenarioService.java
index bfaa49479a..4305aca3ce 100644
--- a/backend/services/test-plan/src/main/java/io/metersphere/plan/service/TestPlanApiScenarioService.java
+++ b/backend/services/test-plan/src/main/java/io/metersphere/plan/service/TestPlanApiScenarioService.java
@@ -1,5 +1,6 @@
 package io.metersphere.plan.service;
 
+import io.metersphere.plan.constants.AssociateCaseType;
 import io.metersphere.plan.domain.TestPlanApiScenario;
 import io.metersphere.plan.domain.TestPlanApiScenarioExample;
 import io.metersphere.plan.dto.TestPlanCaseRunResultCount;
@@ -96,13 +97,9 @@ public class TestPlanApiScenarioService extends TestPlanResourceService {
     }
 
     @Override
-    public void associateCollection(String planId, List<BaseCollectionAssociateRequest> collectionAssociates) {
-        List<BaseCollectionAssociateRequest> scenarioAssociates = collectionAssociates.stream().filter(associate -> StringUtils.equals(associate.getType(), CaseType.SCENARIO_CASE.getKey())).toList();
-        if (CollectionUtils.isNotEmpty(scenarioAssociates)) {
-            scenarioAssociates.forEach(scenarioAssociate -> {
-                // TODO: 调用具体的关联场景用例入库方法  入参{计划ID, 测试集ID, 关联的用例ID集合}
-            });
-        }
+    public void associateCollection(String planId, Map<String, List<BaseCollectionAssociateRequest>> collectionAssociates,String userId) {
+        List<BaseCollectionAssociateRequest> apiScenarios = collectionAssociates.get(AssociateCaseType.API_SCENARIO);
+        // TODO: 调用具体的关联场景用例入库方法  入参{计划ID, 测试集ID, 关联的用例ID集合}
     }
 
 }
diff --git a/backend/services/test-plan/src/main/java/io/metersphere/plan/service/TestPlanBugService.java b/backend/services/test-plan/src/main/java/io/metersphere/plan/service/TestPlanBugService.java
index aa8a2a2921..4f86c2da9a 100644
--- a/backend/services/test-plan/src/main/java/io/metersphere/plan/service/TestPlanBugService.java
+++ b/backend/services/test-plan/src/main/java/io/metersphere/plan/service/TestPlanBugService.java
@@ -118,7 +118,7 @@ public class TestPlanBugService extends TestPlanResourceService {
     }
 
     @Override
-    public void associateCollection(String planId, List<BaseCollectionAssociateRequest> collectionAssociates) {
+    public void associateCollection(String planId, Map<String, List<BaseCollectionAssociateRequest>> collectionAssociates,String userId) {
         // TODO: 暂不支持缺陷关联测试集
     }
 }
diff --git a/backend/services/test-plan/src/main/java/io/metersphere/plan/service/TestPlanFunctionalCaseService.java b/backend/services/test-plan/src/main/java/io/metersphere/plan/service/TestPlanFunctionalCaseService.java
index bb6ba4774d..8690618b2e 100644
--- a/backend/services/test-plan/src/main/java/io/metersphere/plan/service/TestPlanFunctionalCaseService.java
+++ b/backend/services/test-plan/src/main/java/io/metersphere/plan/service/TestPlanFunctionalCaseService.java
@@ -17,6 +17,7 @@ import io.metersphere.functional.mapper.FunctionalCaseMapper;
 import io.metersphere.functional.service.FunctionalCaseAttachmentService;
 import io.metersphere.functional.service.FunctionalCaseModuleService;
 import io.metersphere.functional.service.FunctionalCaseService;
+import io.metersphere.plan.constants.AssociateCaseType;
 import io.metersphere.plan.domain.*;
 import io.metersphere.plan.dto.ResourceLogInsertModule;
 import io.metersphere.plan.dto.TestPlanCaseRunResultCount;
@@ -648,12 +649,8 @@ public class TestPlanFunctionalCaseService extends TestPlanResourceService {
     }
 
     @Override
-    public void associateCollection(String planId, List<BaseCollectionAssociateRequest> collectionAssociates) {
-        List<BaseCollectionAssociateRequest> functionalAssociates = collectionAssociates.stream().filter(associate -> StringUtils.equals(associate.getType(), CaseType.FUNCTIONAL_CASE.getKey())).toList();
-        if (CollectionUtils.isNotEmpty(functionalAssociates)) {
-            functionalAssociates.forEach(functionalAssociate -> {
-                // TODO: 调用具体的关联接口用例入库方法  入参{计划ID, 测试集ID, 关联的用例ID集合}
-            });
-        }
+    public void associateCollection(String planId, Map<String, List<BaseCollectionAssociateRequest>> collectionAssociates,String userId) {
+        List<BaseCollectionAssociateRequest> functionals = collectionAssociates.get(AssociateCaseType.FUNCTIONAL);
+        // TODO: 调用具体的关联接口用例入库方法  入参{计划ID, 测试集ID, 关联的用例ID集合}
     }
 }
diff --git a/backend/services/test-plan/src/main/java/io/metersphere/plan/service/TestPlanResourceService.java b/backend/services/test-plan/src/main/java/io/metersphere/plan/service/TestPlanResourceService.java
index cd8fa28cd5..834df161e5 100644
--- a/backend/services/test-plan/src/main/java/io/metersphere/plan/service/TestPlanResourceService.java
+++ b/backend/services/test-plan/src/main/java/io/metersphere/plan/service/TestPlanResourceService.java
@@ -57,8 +57,9 @@ public abstract class TestPlanResourceService extends TestPlanSortService {
 
     /**
      * 关联用例
-     * @param planId 计划ID
+     *
+     * @param planId               计划ID
      * @param collectionAssociates 测试集关联用例参数
      */
-    public abstract void associateCollection(String planId, List<BaseCollectionAssociateRequest> collectionAssociates);
+    public abstract void associateCollection(String planId, Map<String, List<BaseCollectionAssociateRequest>> collectionAssociates, String userId);
 }
diff --git a/backend/services/test-plan/src/test/java/io/metersphere/plan/controller/TestPlanApiCaseControllerTests.java b/backend/services/test-plan/src/test/java/io/metersphere/plan/controller/TestPlanApiCaseControllerTests.java
index 3e5b3a8c82..2b759f7f1f 100644
--- a/backend/services/test-plan/src/test/java/io/metersphere/plan/controller/TestPlanApiCaseControllerTests.java
+++ b/backend/services/test-plan/src/test/java/io/metersphere/plan/controller/TestPlanApiCaseControllerTests.java
@@ -1,9 +1,12 @@
 package io.metersphere.plan.controller;
 
+import io.metersphere.plan.constants.AssociateCaseType;
 import io.metersphere.plan.dto.request.*;
+import io.metersphere.plan.service.TestPlanApiCaseService;
 import io.metersphere.sdk.util.JSON;
 import io.metersphere.system.base.BaseTest;
 import io.metersphere.system.controller.handler.ResultHolder;
+import jakarta.annotation.Resource;
 import org.junit.jupiter.api.*;
 import org.springframework.boot.test.autoconfigure.web.servlet.AutoConfigureMockMvc;
 import org.springframework.boot.test.context.SpringBootTest;
@@ -12,8 +15,10 @@ import org.springframework.test.context.jdbc.SqlConfig;
 import org.springframework.test.web.servlet.MvcResult;
 
 import java.nio.charset.StandardCharsets;
+import java.util.ArrayList;
 import java.util.HashMap;
 import java.util.List;
+import java.util.Map;
 
 @SpringBootTest(webEnvironment = SpringBootTest.WebEnvironment.RANDOM_PORT)
 @AutoConfigureMockMvc
@@ -27,6 +32,9 @@ public class TestPlanApiCaseControllerTests extends BaseTest {
     public static final String API_CASE_BATCH_DISASSOCIATE = "/test-plan/api/case/batch/disassociate";
     public static final String API_CASE_BATCH_UPDATE_EXECUTOR_URL = "/test-plan/api/case/batch/update/executor";
 
+    @Resource
+    private TestPlanApiCaseService testPlanApiCaseService;
+
     @Test
     @Order(1)
     @Sql(scripts = {"/dml/init_test_plan_api_case.sql"}, config = @SqlConfig(encoding = "utf-8", transactionMode = SqlConfig.TransactionMode.ISOLATED))
@@ -132,4 +140,29 @@ public class TestPlanApiCaseControllerTests extends BaseTest {
         Assertions.assertNotNull(resultHolder);
 
     }
+
+    @Test
+    @Order(6)
+    public void testApiCaseAssociate() throws Exception {
+        // api
+        Map<String, List<BaseCollectionAssociateRequest>> collectionAssociates = new HashMap<>();
+        List<BaseCollectionAssociateRequest> baseCollectionAssociateRequests = new ArrayList<>();
+        BaseCollectionAssociateRequest baseCollectionAssociateRequest = new BaseCollectionAssociateRequest();
+        baseCollectionAssociateRequest.setCollectionId("wxxx_1");
+        baseCollectionAssociateRequest.setIds(List.of("wxxx_api_1"));
+        baseCollectionAssociateRequests.add(baseCollectionAssociateRequest);
+        collectionAssociates.put(AssociateCaseType.API, baseCollectionAssociateRequests);
+        testPlanApiCaseService.associateCollection("wxxx_2", collectionAssociates, "wx");
+
+        //api case
+        Map<String, List<BaseCollectionAssociateRequest>> collectionAssociates1 = new HashMap<>();
+        List<BaseCollectionAssociateRequest> baseCollectionAssociateRequests1 = new ArrayList<>();
+        BaseCollectionAssociateRequest baseCollectionAssociateRequest1 = new BaseCollectionAssociateRequest();
+        baseCollectionAssociateRequest1.setCollectionId("wxxx_1");
+        baseCollectionAssociateRequest1.setIds(List.of("wxxx_api_case_1"));
+        baseCollectionAssociateRequests1.add(baseCollectionAssociateRequest1);
+        collectionAssociates1.put(AssociateCaseType.API_CASE, baseCollectionAssociateRequests1);
+        testPlanApiCaseService.associateCollection("wxxx_2", collectionAssociates1, "wx");
+
+    }
 }