From 52faa190c99dea98613e36c71da2ba23f2c2ad42 Mon Sep 17 00:00:00 2001 From: even <827656971@qq.com> Date: Thu, 24 Jul 2025 16:39:22 +0800 Subject: [PATCH] =?UTF-8?q?=E4=B8=8B=E8=BD=BD=E6=8E=A5=E5=8F=A3=E4=BF=AE?= =?UTF-8?q?=E6=94=B9,ai=E6=B5=8B=E8=AF=95=E7=94=A8=E4=BE=8B=E7=94=9F?= =?UTF-8?q?=E6=88=90=E4=BF=AE=E6=94=B9?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- .../src/main/java/cd/casic/ci/api/TargetController.java | 1 + .../engine/constant/TestCaseGenerationConstant.java | 2 ++ .../engine/worker/afl/TestCaseGenerationWorker.java | 7 +++---- .../service/target/impl/TargetManagerServiceImpl.java | 6 ++---- 4 files changed, 8 insertions(+), 8 deletions(-) diff --git a/modules/module-ci-process-api/src/main/java/cd/casic/ci/api/TargetController.java b/modules/module-ci-process-api/src/main/java/cd/casic/ci/api/TargetController.java index 0dde10d0..8f6f057a 100644 --- a/modules/module-ci-process-api/src/main/java/cd/casic/ci/api/TargetController.java +++ b/modules/module-ci-process-api/src/main/java/cd/casic/ci/api/TargetController.java @@ -2,6 +2,7 @@ package cd.casic.ci.api; import cd.casic.ci.process.dto.req.target.*; +import cd.casic.ci.process.dto.resp.sftpFile.FileBase64DownloadResp; import cd.casic.ci.process.dto.resp.target.TargetManagerResp; import cd.casic.ci.process.dto.resp.target.TargetVersionResp; import cd.casic.ci.process.process.dataObject.base.BaseIdReq; diff --git a/modules/module-ci-process-biz/src/main/java/cd/casic/ci/process/engine/constant/TestCaseGenerationConstant.java b/modules/module-ci-process-biz/src/main/java/cd/casic/ci/process/engine/constant/TestCaseGenerationConstant.java index f90a530c..d188286a 100644 --- a/modules/module-ci-process-biz/src/main/java/cd/casic/ci/process/engine/constant/TestCaseGenerationConstant.java +++ b/modules/module-ci-process-biz/src/main/java/cd/casic/ci/process/engine/constant/TestCaseGenerationConstant.java @@ -11,5 +11,7 @@ public class TestCaseGenerationConstant { public static final String SEED_SOURCE = "filePath"; public static final String ITEM_LIST = "itemList"; public static final String BINARY_NAME="binaryName"; + public static final String COUNT="count"; + public static final String PROMPT = "prompt"; } diff --git a/modules/module-ci-process-biz/src/main/java/cd/casic/ci/process/engine/worker/afl/TestCaseGenerationWorker.java b/modules/module-ci-process-biz/src/main/java/cd/casic/ci/process/engine/worker/afl/TestCaseGenerationWorker.java index 6a77dc4e..5b44b18f 100644 --- a/modules/module-ci-process-biz/src/main/java/cd/casic/ci/process/engine/worker/afl/TestCaseGenerationWorker.java +++ b/modules/module-ci-process-biz/src/main/java/cd/casic/ci/process/engine/worker/afl/TestCaseGenerationWorker.java @@ -128,6 +128,8 @@ public class TestCaseGenerationWorker extends DockerWorker { PipTask taskDef = context.getContextDef() instanceof PipTask ? ((PipTask) context.getContextDef()) : null; Map taskProperties = taskDef.getTaskProperties(); String binaryName = taskProperties.get(BINARY_NAME) instanceof String ? ((String) taskProperties.get("binaryName")) : null; + Integer count = taskProperties.get(COUNT) instanceof Integer ? ((Integer) taskProperties.get(COUNT)) : null; + String prompt = taskProperties.get(PROMPT) instanceof String ? ((String) taskProperties.get(COUNT)) : null; PipPipeline pipeline = (PipPipeline) getContextManager().getContext(taskDef.getPipelineId()).getContextDef(); TargetVersion targetVersion = targetVersionService.getById(pipeline.getTargetVersionId()); String managerId = getVariableNearby(context, AFL_RESOURCE_MANAGER_ID_KEY) instanceof String ? ((String) getVariableNearby(context, AFL_RESOURCE_MANAGER_ID_KEY)) : null; @@ -138,9 +140,6 @@ public class TestCaseGenerationWorker extends DockerWorker { if (Objects.isNull(machineInfo)||Objects.isNull(dockerInfo)) { append(context,"该资源不支持docker或者ssh"); } - // TODO 需要新增的项 - String prompt = "你是一个模糊测试专家，当前程序输入为文本例如 aa aaa 这种"; - Integer count = 10; // 获取编译后的文文件的inputStream JSch jsch = new JSch(); Session session = null; @@ -211,7 +210,7 @@ public class TestCaseGenerationWorker extends DockerWorker { // 解压种子zip exec = (ChannelExec)session.openChannel("exec"); exec.setCommand(cmd); - + exec.connect(); } catch (Exception e) { log.error("ai测试用例生成报错",e); append(context,"测试用例生成失败"); diff --git a/modules/module-ci-process-biz/src/main/java/cd/casic/ci/process/process/service/target/impl/TargetManagerServiceImpl.java b/modules/module-ci-process-biz/src/main/java/cd/casic/ci/process/process/service/target/impl/TargetManagerServiceImpl.java index 551597ee..eca83df6 100644 --- a/modules/module-ci-process-biz/src/main/java/cd/casic/ci/process/process/service/target/impl/TargetManagerServiceImpl.java +++ b/modules/module-ci-process-biz/src/main/java/cd/casic/ci/process/process/service/target/impl/TargetManagerServiceImpl.java @@ -2,6 +2,7 @@ package cd.casic.ci.process.process.service.target.impl; import cd.casic.ci.process.dto.req.target.*; +import cd.casic.ci.process.dto.resp.sftpFile.FileBase64DownloadResp; import cd.casic.ci.process.dto.resp.target.TargetManagerResp; import cd.casic.ci.process.dto.resp.target.TargetVersionResp; import cd.casic.ci.process.process.converter.TargetConverter; @@ -37,10 +38,7 @@ import org.springframework.web.multipart.MultipartFile; import java.io.ByteArrayOutputStream; import java.io.IOException; import java.io.InputStream; -import java.util.ArrayList; -import java.util.Collections; -import java.util.List; -import java.util.UUID; +import java.util.*; /** * @author HopeLi