[Improvement-17288] CodeGenerateException is swallowed in ProcessDefinitionDemo (#17289)

ruanwenjun · web-flow · commit ce7440d5b746 · 2025-06-28T17:33:49.000+08:00
diff --git a/dolphinscheduler-api/src/main/java/org/apache/dolphinscheduler/api/python/PythonGateway.java b/dolphinscheduler-api/src/main/java/org/apache/dolphinscheduler/api/python/PythonGateway.java
@@ -179,7 +179,7 @@ public Map<String, Object> genTaskCodeList(Integer genNum) {
     }
 
     public Map<String, Long> getCodeAndVersion(String projectName, String workflowDefinitionName,
-                                               String taskName) throws CodeGenerateUtils.CodeGenerateException {
+                                               String taskName) {
         Project project = projectMapper.queryByName(projectName);
         Map<String, Long> result = new HashMap<>();
         // project do not exists, mean task not exists too, so we should directly return init value
diff --git a/dolphinscheduler-api/src/main/java/org/apache/dolphinscheduler/api/service/impl/EnvironmentServiceImpl.java b/dolphinscheduler-api/src/main/java/org/apache/dolphinscheduler/api/service/impl/EnvironmentServiceImpl.java
@@ -31,7 +31,6 @@
 import org.apache.dolphinscheduler.common.enums.AuthorizationType;
 import org.apache.dolphinscheduler.common.enums.UserType;
 import org.apache.dolphinscheduler.common.utils.CodeGenerateUtils;
-import org.apache.dolphinscheduler.common.utils.CodeGenerateUtils.CodeGenerateException;
 import org.apache.dolphinscheduler.common.utils.JSONUtils;
 import org.apache.dolphinscheduler.dao.entity.Environment;
 import org.apache.dolphinscheduler.dao.entity.EnvironmentWorkerGroupRelation;
@@ -121,16 +120,7 @@ public Long createEnvironment(User loginUser,
         env.setOperator(loginUser.getId());
         env.setCreateTime(new Date());
         env.setUpdateTime(new Date());
-        long code = 0L;
-        try {
-            code = CodeGenerateUtils.genCode();
-            env.setCode(code);
-        } catch (CodeGenerateException e) {
-            log.error("Generate environment code error.", e);
-        }
-        if (code == 0L) {
-            throw new ServiceException(Status.INTERNAL_SERVER_ERROR_ARGS, "Error generating environment code");
-        }
+        env.setCode(CodeGenerateUtils.genCode());
 
         if (environmentMapper.insert(env) > 0) {
             if (!StringUtils.isEmpty(workerGroups)) {
diff --git a/dolphinscheduler-api/src/main/java/org/apache/dolphinscheduler/api/service/impl/K8SNamespaceServiceImpl.java b/dolphinscheduler-api/src/main/java/org/apache/dolphinscheduler/api/service/impl/K8SNamespaceServiceImpl.java
@@ -139,17 +139,8 @@ public Map<String, Object> registerK8sNamespace(User loginUser, String namespace
             return result;
         }
 
-        long code = 0L;
-        try {
-            code = CodeGenerateUtils.genCode();
-            cluster.setCode(code);
-        } catch (CodeGenerateUtils.CodeGenerateException e) {
-            log.error("Generate cluster code error.", e);
-        }
-        if (code == 0L) {
-            putMsg(result, Status.INTERNAL_SERVER_ERROR_ARGS, "Error generating cluster code");
-            return result;
-        }
+        long code = CodeGenerateUtils.genCode();
+        cluster.setCode(code);
 
         K8sNamespace k8sNamespaceObj = new K8sNamespace();
         Date now = new Date();
diff --git a/dolphinscheduler-api/src/main/java/org/apache/dolphinscheduler/api/service/impl/ProjectParameterServiceImpl.java b/dolphinscheduler-api/src/main/java/org/apache/dolphinscheduler/api/service/impl/ProjectParameterServiceImpl.java
@@ -92,24 +92,18 @@ public Result createProjectParameter(User loginUser, long projectCode, String pr
 
         Date now = new Date();
 
-        try {
-            projectParameter = ProjectParameter
-                    .builder()
-                    .paramName(projectParameterName)
-                    .paramValue(projectParameterValue)
-                    .paramDataType(projectParameterDataType)
-                    .code(CodeGenerateUtils.genCode())
-                    .projectCode(projectCode)
-                    .userId(loginUser.getId())
-                    .operator(loginUser.getId())
-                    .createTime(now)
-                    .updateTime(now)
-                    .build();
-        } catch (CodeGenerateUtils.CodeGenerateException e) {
-            log.error("Generate project parameter code error.", e);
-            putMsg(result, Status.CREATE_PROJECT_PARAMETER_ERROR);
-            return result;
-        }
+        projectParameter = ProjectParameter
+                .builder()
+                .paramName(projectParameterName)
+                .paramValue(projectParameterValue)
+                .paramDataType(projectParameterDataType)
+                .code(CodeGenerateUtils.genCode())
+                .projectCode(projectCode)
+                .userId(loginUser.getId())
+                .operator(loginUser.getId())
+                .createTime(now)
+                .updateTime(now)
+                .build();
 
         if (projectParameterMapper.insert(projectParameter) > 0) {
             log.info("Project parameter is created and id is :{}", projectParameter.getId());
diff --git a/dolphinscheduler-api/src/main/java/org/apache/dolphinscheduler/api/service/impl/ProjectServiceImpl.java b/dolphinscheduler-api/src/main/java/org/apache/dolphinscheduler/api/service/impl/ProjectServiceImpl.java
@@ -31,7 +31,6 @@
 import org.apache.dolphinscheduler.common.enums.AuthorizationType;
 import org.apache.dolphinscheduler.common.enums.UserType;
 import org.apache.dolphinscheduler.common.utils.CodeGenerateUtils;
-import org.apache.dolphinscheduler.common.utils.CodeGenerateUtils.CodeGenerateException;
 import org.apache.dolphinscheduler.dao.entity.Project;
 import org.apache.dolphinscheduler.dao.entity.ProjectUser;
 import org.apache.dolphinscheduler.dao.entity.ProjectWorkflowDefinitionCount;
@@ -120,22 +119,16 @@ public Result createProject(User loginUser, String name, String desc) {
 
         Date now = new Date();
 
-        try {
-            project = Project
-                    .builder()
-                    .name(name)
-                    .code(CodeGenerateUtils.genCode())
-                    .description(desc)
-                    .userId(loginUser.getId())
-                    .userName(loginUser.getUserName())
-                    .createTime(now)
-                    .updateTime(now)
-                    .build();
-        } catch (CodeGenerateException e) {
-            log.error("Generate workflow definition code error.", e);
-            putMsg(result, Status.CREATE_PROJECT_ERROR);
-            return result;
-        }
+        project = Project
+                .builder()
+                .name(name)
+                .code(CodeGenerateUtils.genCode())
+                .description(desc)
+                .userId(loginUser.getId())
+                .userName(loginUser.getUserName())
+                .createTime(now)
+                .updateTime(now)
+                .build();
 
         if (projectMapper.insert(project) > 0) {
             log.info("Project is created and id is :{}", project.getId());
diff --git a/dolphinscheduler-api/src/main/java/org/apache/dolphinscheduler/api/service/impl/TaskDefinitionServiceImpl.java b/dolphinscheduler-api/src/main/java/org/apache/dolphinscheduler/api/service/impl/TaskDefinitionServiceImpl.java
@@ -45,7 +45,6 @@
 import org.apache.dolphinscheduler.common.enums.TaskExecuteType;
 import org.apache.dolphinscheduler.common.enums.TimeoutFlag;
 import org.apache.dolphinscheduler.common.utils.CodeGenerateUtils;
-import org.apache.dolphinscheduler.common.utils.CodeGenerateUtils.CodeGenerateException;
 import org.apache.dolphinscheduler.common.utils.JSONUtils;
 import org.apache.dolphinscheduler.dao.entity.Project;
 import org.apache.dolphinscheduler.dao.entity.TaskDefinition;
@@ -862,14 +861,11 @@ public Map<String, Object> genTaskCodeList(Integer genNum) {
             return result;
         }
         List<Long> taskCodes = new ArrayList<>();
-        try {
-            for (int i = 0; i < genNum; i++) {
-                taskCodes.add(CodeGenerateUtils.genCode());
-            }
-        } catch (CodeGenerateException e) {
-            log.error("Generate task definition code error.", e);
-            putMsg(result, Status.INTERNAL_SERVER_ERROR_ARGS, "Error generating task definition code");
+
+        for (int i = 0; i < genNum; i++) {
+            taskCodes.add(CodeGenerateUtils.genCode());
         }
+
         putMsg(result, Status.SUCCESS);
         // return workflowDefinitionCode
         result.put(Constants.DATA_LIST, taskCodes);
diff --git a/dolphinscheduler-api/src/main/java/org/apache/dolphinscheduler/api/service/impl/WorkflowDefinitionServiceImpl.java b/dolphinscheduler-api/src/main/java/org/apache/dolphinscheduler/api/service/impl/WorkflowDefinitionServiceImpl.java
@@ -75,7 +75,6 @@
 import org.apache.dolphinscheduler.common.lifecycle.ServerLifeCycleManager;
 import org.apache.dolphinscheduler.common.model.TaskNodeRelation;
 import org.apache.dolphinscheduler.common.utils.CodeGenerateUtils;
-import org.apache.dolphinscheduler.common.utils.CodeGenerateUtils.CodeGenerateException;
 import org.apache.dolphinscheduler.common.utils.DateUtils;
 import org.apache.dolphinscheduler.common.utils.JSONUtils;
 import org.apache.dolphinscheduler.dao.entity.DagData;
@@ -349,14 +348,7 @@ public WorkflowDefinition createSingleWorkflowDefinition(User loginUser,
         WorkflowDefinition workflowDefinition = workflowCreateRequest.convert2WorkflowDefinition();
         this.createWorkflowValid(loginUser, workflowDefinition);
 
-        long workflowDefinitionCode;
-        try {
-            workflowDefinitionCode = CodeGenerateUtils.genCode();
-        } catch (CodeGenerateException e) {
-            throw new ServiceException(Status.INTERNAL_SERVER_ERROR_ARGS);
-        }
-
-        workflowDefinition.setCode(workflowDefinitionCode);
+        workflowDefinition.setCode(CodeGenerateUtils.genCode());
         workflowDefinition.setUserId(loginUser.getId());
 
         int create = workflowDefinitionMapper.insert(workflowDefinition);
@@ -1424,7 +1416,7 @@ private DataSource queryDatasourceByNameAndUser(String datasourceName, User logi
     }
 
     private TaskDefinitionLog buildNormalSqlTaskDefinition(String taskName, DataSource dataSource,
-                                                           String sql) throws CodeGenerateException {
+                                                           String sql) {
         TaskDefinitionLog taskDefinition = new TaskDefinitionLog();
         taskDefinition.setName(taskName);
         taskDefinition.setFlag(Flag.YES);
@@ -1480,15 +1472,8 @@ protected boolean checkAndImport(User loginUser,
         workflowDefinition.setId(null);
         workflowDefinition.setProjectCode(projectCode);
         workflowDefinition.setUserId(loginUser.getId());
-        try {
-            workflowDefinition.setCode(CodeGenerateUtils.genCode());
-        } catch (CodeGenerateException e) {
-            log.error(
-                    "Save workflow definition error because generate workflow definition code error, projectCode:{}.",
-                    projectCode, e);
-            putMsg(result, Status.CREATE_WORKFLOW_DEFINITION_ERROR);
-            return false;
-        }
+        workflowDefinition.setCode(CodeGenerateUtils.genCode());
+
         List<TaskDefinition> taskDefinitionList = dagDataSchedule.getTaskDefinitionList();
         Map<Long, Long> taskCodeMap = new HashMap<>();
         Date now = new Date();
@@ -1503,16 +1488,10 @@ protected boolean checkAndImport(User loginUser,
             taskDefinitionLog.setUpdateTime(now);
             taskDefinitionLog.setOperator(loginUser.getId());
             taskDefinitionLog.setOperateTime(now);
-            try {
-                long code = CodeGenerateUtils.genCode();
-                taskCodeMap.put(taskDefinitionLog.getCode(), code);
-                taskDefinitionLog.setCode(code);
-            } catch (CodeGenerateException e) {
-                log.error("Generate task definition code error, projectCode:{}, workflowDefinitionCode:{}",
-                        projectCode, workflowDefinition.getCode(), e);
-                putMsg(result, Status.INTERNAL_SERVER_ERROR_ARGS, "Error generating task definition code");
-                return false;
-            }
+            long code = CodeGenerateUtils.genCode();
+            taskCodeMap.put(taskDefinitionLog.getCode(), code);
+            taskDefinitionLog.setCode(code);
+
             taskDefinitionLogList.add(taskDefinitionLog);
         }
         int insert = taskDefinitionMapper.batchInsert(taskDefinitionLogList);
@@ -2120,15 +2099,8 @@ protected void doBatchOperateWorkflowDefinition(User loginUser,
                 List<TaskDefinitionLog> taskDefinitionLogs =
                         taskDefinitionLogDao.queryTaskDefineLogList(workflowTaskRelations);
                 Map<Long, Long> taskCodeMap = new HashMap<>();
-                taskDefinitionLogs.forEach(taskDefinitionLog -> {
-                    try {
-                        taskCodeMap.put(taskDefinitionLog.getCode(), CodeGenerateUtils.genCode());
-                    } catch (CodeGenerateException e) {
-                        log.error("Generate task definition code error, projectCode:{}.", targetProjectCode, e);
-                        putMsg(result, Status.INTERNAL_SERVER_ERROR_ARGS);
-                        throw new ServiceException(Status.INTERNAL_SERVER_ERROR_ARGS);
-                    }
-                });
+                taskDefinitionLogs.forEach(
+                        taskDefinitionLog -> taskCodeMap.put(taskDefinitionLog.getCode(), CodeGenerateUtils.genCode()));
                 for (TaskDefinitionLog taskDefinitionLog : taskDefinitionLogs) {
                     taskDefinitionLog.setCode(taskCodeMap.get(taskDefinitionLog.getCode()));
                     taskDefinitionLog.setProjectCode(targetProjectCode);
@@ -2164,13 +2136,8 @@ protected void doBatchOperateWorkflowDefinition(User loginUser,
                     }
                 }
                 final long oldWorkflowDefinitionCode = workflowDefinition.getCode();
-                try {
-                    workflowDefinition.setCode(CodeGenerateUtils.genCode());
-                } catch (CodeGenerateException e) {
-                    log.error("Generate workflow definition code error, projectCode:{}.", targetProjectCode, e);
-                    putMsg(result, Status.INTERNAL_SERVER_ERROR_ARGS);
-                    throw new ServiceException(Status.INTERNAL_SERVER_ERROR_ARGS);
-                }
+                workflowDefinition.setCode(CodeGenerateUtils.genCode());
+
                 workflowDefinition.setId(null);
                 workflowDefinition.setUserId(loginUser.getId());
                 workflowDefinition.setName(getNewName(workflowDefinition.getName(), COPY_SUFFIX));
diff --git a/dolphinscheduler-api/src/test/java/org/apache/dolphinscheduler/api/python/PythonGatewayTest.java b/dolphinscheduler-api/src/test/java/org/apache/dolphinscheduler/api/python/PythonGatewayTest.java
@@ -18,7 +18,6 @@
 package org.apache.dolphinscheduler.api.python;
 
 import org.apache.dolphinscheduler.api.service.ResourcesService;
-import org.apache.dolphinscheduler.common.utils.CodeGenerateUtils;
 import org.apache.dolphinscheduler.dao.entity.Project;
 import org.apache.dolphinscheduler.dao.entity.TaskDefinition;
 import org.apache.dolphinscheduler.dao.entity.User;
@@ -62,7 +61,7 @@ public class PythonGatewayTest {
     private ResourcesService resourcesService;
 
     @Test
-    public void testGetCodeAndVersion() throws CodeGenerateUtils.CodeGenerateException {
+    public void testGetCodeAndVersion() {
         Project project = getTestProject();
         Mockito.when(projectMapper.queryByName(project.getName())).thenReturn(project);
 
diff --git a/dolphinscheduler-api/src/test/java/org/apache/dolphinscheduler/api/service/ProjectParameterServiceTest.java b/dolphinscheduler-api/src/test/java/org/apache/dolphinscheduler/api/service/ProjectParameterServiceTest.java
@@ -21,6 +21,7 @@
 import static org.junit.jupiter.api.Assertions.assertEquals;
 import static org.junit.jupiter.api.Assertions.assertNotNull;
 import static org.junit.jupiter.api.Assertions.assertNull;
+import static org.junit.jupiter.api.Assertions.assertThrows;
 import static org.mockito.ArgumentMatchers.any;
 import static org.mockito.ArgumentMatchers.anyLong;
 import static org.mockito.Mockito.when;
@@ -90,9 +91,9 @@ public void testCreateProjectParameter() {
         try (MockedStatic<CodeGenerateUtils> ignored = Mockito.mockStatic(CodeGenerateUtils.class)) {
             when(CodeGenerateUtils.genCode()).thenThrow(CodeGenerateUtils.CodeGenerateException.class);
 
-            result = projectParameterService.createProjectParameter(loginUser, projectCode, "key", "value",
-                    DataType.VARCHAR.name());
-            assertEquals(Status.CREATE_PROJECT_PARAMETER_ERROR.getCode(), result.getCode());
+            assertThrows(CodeGenerateUtils.CodeGenerateException.class,
+                    () -> projectParameterService.createProjectParameter(loginUser, projectCode, "key", "value",
+                            DataType.VARCHAR.name()));
         }
 
         // PROJECT_PARAMETER_ALREADY_EXISTS
diff --git a/dolphinscheduler-api/src/test/java/org/apache/dolphinscheduler/api/service/impl/EnvironmentServiceTest.java b/dolphinscheduler-api/src/test/java/org/apache/dolphinscheduler/api/service/impl/EnvironmentServiceTest.java
@@ -23,6 +23,7 @@
 import static org.apache.dolphinscheduler.api.constants.ApiFuncIdentificationConstant.ENVIRONMENT_DELETE;
 import static org.apache.dolphinscheduler.api.constants.ApiFuncIdentificationConstant.ENVIRONMENT_UPDATE;
 import static org.junit.jupiter.api.Assertions.assertEquals;
+import static org.junit.jupiter.api.Assertions.assertThrows;
 import static org.mockito.ArgumentMatchers.any;
 import static org.mockito.Mockito.when;
 
@@ -140,7 +141,7 @@ public void testCreateEnvironment() {
         try (MockedStatic<CodeGenerateUtils> ignored = Mockito.mockStatic(CodeGenerateUtils.class)) {
             when(CodeGenerateUtils.genCode()).thenThrow(CodeGenerateUtils.CodeGenerateException.class);
 
-            assertThrowsServiceException(Status.INTERNAL_SERVER_ERROR_ARGS,
+            assertThrows(CodeGenerateUtils.CodeGenerateException.class,
                     () -> environmentService.createEnvironment(adminUser, "testName", "test", "test", workerGroups));
         }
     }
diff --git a/dolphinscheduler-service/src/main/java/org/apache/dolphinscheduler/service/process/ProcessService.java b/dolphinscheduler-service/src/main/java/org/apache/dolphinscheduler/service/process/ProcessService.java
@@ -21,7 +21,6 @@
 import org.apache.dolphinscheduler.common.enums.TaskGroupQueueStatus;
 import org.apache.dolphinscheduler.common.graph.DAG;
 import org.apache.dolphinscheduler.common.model.TaskNodeRelation;
-import org.apache.dolphinscheduler.common.utils.CodeGenerateUtils;
 import org.apache.dolphinscheduler.dao.entity.Command;
 import org.apache.dolphinscheduler.dao.entity.DagData;
 import org.apache.dolphinscheduler.dao.entity.DataSource;
@@ -46,7 +45,7 @@
 public interface ProcessService {
 
     WorkflowInstance constructWorkflowInstance(Command command,
-                                               String host) throws CronParseException, CodeGenerateUtils.CodeGenerateException;
+                                               String host) throws CronParseException;
 
     Optional<WorkflowInstance> findWorkflowInstanceDetailById(int workflowInstanceId);
 
diff --git a/dolphinscheduler-service/src/main/java/org/apache/dolphinscheduler/service/process/ProcessServiceImpl.java b/dolphinscheduler-service/src/main/java/org/apache/dolphinscheduler/service/process/ProcessServiceImpl.java
@@ -38,7 +38,6 @@
 import org.apache.dolphinscheduler.common.graph.DAG;
 import org.apache.dolphinscheduler.common.model.TaskNodeRelation;
 import org.apache.dolphinscheduler.common.utils.CodeGenerateUtils;
-import org.apache.dolphinscheduler.common.utils.CodeGenerateUtils.CodeGenerateException;
 import org.apache.dolphinscheduler.common.utils.DateUtils;
 import org.apache.dolphinscheduler.common.utils.JSONUtils;
 import org.apache.dolphinscheduler.dao.entity.Cluster;
@@ -492,7 +491,7 @@ private Boolean checkCmdParam(Command command, Map<String, String> cmdParam) {
      */
     @Override
     public @Nullable WorkflowInstance constructWorkflowInstance(Command command,
-                                                                String host) throws CronParseException, CodeGenerateException {
+                                                                String host) throws CronParseException {
         WorkflowInstance workflowInstance;
         WorkflowDefinition workflowDefinition;
         CommandType commandType = command.getCommandType();
diff --git a/dolphinscheduler-tools/src/main/java/org/apache/dolphinscheduler/tools/demo/CreateWorkflowDemo.java b/dolphinscheduler-tools/src/main/java/org/apache/dolphinscheduler/tools/demo/CreateWorkflowDemo.java
@@ -33,27 +33,27 @@
                 "org.apache.dolphinscheduler.tools.datasource.*",
         })
 })
-public class CreateProcessDemo {
+public class CreateWorkflowDemo {
 
     public static void main(String[] args) {
-        SpringApplication.run(CreateProcessDemo.class, args);
+        SpringApplication.run(CreateWorkflowDemo.class, args);
     }
 
     @Component
     @Profile("demo")
     @Slf4j
     static class DemoRunner implements CommandLineRunner {
 
-        private final ProcessDefinitionDemo processDefinitionDemo;
+        private final WorkflowDefinitionDemo workflowDefinitionDemo;
 
-        DemoRunner(ProcessDefinitionDemo processDefinitionDemo) {
-            this.processDefinitionDemo = processDefinitionDemo;
+        DemoRunner(WorkflowDefinitionDemo workflowDefinitionDemo) {
+            this.workflowDefinitionDemo = workflowDefinitionDemo;
         }
 
         @Override
         public void run(String... args) throws Exception {
-            processDefinitionDemo.createProcessDefinitionDemo();
-            log.info("create process definition demo success");
+            workflowDefinitionDemo.createWorkflowDefinitionDemo();
+            log.info("create workflow definition demo success");
         }
     }
 }
diff --git a/dolphinscheduler-tools/src/main/java/org/apache/dolphinscheduler/tools/demo/ProxyWorkflowDefinitionController.java b/dolphinscheduler-tools/src/main/java/org/apache/dolphinscheduler/tools/demo/ProxyWorkflowDefinitionController.java
@@ -29,7 +29,7 @@
 import org.springframework.stereotype.Service;
 
 @Service
-public class ProxyProcessDefinitionController {
+public class ProxyWorkflowDefinitionController {
 
     @Value("${demo.api-server-port}")
     private String ServerPort;
diff --git a/dolphinscheduler-tools/src/main/java/org/apache/dolphinscheduler/tools/demo/WorkflowDefinitionDemo.java b/dolphinscheduler-tools/src/main/java/org/apache/dolphinscheduler/tools/demo/WorkflowDefinitionDemo.java

Original file line number	Diff line number	Diff line change
`@@ -179,7 +179,7 @@ public Map<String, Object> genTaskCodeList(Integer genNum) {`
`179`	`179`	`}`
`180`	`180`
`181`	`181`	`public Map<String, Long> getCodeAndVersion(String projectName, String workflowDefinitionName,`
`182`		`- String taskName) throws CodeGenerateUtils.CodeGenerateException {`
	`182`	`+ String taskName) {`
`183`	`183`	`Project project = projectMapper.queryByName(projectName);`
`184`	`184`	`Map<String, Long> result = new HashMap<>();`
`185`	`185`	`// project do not exists, mean task not exists too, so we should directly return init value`
Original file line number	Diff line number	Diff line change
`@@ -23,6 +23,7 @@`
`23`	`23`	`import static org.apache.dolphinscheduler.api.constants.ApiFuncIdentificationConstant.ENVIRONMENT_DELETE;`
`24`	`24`	`import static org.apache.dolphinscheduler.api.constants.ApiFuncIdentificationConstant.ENVIRONMENT_UPDATE;`
`25`	`25`	`import static org.junit.jupiter.api.Assertions.assertEquals;`
	`26`	`+import static org.junit.jupiter.api.Assertions.assertThrows;`
`26`	`27`	`import static org.mockito.ArgumentMatchers.any;`
`27`	`28`	`import static org.mockito.Mockito.when;`
`28`	`29`
`@@ -140,7 +141,7 @@ public void testCreateEnvironment() {`
`140`	`141`	`try (MockedStatic<CodeGenerateUtils> ignored = Mockito.mockStatic(CodeGenerateUtils.class)) {`
`141`	`142`	`when(CodeGenerateUtils.genCode()).thenThrow(CodeGenerateUtils.CodeGenerateException.class);`
`142`	`143`
`143`		`- assertThrowsServiceException(Status.INTERNAL_SERVER_ERROR_ARGS,`
	`144`	`+ assertThrows(CodeGenerateUtils.CodeGenerateException.class,`
`144`	`145`	`() -> environmentService.createEnvironment(adminUser, "testName", "test", "test", workerGroups));`
`145`	`146`	`}`
`146`	`147`	`}`