增加工作流代码可正常启动
This commit is contained in:
戚辰先生
parent
2c39b79515
commit
bdbd4528a5
@ -1,325 +0,0 @@
|
|||||||
package com.qqchen.deploy.backend.workflow.engine;
|
|
||||||
|
|
||||||
import com.fasterxml.jackson.databind.ObjectMapper;
|
|
||||||
import com.qqchen.deploy.backend.workflow.dto.WorkflowDefinitionDTO;
|
|
||||||
import com.qqchen.deploy.backend.workflow.engine.context.DefaultWorkflowContext;
|
|
||||||
import com.qqchen.deploy.backend.workflow.engine.executor.*;
|
|
||||||
import com.qqchen.deploy.backend.workflow.entity.NodeInstance;
|
|
||||||
import com.qqchen.deploy.backend.workflow.entity.WorkflowDefinition;
|
|
||||||
import com.qqchen.deploy.backend.workflow.entity.WorkflowInstance;
|
|
||||||
import com.qqchen.deploy.backend.workflow.enums.NodeStatusEnum;
|
|
||||||
import com.qqchen.deploy.backend.workflow.enums.NodeTypeEnum;
|
|
||||||
import com.qqchen.deploy.backend.workflow.enums.WorkflowStatusEnum;
|
|
||||||
import com.qqchen.deploy.backend.workflow.repository.INodeInstanceRepository;
|
|
||||||
import com.qqchen.deploy.backend.workflow.repository.IWorkflowDefinitionRepository;
|
|
||||||
import com.qqchen.deploy.backend.workflow.repository.IWorkflowInstanceRepository;
|
|
||||||
import com.qqchen.deploy.backend.workflow.service.IWorkflowDefinitionService;
|
|
||||||
import com.qqchen.deploy.backend.workflow.service.IWorkflowLogService;
|
|
||||||
import com.qqchen.deploy.backend.workflow.service.IWorkflowVariableService;
|
|
||||||
import jakarta.annotation.Resource;
|
|
||||||
import lombok.extern.slf4j.Slf4j;
|
|
||||||
import org.junit.jupiter.api.BeforeEach;
|
|
||||||
import org.junit.jupiter.api.Test;
|
|
||||||
import org.springframework.boot.test.context.SpringBootTest;
|
|
||||||
import org.springframework.boot.test.mock.mockito.MockBean;
|
|
||||||
import org.springframework.context.annotation.Bean;
|
|
||||||
import org.springframework.context.annotation.Configuration;
|
|
||||||
|
|
||||||
import java.util.HashMap;
|
|
||||||
import java.util.List;
|
|
||||||
import java.util.Map;
|
|
||||||
import java.util.Optional;
|
|
||||||
|
|
||||||
import static org.junit.jupiter.api.Assertions.*;
|
|
||||||
import static org.mockito.ArgumentMatchers.any;
|
|
||||||
import static org.mockito.ArgumentMatchers.anyString;
|
|
||||||
import static org.mockito.Mockito.*;
|
|
||||||
|
|
||||||
@Slf4j
|
|
||||||
@SpringBootTest(classes = {
|
|
||||||
WorkflowEngineTest.TestConfig.class,
|
|
||||||
DefaultWorkflowEngine.class,
|
|
||||||
ObjectMapper.class
|
|
||||||
})
|
|
||||||
class WorkflowEngineTest {
|
|
||||||
|
|
||||||
@Configuration
|
|
||||||
static class TestConfig {
|
|
||||||
@Bean
|
|
||||||
Map<NodeTypeEnum, NodeExecutor> nodeExecutors() {
|
|
||||||
Map<NodeTypeEnum, NodeExecutor> executors = new HashMap<>();
|
|
||||||
StartNodeExecutor startExecutor = mock(StartNodeExecutor.class);
|
|
||||||
doNothing().when(startExecutor).execute(any(), any());
|
|
||||||
|
|
||||||
EndNodeExecutor endExecutor = mock(EndNodeExecutor.class);
|
|
||||||
doNothing().when(endExecutor).execute(any(), any());
|
|
||||||
|
|
||||||
TaskNodeExecutor taskExecutor = mock(TaskNodeExecutor.class);
|
|
||||||
doNothing().when(taskExecutor).execute(any(), any());
|
|
||||||
|
|
||||||
GatewayNodeExecutor gatewayExecutor = mock(GatewayNodeExecutor.class);
|
|
||||||
doNothing().when(gatewayExecutor).execute(any(), any());
|
|
||||||
|
|
||||||
executors.put(NodeTypeEnum.START, startExecutor);
|
|
||||||
executors.put(NodeTypeEnum.END, endExecutor);
|
|
||||||
executors.put(NodeTypeEnum.TASK, taskExecutor);
|
|
||||||
executors.put(NodeTypeEnum.GATEWAY, gatewayExecutor);
|
|
||||||
return executors;
|
|
||||||
}
|
|
||||||
|
|
||||||
@Bean
|
|
||||||
IWorkflowVariableService workflowVariableService() {
|
|
||||||
IWorkflowVariableService service = mock(IWorkflowVariableService.class);
|
|
||||||
doNothing().when(service).setVariable(any(), anyString(), any());
|
|
||||||
return service;
|
|
||||||
}
|
|
||||||
|
|
||||||
@Bean
|
|
||||||
IWorkflowLogService workflowLogService() {
|
|
||||||
return mock(IWorkflowLogService.class);
|
|
||||||
}
|
|
||||||
|
|
||||||
@Bean
|
|
||||||
DefaultWorkflowContext.Factory workflowContextFactory(
|
|
||||||
IWorkflowVariableService variableService,
|
|
||||||
IWorkflowLogService logService
|
|
||||||
) {
|
|
||||||
return new DefaultWorkflowContext.Factory(variableService, logService);
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
@Resource
|
|
||||||
private WorkflowEngine workflowEngine;
|
|
||||||
|
|
||||||
@MockBean
|
|
||||||
private IWorkflowDefinitionService workflowDefinitionService;
|
|
||||||
|
|
||||||
@MockBean
|
|
||||||
private IWorkflowDefinitionRepository workflowDefinitionRepository;
|
|
||||||
|
|
||||||
@MockBean
|
|
||||||
private IWorkflowInstanceRepository workflowInstanceRepository;
|
|
||||||
|
|
||||||
@MockBean
|
|
||||||
private INodeInstanceRepository nodeInstanceRepository;
|
|
||||||
|
|
||||||
@Resource
|
|
||||||
private ObjectMapper objectMapper;
|
|
||||||
|
|
||||||
private NodeInstance savedStartNode;
|
|
||||||
|
|
||||||
@BeforeEach
|
|
||||||
void setUp() {
|
|
||||||
log.info("Setting up workflow engine test mocks...");
|
|
||||||
|
|
||||||
// Mock workflowDefinitionService
|
|
||||||
when(workflowDefinitionService.create(any())).thenAnswer(i -> {
|
|
||||||
WorkflowDefinitionDTO dto = i.getArgument(0);
|
|
||||||
dto.setId(1L);
|
|
||||||
log.info("Created workflow definition with ID: {}", dto.getId());
|
|
||||||
return dto;
|
|
||||||
});
|
|
||||||
|
|
||||||
// Mock workflowDefinitionRepository
|
|
||||||
when(workflowDefinitionRepository.findByCodeAndDeletedFalse(anyString()))
|
|
||||||
.thenAnswer(i -> {
|
|
||||||
WorkflowDefinition definition = new WorkflowDefinition();
|
|
||||||
definition.setId(1L);
|
|
||||||
definition.setName("简单测试工作流");
|
|
||||||
definition.setCode("SIMPLE_TEST_WORKFLOW");
|
|
||||||
definition.setStatus(WorkflowStatusEnum.PUBLISHED);
|
|
||||||
definition.setNodeConfig(createSimpleWorkflow().getNodeConfig());
|
|
||||||
definition.setTransitionConfig(createSimpleWorkflow().getTransitionConfig());
|
|
||||||
log.info("Found workflow definition: code={}, status={}", definition.getCode(), definition.getStatus());
|
|
||||||
return definition;
|
|
||||||
});
|
|
||||||
|
|
||||||
// Mock workflowInstanceRepository
|
|
||||||
when(workflowInstanceRepository.save(any())).thenAnswer(i -> {
|
|
||||||
WorkflowInstance instance = i.getArgument(0);
|
|
||||||
if (instance.getId() == null) {
|
|
||||||
instance.setId(1L);
|
|
||||||
}
|
|
||||||
log.info("Saved workflow instance: id={}, status={}", instance.getId(), instance.getStatus());
|
|
||||||
return instance;
|
|
||||||
});
|
|
||||||
|
|
||||||
// Mock nodeInstanceRepository
|
|
||||||
when(nodeInstanceRepository.save(any())).thenAnswer(i -> {
|
|
||||||
NodeInstance node = i.getArgument(0);
|
|
||||||
if (node.getId() == null) {
|
|
||||||
node.setId(System.nanoTime());
|
|
||||||
}
|
|
||||||
savedStartNode = node;
|
|
||||||
|
|
||||||
// Create and set the workflow instance if it doesn't exist
|
|
||||||
if (node.getWorkflowInstance() == null) {
|
|
||||||
WorkflowInstance instance = new WorkflowInstance();
|
|
||||||
instance.setId(1L);
|
|
||||||
instance.setStatus(WorkflowStatusEnum.RUNNING);
|
|
||||||
node.setWorkflowInstance(instance);
|
|
||||||
}
|
|
||||||
|
|
||||||
log.info("Saved node instance: id={}, type={}, status={}, workflowInstanceId={}",
|
|
||||||
node.getId(), node.getNodeType(), node.getStatus(), node.getWorkflowInstance().getId());
|
|
||||||
return node;
|
|
||||||
});
|
|
||||||
|
|
||||||
when(nodeInstanceRepository.findById(any())).thenAnswer(i -> {
|
|
||||||
if (savedStartNode != null && savedStartNode.getId().equals(i.getArgument(0))) {
|
|
||||||
log.info("Found node instance by id: {}, type={}, status={}",
|
|
||||||
savedStartNode.getId(), savedStartNode.getNodeType(), savedStartNode.getStatus());
|
|
||||||
return Optional.of(savedStartNode);
|
|
||||||
}
|
|
||||||
log.warn("Node instance not found with id: {}", (Object)i.getArgument(0));
|
|
||||||
return Optional.empty();
|
|
||||||
});
|
|
||||||
|
|
||||||
when(nodeInstanceRepository.findByWorkflowInstanceIdOrderByCreateTime(any()))
|
|
||||||
.thenAnswer(i -> {
|
|
||||||
WorkflowInstance instance = new WorkflowInstance();
|
|
||||||
instance.setId(1L);
|
|
||||||
instance.setStatus(WorkflowStatusEnum.RUNNING);
|
|
||||||
|
|
||||||
List<NodeInstance> nodes = List.of(
|
|
||||||
createNodeInstance(NodeTypeEnum.START, NodeStatusEnum.COMPLETED),
|
|
||||||
createNodeInstance(NodeTypeEnum.TASK, NodeStatusEnum.COMPLETED),
|
|
||||||
createNodeInstance(NodeTypeEnum.END, NodeStatusEnum.COMPLETED)
|
|
||||||
);
|
|
||||||
|
|
||||||
// Set workflow instance for each node
|
|
||||||
nodes.forEach(node -> node.setWorkflowInstance(instance));
|
|
||||||
|
|
||||||
log.info("Found {} nodes for workflow instance id={}", nodes.size(), i.getArgument(0));
|
|
||||||
nodes.forEach(node -> log.info("Node: type={}, status={}", node.getNodeType(), node.getStatus()));
|
|
||||||
|
|
||||||
return nodes;
|
|
||||||
});
|
|
||||||
|
|
||||||
log.info("Test setup completed");
|
|
||||||
}
|
|
||||||
|
|
||||||
@Test
|
|
||||||
void testSimpleWorkflow() {
|
|
||||||
log.info("Starting simple workflow test");
|
|
||||||
|
|
||||||
// 1. 创建工作流定义
|
|
||||||
WorkflowDefinitionDTO definition = createSimpleWorkflow();
|
|
||||||
log.info("Created workflow definition: name={}, code={}", definition.getName(), definition.getCode());
|
|
||||||
|
|
||||||
// 2. 启动工作流实例
|
|
||||||
Map<String, Object> variables = new HashMap<>();
|
|
||||||
variables.put("input", "test");
|
|
||||||
log.info("Starting workflow with variables: {}", variables);
|
|
||||||
|
|
||||||
WorkflowInstance instance = workflowEngine.startWorkflow(
|
|
||||||
definition.getCode(),
|
|
||||||
"TEST-" + System.currentTimeMillis(),
|
|
||||||
variables
|
|
||||||
);
|
|
||||||
|
|
||||||
// 3. 验证工作流实例状态
|
|
||||||
assertNotNull(instance);
|
|
||||||
assertEquals(WorkflowStatusEnum.RUNNING, instance.getStatus());
|
|
||||||
log.info("Workflow instance started: id={}, status={}", instance.getId(), instance.getStatus());
|
|
||||||
|
|
||||||
// 4. 验证节点执行状态
|
|
||||||
List<NodeInstance> nodes = nodeInstanceRepository.findByWorkflowInstanceIdOrderByCreateTime(instance.getId());
|
|
||||||
assertEquals(3, nodes.size());
|
|
||||||
log.info("Found {} nodes in workflow instance", nodes.size());
|
|
||||||
|
|
||||||
// 验证开始节点
|
|
||||||
NodeInstance startNode = nodes.get(0);
|
|
||||||
assertEquals(NodeTypeEnum.START, startNode.getNodeType());
|
|
||||||
assertEquals(NodeStatusEnum.COMPLETED, startNode.getStatus());
|
|
||||||
log.info("Start node validation passed: id={}, status={}", startNode.getId(), startNode.getStatus());
|
|
||||||
|
|
||||||
// 验证Shell节点
|
|
||||||
NodeInstance shellNode = nodes.get(1);
|
|
||||||
assertEquals(NodeTypeEnum.TASK, shellNode.getNodeType());
|
|
||||||
assertEquals(NodeStatusEnum.COMPLETED, shellNode.getStatus());
|
|
||||||
log.info("Shell node validation passed: id={}, status={}", shellNode.getId(), shellNode.getStatus());
|
|
||||||
|
|
||||||
// 验证结束节点
|
|
||||||
NodeInstance endNode = nodes.get(2);
|
|
||||||
assertEquals(NodeTypeEnum.END, endNode.getNodeType());
|
|
||||||
assertEquals(NodeStatusEnum.COMPLETED, endNode.getStatus());
|
|
||||||
log.info("End node validation passed: id={}, status={}", endNode.getId(), endNode.getStatus());
|
|
||||||
|
|
||||||
log.info("Simple workflow test completed successfully");
|
|
||||||
}
|
|
||||||
|
|
||||||
private WorkflowDefinitionDTO createSimpleWorkflow() {
|
|
||||||
WorkflowDefinitionDTO dto = new WorkflowDefinitionDTO();
|
|
||||||
dto.setId(1L);
|
|
||||||
dto.setName("简单测试工作流");
|
|
||||||
dto.setCode("SIMPLE_TEST_WORKFLOW");
|
|
||||||
dto.setStatus(WorkflowStatusEnum.PUBLISHED);
|
|
||||||
|
|
||||||
// 节点配置
|
|
||||||
String nodeConfig = """
|
|
||||||
{
|
|
||||||
"nodes": [
|
|
||||||
{
|
|
||||||
"id": "start",
|
|
||||||
"name": "开始",
|
|
||||||
"type": "START"
|
|
||||||
},
|
|
||||||
{
|
|
||||||
"id": "shell",
|
|
||||||
"name": "Shell任务",
|
|
||||||
"type": "TASK",
|
|
||||||
"config": {
|
|
||||||
"type": "SHELL",
|
|
||||||
"script": "echo 'Hello World'",
|
|
||||||
"workingDirectory": "/tmp",
|
|
||||||
"successExitCode": 0
|
|
||||||
}
|
|
||||||
},
|
|
||||||
{
|
|
||||||
"id": "end",
|
|
||||||
"name": "结束",
|
|
||||||
"type": "END"
|
|
||||||
}
|
|
||||||
]
|
|
||||||
}
|
|
||||||
""";
|
|
||||||
dto.setNodeConfig(nodeConfig);
|
|
||||||
|
|
||||||
// 流转配置
|
|
||||||
String transitionConfig = """
|
|
||||||
[
|
|
||||||
{
|
|
||||||
"sourceNodeId": "start",
|
|
||||||
"targetNodeId": "shell",
|
|
||||||
"condition": null,
|
|
||||||
"priority": 1
|
|
||||||
},
|
|
||||||
{
|
|
||||||
"sourceNodeId": "shell",
|
|
||||||
"targetNodeId": "end",
|
|
||||||
"condition": null,
|
|
||||||
"priority": 1
|
|
||||||
}
|
|
||||||
]
|
|
||||||
""";
|
|
||||||
dto.setTransitionConfig(transitionConfig);
|
|
||||||
|
|
||||||
return dto;
|
|
||||||
}
|
|
||||||
|
|
||||||
private NodeInstance createNodeInstance(NodeTypeEnum type, NodeStatusEnum status) {
|
|
||||||
NodeInstance node = new NodeInstance();
|
|
||||||
node.setId(System.nanoTime());
|
|
||||||
node.setNodeType(type);
|
|
||||||
node.setStatus(status);
|
|
||||||
|
|
||||||
// Create and set a workflow instance
|
|
||||||
WorkflowInstance instance = new WorkflowInstance();
|
|
||||||
instance.setId(1L);
|
|
||||||
instance.setStatus(WorkflowStatusEnum.RUNNING);
|
|
||||||
node.setWorkflowInstance(instance);
|
|
||||||
|
|
||||||
return node;
|
|
||||||
}
|
|
||||||
}
|
|
||||||
@ -1,138 +0,0 @@
|
|||||||
package com.qqchen.deploy.backend.workflow.repository;
|
|
||||||
|
|
||||||
import com.qqchen.deploy.backend.workflow.entity.NodeDefinition;
|
|
||||||
import com.qqchen.deploy.backend.workflow.entity.WorkflowDefinition;
|
|
||||||
import com.qqchen.deploy.backend.workflow.enums.NodeTypeEnum;
|
|
||||||
import com.qqchen.deploy.backend.workflow.enums.WorkflowStatusEnum;
|
|
||||||
import org.junit.jupiter.api.BeforeEach;
|
|
||||||
import org.junit.jupiter.api.Test;
|
|
||||||
import org.springframework.beans.factory.annotation.Autowired;
|
|
||||||
import org.springframework.boot.test.autoconfigure.orm.jpa.DataJpaTest;
|
|
||||||
import org.springframework.test.context.ActiveProfiles;
|
|
||||||
|
|
||||||
import java.util.List;
|
|
||||||
|
|
||||||
import static org.assertj.core.api.Assertions.assertThat;
|
|
||||||
|
|
||||||
@DataJpaTest
|
|
||||||
@ActiveProfiles("test")
|
|
||||||
class NodeDefinitionRepositoryTest {
|
|
||||||
|
|
||||||
@Autowired
|
|
||||||
private INodeDefinitionRepository nodeDefinitionRepository;
|
|
||||||
|
|
||||||
@Autowired
|
|
||||||
private IWorkflowDefinitionRepository workflowDefinitionRepository;
|
|
||||||
|
|
||||||
private WorkflowDefinition workflowDefinition;
|
|
||||||
|
|
||||||
@BeforeEach
|
|
||||||
void setUp() {
|
|
||||||
// 创建工作流定义
|
|
||||||
workflowDefinition = new WorkflowDefinition();
|
|
||||||
workflowDefinition.setCode("TEST-WF");
|
|
||||||
workflowDefinition.setName("Test Workflow");
|
|
||||||
workflowDefinition.setStatus(WorkflowStatusEnum.DRAFT);
|
|
||||||
workflowDefinition.setVersion(1);
|
|
||||||
workflowDefinition.setEnabled(true);
|
|
||||||
workflowDefinition = workflowDefinitionRepository.save(workflowDefinition);
|
|
||||||
}
|
|
||||||
|
|
||||||
@Test
|
|
||||||
void findByWorkflowDefinitionId_ShouldReturnNodesOrderedByOrderNum() {
|
|
||||||
// Given
|
|
||||||
NodeDefinition node1 = createNodeDefinition("NODE-001", "Node 1", NodeTypeEnum.START, 1);
|
|
||||||
NodeDefinition node2 = createNodeDefinition("NODE-002", "Node 2", NodeTypeEnum.TASK, 2);
|
|
||||||
NodeDefinition node3 = createNodeDefinition("NODE-003", "Node 3", NodeTypeEnum.END, 3);
|
|
||||||
|
|
||||||
nodeDefinitionRepository.save(node1);
|
|
||||||
nodeDefinitionRepository.save(node2);
|
|
||||||
nodeDefinitionRepository.save(node3);
|
|
||||||
|
|
||||||
// When
|
|
||||||
List<NodeDefinition> results = nodeDefinitionRepository.findByWorkflowDefinitionId(workflowDefinition.getId());
|
|
||||||
|
|
||||||
// Then
|
|
||||||
assertThat(results).hasSize(3);
|
|
||||||
assertThat(results.get(0).getOrderNum()).isEqualTo(1);
|
|
||||||
assertThat(results.get(1).getOrderNum()).isEqualTo(2);
|
|
||||||
assertThat(results.get(2).getOrderNum()).isEqualTo(3);
|
|
||||||
}
|
|
||||||
|
|
||||||
@Test
|
|
||||||
void findByWorkflowDefinitionIdAndType_ShouldReturnNodesWithSpecificType() {
|
|
||||||
// Given
|
|
||||||
NodeDefinition startNode = createNodeDefinition("START-001", "Start Node", NodeTypeEnum.START, 1);
|
|
||||||
NodeDefinition taskNode1 = createNodeDefinition("TASK-001", "Task Node 1", NodeTypeEnum.TASK, 2);
|
|
||||||
NodeDefinition taskNode2 = createNodeDefinition("TASK-002", "Task Node 2", NodeTypeEnum.TASK, 3);
|
|
||||||
NodeDefinition endNode = createNodeDefinition("END-001", "End Node", NodeTypeEnum.END, 4);
|
|
||||||
|
|
||||||
nodeDefinitionRepository.save(startNode);
|
|
||||||
nodeDefinitionRepository.save(taskNode1);
|
|
||||||
nodeDefinitionRepository.save(taskNode2);
|
|
||||||
nodeDefinitionRepository.save(endNode);
|
|
||||||
|
|
||||||
// When
|
|
||||||
List<NodeDefinition> results = nodeDefinitionRepository.findByWorkflowDefinitionIdAndType(
|
|
||||||
workflowDefinition.getId(), NodeTypeEnum.TASK);
|
|
||||||
|
|
||||||
// Then
|
|
||||||
assertThat(results).hasSize(2);
|
|
||||||
assertThat(results).allMatch(node -> node.getType() == NodeTypeEnum.TASK);
|
|
||||||
}
|
|
||||||
|
|
||||||
@Test
|
|
||||||
void findByWorkflowDefinitionIdAndCode_ShouldReturnSpecificNode() {
|
|
||||||
// Given
|
|
||||||
NodeDefinition node = createNodeDefinition("NODE-001", "Test Node", NodeTypeEnum.TASK, 1);
|
|
||||||
nodeDefinitionRepository.save(node);
|
|
||||||
|
|
||||||
// When
|
|
||||||
NodeDefinition result = nodeDefinitionRepository.findByWorkflowDefinitionIdAndCodeAndDeletedFalse(
|
|
||||||
workflowDefinition.getId(), "NODE-001");
|
|
||||||
|
|
||||||
// Then
|
|
||||||
assertThat(result).isNotNull();
|
|
||||||
assertThat(result.getCode()).isEqualTo("NODE-001");
|
|
||||||
}
|
|
||||||
|
|
||||||
@Test
|
|
||||||
void existsByWorkflowDefinitionIdAndCode_ShouldReturnTrue_WhenNodeExists() {
|
|
||||||
// Given
|
|
||||||
NodeDefinition node = createNodeDefinition("NODE-001", "Test Node", NodeTypeEnum.TASK, 1);
|
|
||||||
nodeDefinitionRepository.save(node);
|
|
||||||
|
|
||||||
// When
|
|
||||||
boolean exists = nodeDefinitionRepository.existsByWorkflowDefinitionIdAndCodeAndDeletedFalse(
|
|
||||||
workflowDefinition.getId(), "NODE-001");
|
|
||||||
|
|
||||||
// Then
|
|
||||||
assertThat(exists).isTrue();
|
|
||||||
}
|
|
||||||
|
|
||||||
@Test
|
|
||||||
void deleteByWorkflowDefinitionId_ShouldMarkNodesAsDeleted() {
|
|
||||||
// Given
|
|
||||||
NodeDefinition node1 = createNodeDefinition("NODE-001", "Node 1", NodeTypeEnum.START, 1);
|
|
||||||
NodeDefinition node2 = createNodeDefinition("NODE-002", "Node 2", NodeTypeEnum.TASK, 2);
|
|
||||||
nodeDefinitionRepository.save(node1);
|
|
||||||
nodeDefinitionRepository.save(node2);
|
|
||||||
|
|
||||||
// When
|
|
||||||
nodeDefinitionRepository.deleteByWorkflowDefinitionId(workflowDefinition.getId());
|
|
||||||
|
|
||||||
// Then
|
|
||||||
List<NodeDefinition> remainingNodes = nodeDefinitionRepository.findByWorkflowDefinitionId(workflowDefinition.getId());
|
|
||||||
assertThat(remainingNodes).isEmpty();
|
|
||||||
}
|
|
||||||
|
|
||||||
private NodeDefinition createNodeDefinition(String code, String name, NodeTypeEnum type, Integer orderNum) {
|
|
||||||
NodeDefinition node = new NodeDefinition();
|
|
||||||
node.setCode(code);
|
|
||||||
node.setName(name);
|
|
||||||
node.setType(type);
|
|
||||||
node.setOrderNum(orderNum);
|
|
||||||
node.setWorkflowDefinition(workflowDefinition);
|
|
||||||
return node;
|
|
||||||
}
|
|
||||||
}
|
|
||||||
@ -1,131 +0,0 @@
|
|||||||
package com.qqchen.deploy.backend.workflow.repository;
|
|
||||||
|
|
||||||
import com.qqchen.deploy.backend.workflow.entity.WorkflowDefinition;
|
|
||||||
import com.qqchen.deploy.backend.workflow.enums.WorkflowStatusEnum;
|
|
||||||
import org.junit.jupiter.api.Test;
|
|
||||||
import org.springframework.beans.factory.annotation.Autowired;
|
|
||||||
import org.springframework.boot.test.autoconfigure.orm.jpa.DataJpaTest;
|
|
||||||
import org.springframework.test.context.ActiveProfiles;
|
|
||||||
|
|
||||||
import java.util.List;
|
|
||||||
import java.util.Optional;
|
|
||||||
|
|
||||||
import static org.assertj.core.api.Assertions.assertThat;
|
|
||||||
|
|
||||||
@DataJpaTest
|
|
||||||
@ActiveProfiles("test")
|
|
||||||
class WorkflowDefinitionRepositoryTest {
|
|
||||||
|
|
||||||
@Autowired
|
|
||||||
private IWorkflowDefinitionRepository workflowDefinitionRepository;
|
|
||||||
|
|
||||||
@Test
|
|
||||||
void findLatestByCode_ShouldReturnLatestVersion() {
|
|
||||||
// Given
|
|
||||||
WorkflowDefinition def1 = createWorkflowDefinition("TEST-001", "Test Workflow 1", 1);
|
|
||||||
WorkflowDefinition def2 = createWorkflowDefinition("TEST-001", "Test Workflow 1", 2);
|
|
||||||
workflowDefinitionRepository.save(def1);
|
|
||||||
workflowDefinitionRepository.save(def2);
|
|
||||||
|
|
||||||
// When
|
|
||||||
Optional<WorkflowDefinition> result = workflowDefinitionRepository.findLatestByCode("TEST-001");
|
|
||||||
|
|
||||||
// Then
|
|
||||||
assertThat(result).isPresent();
|
|
||||||
assertThat(result.get().getVersion()).isEqualTo(2);
|
|
||||||
}
|
|
||||||
|
|
||||||
@Test
|
|
||||||
void findByCodeAndVersion_ShouldReturnSpecificVersion() {
|
|
||||||
// Given
|
|
||||||
WorkflowDefinition def1 = createWorkflowDefinition("TEST-002", "Test Workflow 2", 1);
|
|
||||||
WorkflowDefinition def2 = createWorkflowDefinition("TEST-002", "Test Workflow 2", 2);
|
|
||||||
workflowDefinitionRepository.save(def1);
|
|
||||||
workflowDefinitionRepository.save(def2);
|
|
||||||
|
|
||||||
// When
|
|
||||||
Optional<WorkflowDefinition> result = workflowDefinitionRepository.findByCodeAndVersionAndDeletedFalse("TEST-002", 1);
|
|
||||||
|
|
||||||
// Then
|
|
||||||
assertThat(result).isPresent();
|
|
||||||
assertThat(result.get().getVersion()).isEqualTo(1);
|
|
||||||
}
|
|
||||||
|
|
||||||
@Test
|
|
||||||
void findAllVersionsByCode_ShouldReturnAllVersionsOrderedByVersionDesc() {
|
|
||||||
// Given
|
|
||||||
WorkflowDefinition def1 = createWorkflowDefinition("TEST-003", "Test Workflow 3", 1);
|
|
||||||
WorkflowDefinition def2 = createWorkflowDefinition("TEST-003", "Test Workflow 3", 2);
|
|
||||||
WorkflowDefinition def3 = createWorkflowDefinition("TEST-003", "Test Workflow 3", 3);
|
|
||||||
workflowDefinitionRepository.save(def1);
|
|
||||||
workflowDefinitionRepository.save(def2);
|
|
||||||
workflowDefinitionRepository.save(def3);
|
|
||||||
|
|
||||||
// When
|
|
||||||
List<WorkflowDefinition> results = workflowDefinitionRepository.findAllVersionsByCode("TEST-003");
|
|
||||||
|
|
||||||
// Then
|
|
||||||
assertThat(results).hasSize(3);
|
|
||||||
assertThat(results.get(0).getVersion()).isEqualTo(3);
|
|
||||||
assertThat(results.get(1).getVersion()).isEqualTo(2);
|
|
||||||
assertThat(results.get(2).getVersion()).isEqualTo(1);
|
|
||||||
}
|
|
||||||
|
|
||||||
@Test
|
|
||||||
void findByStatus_ShouldReturnWorkflowsWithSpecificStatus() {
|
|
||||||
// Given
|
|
||||||
WorkflowDefinition def1 = createWorkflowDefinition("TEST-004", "Test Workflow 4", 1);
|
|
||||||
def1.setStatus(WorkflowStatusEnum.PUBLISHED);
|
|
||||||
WorkflowDefinition def2 = createWorkflowDefinition("TEST-005", "Test Workflow 5", 1);
|
|
||||||
def2.setStatus(WorkflowStatusEnum.DRAFT);
|
|
||||||
workflowDefinitionRepository.save(def1);
|
|
||||||
workflowDefinitionRepository.save(def2);
|
|
||||||
|
|
||||||
// When
|
|
||||||
List<WorkflowDefinition> results = workflowDefinitionRepository.findByStatusAndDeletedFalse(WorkflowStatusEnum.PUBLISHED);
|
|
||||||
|
|
||||||
// Then
|
|
||||||
assertThat(results).hasSize(1);
|
|
||||||
assertThat(results.get(0).getStatus()).isEqualTo(WorkflowStatusEnum.PUBLISHED);
|
|
||||||
}
|
|
||||||
|
|
||||||
@Test
|
|
||||||
void existsByCode_ShouldReturnTrueWhenCodeExists() {
|
|
||||||
// Given
|
|
||||||
WorkflowDefinition def = createWorkflowDefinition("TEST-006", "Test Workflow 6", 1);
|
|
||||||
workflowDefinitionRepository.save(def);
|
|
||||||
|
|
||||||
// When
|
|
||||||
boolean exists = workflowDefinitionRepository.existsByCodeAndDeletedFalse("TEST-006");
|
|
||||||
|
|
||||||
// Then
|
|
||||||
assertThat(exists).isTrue();
|
|
||||||
}
|
|
||||||
|
|
||||||
@Test
|
|
||||||
void findLatestVersionByCode_ShouldReturnHighestVersion() {
|
|
||||||
// Given
|
|
||||||
WorkflowDefinition def1 = createWorkflowDefinition("TEST-007", "Test Workflow 7", 1);
|
|
||||||
WorkflowDefinition def2 = createWorkflowDefinition("TEST-007", "Test Workflow 7", 2);
|
|
||||||
WorkflowDefinition def3 = createWorkflowDefinition("TEST-007", "Test Workflow 7", 3);
|
|
||||||
workflowDefinitionRepository.save(def1);
|
|
||||||
workflowDefinitionRepository.save(def2);
|
|
||||||
workflowDefinitionRepository.save(def3);
|
|
||||||
|
|
||||||
// When
|
|
||||||
Integer latestVersion = workflowDefinitionRepository.findLatestVersionByCode("TEST-007");
|
|
||||||
|
|
||||||
// Then
|
|
||||||
assertThat(latestVersion).isEqualTo(3);
|
|
||||||
}
|
|
||||||
|
|
||||||
private WorkflowDefinition createWorkflowDefinition(String code, String name, Integer version) {
|
|
||||||
WorkflowDefinition def = new WorkflowDefinition();
|
|
||||||
def.setCode(code);
|
|
||||||
def.setName(name);
|
|
||||||
def.setVersion(version);
|
|
||||||
def.setStatus(WorkflowStatusEnum.DRAFT);
|
|
||||||
def.setEnabled(true);
|
|
||||||
return def;
|
|
||||||
}
|
|
||||||
}
|
|
||||||
@ -1,168 +0,0 @@
|
|||||||
package com.qqchen.deploy.backend.workflow.service;
|
|
||||||
|
|
||||||
import com.qqchen.deploy.backend.framework.exception.BusinessException;
|
|
||||||
import com.qqchen.deploy.backend.workflow.converter.WorkflowDefinitionConverter;
|
|
||||||
import com.qqchen.deploy.backend.workflow.dto.WorkflowDefinitionDTO;
|
|
||||||
import com.qqchen.deploy.backend.workflow.entity.WorkflowDefinition;
|
|
||||||
import com.qqchen.deploy.backend.workflow.enums.WorkflowStatusEnum;
|
|
||||||
import com.qqchen.deploy.backend.workflow.repository.INodeDefinitionRepository;
|
|
||||||
import com.qqchen.deploy.backend.workflow.repository.IWorkflowDefinitionRepository;
|
|
||||||
import com.qqchen.deploy.backend.workflow.service.impl.WorkflowDefinitionServiceImpl;
|
|
||||||
import org.junit.jupiter.api.BeforeEach;
|
|
||||||
import org.junit.jupiter.api.Test;
|
|
||||||
import org.junit.jupiter.api.extension.ExtendWith;
|
|
||||||
import org.mockito.InjectMocks;
|
|
||||||
import org.mockito.Mock;
|
|
||||||
import org.mockito.junit.jupiter.MockitoExtension;
|
|
||||||
|
|
||||||
import java.util.Arrays;
|
|
||||||
import java.util.List;
|
|
||||||
import java.util.Optional;
|
|
||||||
|
|
||||||
import static org.assertj.core.api.Assertions.assertThat;
|
|
||||||
import static org.assertj.core.api.Assertions.assertThatThrownBy;
|
|
||||||
import static org.mockito.ArgumentMatchers.any;
|
|
||||||
import static org.mockito.Mockito.*;
|
|
||||||
|
|
||||||
@ExtendWith(MockitoExtension.class)
|
|
||||||
class WorkflowDefinitionServiceTest {
|
|
||||||
|
|
||||||
@Mock
|
|
||||||
private IWorkflowDefinitionRepository workflowDefinitionRepository;
|
|
||||||
|
|
||||||
@Mock
|
|
||||||
private INodeDefinitionRepository nodeDefinitionRepository;
|
|
||||||
|
|
||||||
@Mock
|
|
||||||
private WorkflowDefinitionConverter workflowDefinitionConverter;
|
|
||||||
|
|
||||||
@InjectMocks
|
|
||||||
private WorkflowDefinitionServiceImpl workflowDefinitionService;
|
|
||||||
|
|
||||||
private WorkflowDefinition mockWorkflowDefinition;
|
|
||||||
private WorkflowDefinitionDTO mockWorkflowDefinitionDTO;
|
|
||||||
|
|
||||||
@BeforeEach
|
|
||||||
void setUp() {
|
|
||||||
mockWorkflowDefinition = new WorkflowDefinition();
|
|
||||||
mockWorkflowDefinition.setId(1L);
|
|
||||||
mockWorkflowDefinition.setCode("TEST-001");
|
|
||||||
mockWorkflowDefinition.setName("Test Workflow");
|
|
||||||
mockWorkflowDefinition.setVersion(1);
|
|
||||||
mockWorkflowDefinition.setStatus(WorkflowStatusEnum.DRAFT);
|
|
||||||
mockWorkflowDefinition.setEnabled(true);
|
|
||||||
|
|
||||||
mockWorkflowDefinitionDTO = new WorkflowDefinitionDTO();
|
|
||||||
mockWorkflowDefinitionDTO.setId(1L);
|
|
||||||
mockWorkflowDefinitionDTO.setCode("TEST-001");
|
|
||||||
mockWorkflowDefinitionDTO.setName("Test Workflow");
|
|
||||||
mockWorkflowDefinitionDTO.setVersion(1);
|
|
||||||
mockWorkflowDefinitionDTO.setStatus(WorkflowStatusEnum.DRAFT);
|
|
||||||
mockWorkflowDefinitionDTO.setEnabled(true);
|
|
||||||
}
|
|
||||||
|
|
||||||
@Test
|
|
||||||
void create_ShouldCreateNewWorkflowDefinition() {
|
|
||||||
// Given
|
|
||||||
when(workflowDefinitionRepository.existsByCodeAndDeletedFalse(anyString())).thenReturn(false);
|
|
||||||
when(workflowDefinitionRepository.save(any(WorkflowDefinition.class))).thenReturn(mockWorkflowDefinition);
|
|
||||||
when(workflowDefinitionConverter.toEntity(any(WorkflowDefinitionDTO.class))).thenReturn(mockWorkflowDefinition);
|
|
||||||
when(workflowDefinitionConverter.toDto(any(WorkflowDefinition.class))).thenReturn(mockWorkflowDefinitionDTO);
|
|
||||||
|
|
||||||
// When
|
|
||||||
WorkflowDefinitionDTO result = workflowDefinitionService.create(mockWorkflowDefinitionDTO);
|
|
||||||
|
|
||||||
// Then
|
|
||||||
assertThat(result).isNotNull();
|
|
||||||
assertThat(result.getStatus()).isEqualTo(WorkflowStatusEnum.DRAFT);
|
|
||||||
assertThat(result.getVersion()).isEqualTo(1);
|
|
||||||
verify(workflowDefinitionRepository).save(any(WorkflowDefinition.class));
|
|
||||||
}
|
|
||||||
|
|
||||||
@Test
|
|
||||||
void create_ShouldThrowException_WhenCodeExists() {
|
|
||||||
// Given
|
|
||||||
when(workflowDefinitionRepository.existsByCodeAndDeletedFalse(anyString())).thenReturn(true);
|
|
||||||
|
|
||||||
// When/Then
|
|
||||||
assertThatThrownBy(() -> workflowDefinitionService.create(mockWorkflowDefinitionDTO))
|
|
||||||
.isInstanceOf(BusinessException.class)
|
|
||||||
.hasMessage("工作流编码已存在");
|
|
||||||
}
|
|
||||||
|
|
||||||
@Test
|
|
||||||
void publish_ShouldPublishWorkflowDefinition() {
|
|
||||||
// Given
|
|
||||||
mockWorkflowDefinition.setStatus(WorkflowStatusEnum.DRAFT);
|
|
||||||
when(workflowDefinitionRepository.findById(anyLong())).thenReturn(Optional.of(mockWorkflowDefinition));
|
|
||||||
when(workflowDefinitionRepository.save(any(WorkflowDefinition.class))).thenReturn(mockWorkflowDefinition);
|
|
||||||
when(workflowDefinitionConverter.toDto(any(WorkflowDefinition.class))).thenReturn(mockWorkflowDefinitionDTO);
|
|
||||||
|
|
||||||
// When
|
|
||||||
WorkflowDefinitionDTO result = workflowDefinitionService.publish(1L);
|
|
||||||
|
|
||||||
// Then
|
|
||||||
assertThat(result).isNotNull();
|
|
||||||
assertThat(result.getStatus()).isEqualTo(WorkflowStatusEnum.PUBLISHED);
|
|
||||||
verify(workflowDefinitionRepository).save(any(WorkflowDefinition.class));
|
|
||||||
}
|
|
||||||
|
|
||||||
@Test
|
|
||||||
void publish_ShouldThrowException_WhenNotDraft() {
|
|
||||||
// Given
|
|
||||||
mockWorkflowDefinition.setStatus(WorkflowStatusEnum.PUBLISHED);
|
|
||||||
when(workflowDefinitionRepository.findById(anyLong())).thenReturn(Optional.of(mockWorkflowDefinition));
|
|
||||||
|
|
||||||
// When/Then
|
|
||||||
assertThatThrownBy(() -> workflowDefinitionService.publish(1L))
|
|
||||||
.isInstanceOf(BusinessException.class)
|
|
||||||
.hasMessage("只有草稿状态的工作流定义可以发布");
|
|
||||||
}
|
|
||||||
|
|
||||||
@Test
|
|
||||||
void createNewVersion_ShouldCreateNewVersion() {
|
|
||||||
// Given
|
|
||||||
when(workflowDefinitionRepository.findById(anyLong())).thenReturn(Optional.of(mockWorkflowDefinition));
|
|
||||||
when(workflowDefinitionRepository.findLatestVersionByCode(anyString())).thenReturn(1);
|
|
||||||
when(workflowDefinitionRepository.save(any(WorkflowDefinition.class))).thenReturn(mockWorkflowDefinition);
|
|
||||||
when(workflowDefinitionConverter.toDto(any(WorkflowDefinition.class))).thenReturn(mockWorkflowDefinitionDTO);
|
|
||||||
|
|
||||||
// When
|
|
||||||
WorkflowDefinitionDTO result = workflowDefinitionService.createNewVersion(1L);
|
|
||||||
|
|
||||||
// Then
|
|
||||||
assertThat(result).isNotNull();
|
|
||||||
assertThat(result.getStatus()).isEqualTo(WorkflowStatusEnum.DRAFT);
|
|
||||||
assertThat(result.getVersion()).isEqualTo(2);
|
|
||||||
verify(workflowDefinitionRepository).save(any(WorkflowDefinition.class));
|
|
||||||
}
|
|
||||||
|
|
||||||
@Test
|
|
||||||
void findAllVersions_ShouldReturnAllVersions() {
|
|
||||||
// Given
|
|
||||||
WorkflowDefinition version1 = createWorkflowDefinition(1);
|
|
||||||
WorkflowDefinition version2 = createWorkflowDefinition(2);
|
|
||||||
List<WorkflowDefinition> versions = Arrays.asList(version2, version1);
|
|
||||||
|
|
||||||
when(workflowDefinitionRepository.findAllVersionsByCode(anyString())).thenReturn(versions);
|
|
||||||
when(workflowDefinitionConverter.toDto(any(WorkflowDefinition.class))).thenReturn(mockWorkflowDefinitionDTO);
|
|
||||||
|
|
||||||
// When
|
|
||||||
List<WorkflowDefinitionDTO> results = workflowDefinitionService.findAllVersions("TEST-001");
|
|
||||||
|
|
||||||
// Then
|
|
||||||
assertThat(results).hasSize(2);
|
|
||||||
verify(workflowDefinitionRepository).findAllVersionsByCode("TEST-001");
|
|
||||||
}
|
|
||||||
|
|
||||||
private WorkflowDefinition createWorkflowDefinition(int version) {
|
|
||||||
WorkflowDefinition def = new WorkflowDefinition();
|
|
||||||
def.setId((long) version);
|
|
||||||
def.setCode("TEST-001");
|
|
||||||
def.setName("Test Workflow");
|
|
||||||
def.setVersion(version);
|
|
||||||
def.setStatus(WorkflowStatusEnum.DRAFT);
|
|
||||||
def.setEnabled(true);
|
|
||||||
return def;
|
|
||||||
}
|
|
||||||
}
|
|
||||||
Loading…
Reference in New Issue
Block a user