You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
131 lines
5.1 KiB
131 lines
5.1 KiB
/* |
|
* Licensed to the Apache Software Foundation (ASF) under one or more |
|
* contributor license agreements. See the NOTICE file distributed with |
|
* this work for additional information regarding copyright ownership. |
|
* The ASF licenses this file to You under the Apache License, Version 2.0 |
|
* (the "License"); you may not use this file except in compliance with |
|
* the License. You may obtain a copy of the License at |
|
* |
|
* http://www.apache.org/licenses/LICENSE-2.0 |
|
* |
|
* Unless required by applicable law or agreed to in writing, software |
|
* distributed under the License is distributed on an "AS IS" BASIS, |
|
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. |
|
* See the License for the specific language governing permissions and |
|
* limitations under the License. |
|
*/ |
|
|
|
package org.apache.dolphinscheduler.server.master; |
|
|
|
// @ExtendWith(MockitoExtension.class) |
|
// @MockitoSettings(strictness = Strictness.LENIENT) |
|
// public class SwitchTaskTest { |
|
// |
|
// private ProcessService processService; |
|
// |
|
// private ProcessInstance processInstance; |
|
// |
|
// private TaskInstanceDao taskInstanceDao; |
|
// |
|
// private TaskDefinitionDao taskDefinitionDao; |
|
// |
|
// @BeforeEach |
|
// public void before() { |
|
// ApplicationContext applicationContext = Mockito.mock(ApplicationContext.class); |
|
// SpringApplicationContext springApplicationContext = new SpringApplicationContext(); |
|
// springApplicationContext.setApplicationContext(applicationContext); |
|
// |
|
// MasterConfig config = new MasterConfig(); |
|
// Mockito.when(applicationContext.getBean(MasterConfig.class)).thenReturn(config); |
|
// config.setTaskCommitRetryTimes(3); |
|
// config.setTaskCommitInterval(Duration.ofSeconds(1)); |
|
// |
|
// processService = Mockito.mock(ProcessService.class); |
|
// Mockito.when(applicationContext.getBean(ProcessService.class)).thenReturn(processService); |
|
// |
|
// taskInstanceDao = Mockito.mock(TaskInstanceDao.class); |
|
// Mockito.when(applicationContext.getBean(TaskInstanceDao.class)).thenReturn(taskInstanceDao); |
|
// |
|
// taskDefinitionDao = Mockito.mock(TaskDefinitionDao.class); |
|
// Mockito.when(SpringApplicationContext.getBean(TaskDefinitionDao.class)).thenReturn(taskDefinitionDao); |
|
// |
|
// processInstance = getProcessInstance(); |
|
// Mockito.when(processService |
|
// .findProcessInstanceById(processInstance.getId())) |
|
// .thenReturn(processInstance); |
|
// } |
|
// |
|
// private TaskInstance testBasicInit(WorkflowExecutionStatus expectResult) { |
|
// TaskDefinition taskDefinition = new TaskDefinition(); |
|
// taskDefinition.setTimeoutFlag(TimeoutFlag.OPEN); |
|
// taskDefinition.setTimeoutNotifyStrategy(TaskTimeoutStrategy.WARN); |
|
// taskDefinition.setTimeout(0); |
|
// Mockito.when(taskDefinitionDao.findTaskDefinition(1L, 1)) |
|
// .thenReturn(taskDefinition); |
|
// TaskInstance taskInstance = getTaskInstance(getTaskNode(), processInstance); |
|
// |
|
// // for MasterBaseTaskExecThread.submit |
|
// Mockito.when(processService |
|
// .submitTask(processInstance, taskInstance)) |
|
// .thenReturn(taskInstance); |
|
// // for MasterBaseTaskExecThread.call |
|
// Mockito.when(taskInstanceDao |
|
// .findTaskInstanceById(taskInstance.getId())) |
|
// .thenReturn(taskInstance); |
|
// // for SwitchTaskExecThread.initTaskParameters |
|
// Mockito.when(taskInstanceDao.upsertTaskInstance(taskInstance)) |
|
// .thenReturn(true); |
|
// // for SwitchTaskExecThread.updateTaskState |
|
// Mockito.when(taskInstanceDao |
|
// .updateTaskInstance(taskInstance)) |
|
// .thenReturn(true); |
|
// |
|
// return taskInstance; |
|
// } |
|
// |
|
// private SwitchParameters getTaskNode() { |
|
// SwitchParameters conditionsParameters = new SwitchParameters(); |
|
// |
|
// SwitchResultVo switchResultVo1 = new SwitchResultVo(); |
|
// switchResultVo1.setCondition(" 2 == 1"); |
|
// switchResultVo1.setNextNode("t1"); |
|
// SwitchResultVo switchResultVo2 = new SwitchResultVo(); |
|
// switchResultVo2.setCondition(" 2 == 2"); |
|
// switchResultVo2.setNextNode("t2"); |
|
// SwitchResultVo switchResultVo3 = new SwitchResultVo(); |
|
// switchResultVo3.setCondition(" 3 == 2"); |
|
// switchResultVo3.setNextNode("t3"); |
|
// List<SwitchResultVo> list = new ArrayList<>(); |
|
// list.add(switchResultVo1); |
|
// list.add(switchResultVo2); |
|
// list.add(switchResultVo3); |
|
// conditionsParameters.setDependTaskList(list); |
|
// conditionsParameters.setNextNode("t"); |
|
// conditionsParameters.setRelation("AND"); |
|
// |
|
// return conditionsParameters; |
|
// } |
|
// |
|
// private ProcessInstance getProcessInstance() { |
|
// ProcessInstance processInstance = new ProcessInstance(); |
|
// processInstance.setId(1000); |
|
// processInstance.setState(WorkflowExecutionStatus.RUNNING_EXECUTION); |
|
// processInstance.setProcessDefinitionCode(1L); |
|
// return processInstance; |
|
// } |
|
// |
|
// private TaskInstance getTaskInstance(SwitchParameters conditionsParameters, ProcessInstance processInstance) { |
|
// TaskInstance taskInstance = new TaskInstance(); |
|
// taskInstance.setId(1000); |
|
// Map<String, Object> taskParamsMap = new HashMap<>(); |
|
// taskParamsMap.put(Constants.SWITCH_RESULT, ""); |
|
// taskInstance.setTaskParams(JSONUtils.toJsonString(taskParamsMap)); |
|
// taskInstance.setSwitchDependency(conditionsParameters); |
|
// taskInstance.setName("C"); |
|
// taskInstance.setTaskType("SWITCH"); |
|
// taskInstance.setProcessInstanceId(processInstance.getId()); |
|
// taskInstance.setTaskCode(1L); |
|
// taskInstance.setTaskDefinitionVersion(1); |
|
// return taskInstance; |
|
// } |
|
// }
|
|
|