Wenjun Ruan
8 months ago
committed by
GitHub
12 changed files with 463 additions and 297 deletions
@ -0,0 +1,104 @@ |
|||||||
|
/* |
||||||
|
* 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.dao.repository.impl; |
||||||
|
|
||||||
|
import static org.apache.commons.collections4.CollectionUtils.isEmpty; |
||||||
|
import static org.junit.jupiter.api.Assertions.assertEquals; |
||||||
|
import static org.junit.jupiter.api.Assertions.assertTrue; |
||||||
|
|
||||||
|
import org.apache.dolphinscheduler.common.enums.WorkflowExecutionStatus; |
||||||
|
import org.apache.dolphinscheduler.dao.BaseDaoTest; |
||||||
|
import org.apache.dolphinscheduler.dao.entity.ProcessInstance; |
||||||
|
import org.apache.dolphinscheduler.dao.repository.ProcessInstanceDao; |
||||||
|
|
||||||
|
import org.junit.jupiter.api.Test; |
||||||
|
import org.springframework.beans.factory.annotation.Autowired; |
||||||
|
|
||||||
|
class ProcessInstanceDaoImplTest extends BaseDaoTest { |
||||||
|
|
||||||
|
@Autowired |
||||||
|
private ProcessInstanceDao processInstanceDao; |
||||||
|
|
||||||
|
@Test |
||||||
|
void queryByWorkflowCodeVersionStatus_EMPTY_INSTANCE() { |
||||||
|
long workflowDefinitionCode = 1L; |
||||||
|
int workflowDefinitionVersion = 1; |
||||||
|
int[] status = WorkflowExecutionStatus.getNeedFailoverWorkflowInstanceState(); |
||||||
|
|
||||||
|
assertTrue(isEmpty(processInstanceDao.queryByWorkflowCodeVersionStatus(workflowDefinitionCode, |
||||||
|
workflowDefinitionVersion, status))); |
||||||
|
} |
||||||
|
|
||||||
|
@Test |
||||||
|
void queryByWorkflowCodeVersionStatus_EXIST_NOT_FINISH_INSTANCE() { |
||||||
|
long workflowDefinitionCode = 1L; |
||||||
|
int workflowDefinitionVersion = 1; |
||||||
|
int[] status = WorkflowExecutionStatus.getNotTerminalStatus(); |
||||||
|
|
||||||
|
assertTrue(isEmpty(processInstanceDao.queryByWorkflowCodeVersionStatus(workflowDefinitionCode, |
||||||
|
workflowDefinitionVersion, status))); |
||||||
|
|
||||||
|
processInstanceDao.insert(createWorkflowInstance(workflowDefinitionCode, workflowDefinitionVersion, |
||||||
|
WorkflowExecutionStatus.SUBMITTED_SUCCESS)); |
||||||
|
processInstanceDao.insert(createWorkflowInstance(workflowDefinitionCode, workflowDefinitionVersion, |
||||||
|
WorkflowExecutionStatus.RUNNING_EXECUTION)); |
||||||
|
processInstanceDao.insert(createWorkflowInstance(workflowDefinitionCode, workflowDefinitionVersion, |
||||||
|
WorkflowExecutionStatus.DELAY_EXECUTION)); |
||||||
|
processInstanceDao.insert(createWorkflowInstance(workflowDefinitionCode, workflowDefinitionVersion, |
||||||
|
WorkflowExecutionStatus.READY_PAUSE)); |
||||||
|
processInstanceDao.insert(createWorkflowInstance(workflowDefinitionCode, workflowDefinitionVersion, |
||||||
|
WorkflowExecutionStatus.READY_STOP)); |
||||||
|
processInstanceDao.insert(createWorkflowInstance(workflowDefinitionCode, workflowDefinitionVersion, |
||||||
|
WorkflowExecutionStatus.SERIAL_WAIT)); |
||||||
|
processInstanceDao.insert(createWorkflowInstance(workflowDefinitionCode, workflowDefinitionVersion, |
||||||
|
WorkflowExecutionStatus.READY_BLOCK)); |
||||||
|
processInstanceDao.insert(createWorkflowInstance(workflowDefinitionCode, workflowDefinitionVersion, |
||||||
|
WorkflowExecutionStatus.WAIT_TO_RUN)); |
||||||
|
assertEquals(8, processInstanceDao |
||||||
|
.queryByWorkflowCodeVersionStatus(workflowDefinitionCode, workflowDefinitionVersion, status).size()); |
||||||
|
} |
||||||
|
|
||||||
|
@Test |
||||||
|
void queryByWorkflowCodeVersionStatus_EXIST_FINISH_INSTANCE() { |
||||||
|
long workflowDefinitionCode = 1L; |
||||||
|
int workflowDefinitionVersion = 1; |
||||||
|
int[] status = WorkflowExecutionStatus.getNotTerminalStatus(); |
||||||
|
|
||||||
|
processInstanceDao.insert(createWorkflowInstance(workflowDefinitionCode, workflowDefinitionVersion, |
||||||
|
WorkflowExecutionStatus.PAUSE)); |
||||||
|
processInstanceDao.insert(createWorkflowInstance(workflowDefinitionCode, workflowDefinitionVersion, |
||||||
|
WorkflowExecutionStatus.STOP)); |
||||||
|
processInstanceDao.insert(createWorkflowInstance(workflowDefinitionCode, workflowDefinitionVersion, |
||||||
|
WorkflowExecutionStatus.FAILURE)); |
||||||
|
processInstanceDao.insert(createWorkflowInstance(workflowDefinitionCode, workflowDefinitionVersion, |
||||||
|
WorkflowExecutionStatus.SUCCESS)); |
||||||
|
assertTrue(isEmpty(processInstanceDao.queryByWorkflowCodeVersionStatus(workflowDefinitionCode, |
||||||
|
workflowDefinitionVersion, status))); |
||||||
|
} |
||||||
|
|
||||||
|
private ProcessInstance createWorkflowInstance(Long workflowDefinitionCode, int workflowDefinitionVersion, |
||||||
|
WorkflowExecutionStatus status) { |
||||||
|
ProcessInstance processInstance = new ProcessInstance(); |
||||||
|
processInstance.setName("WorkflowInstance" + System.currentTimeMillis()); |
||||||
|
processInstance.setProcessDefinitionCode(workflowDefinitionCode); |
||||||
|
processInstance.setProcessDefinitionVersion(workflowDefinitionVersion); |
||||||
|
processInstance.setState(status); |
||||||
|
return processInstance; |
||||||
|
} |
||||||
|
|
||||||
|
} |
Loading…
Reference in new issue