陈家名
2 years ago
committed by
GitHub
9 changed files with 185 additions and 14 deletions
@ -0,0 +1,123 @@
|
||||
/* |
||||
* 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.api.python; |
||||
|
||||
import org.apache.dolphinscheduler.common.utils.CodeGenerateUtils; |
||||
import org.apache.dolphinscheduler.dao.entity.ProcessDefinition; |
||||
import org.apache.dolphinscheduler.dao.entity.Project; |
||||
import org.apache.dolphinscheduler.dao.entity.TaskDefinition; |
||||
import org.apache.dolphinscheduler.dao.mapper.ProcessDefinitionMapper; |
||||
import org.apache.dolphinscheduler.dao.mapper.ProjectMapper; |
||||
import org.apache.dolphinscheduler.dao.mapper.TaskDefinitionMapper; |
||||
import org.junit.Assert; |
||||
import org.junit.Test; |
||||
import org.junit.runner.RunWith; |
||||
import org.mockito.InjectMocks; |
||||
import org.mockito.Mock; |
||||
import org.mockito.Mockito; |
||||
import org.mockito.junit.MockitoJUnitRunner; |
||||
|
||||
import java.util.Date; |
||||
import java.util.Map; |
||||
|
||||
/** |
||||
* python gate test |
||||
*/ |
||||
@RunWith(MockitoJUnitRunner.class) |
||||
public class PythonGatewayTest { |
||||
|
||||
@InjectMocks |
||||
private PythonGateway pythonGateway; |
||||
|
||||
@Mock |
||||
private ProjectMapper projectMapper; |
||||
|
||||
@Mock |
||||
private ProcessDefinitionMapper processDefinitionMapper; |
||||
|
||||
@Mock |
||||
private TaskDefinitionMapper taskDefinitionMapper; |
||||
|
||||
@Test |
||||
public void testGetCodeAndVersion() throws CodeGenerateUtils.CodeGenerateException { |
||||
Project project = getTestProject(); |
||||
Mockito.when(projectMapper.queryByName(project.getName())).thenReturn(project); |
||||
|
||||
ProcessDefinition processDefinition = getTestProcessDefinition(); |
||||
Mockito.when(processDefinitionMapper.queryByDefineName(project.getCode(), processDefinition.getName())).thenReturn(processDefinition); |
||||
|
||||
TaskDefinition taskDefinition = getTestTaskDefinition(); |
||||
Mockito.when(taskDefinitionMapper.queryByName(project.getCode(), processDefinition.getCode(), taskDefinition.getName())).thenReturn(taskDefinition); |
||||
|
||||
Map<String, Long> result = pythonGateway.getCodeAndVersion(project.getName(), processDefinition.getName(), taskDefinition.getName()); |
||||
Assert.assertEquals(result.get("code").longValue(), taskDefinition.getCode()); |
||||
} |
||||
|
||||
@Test |
||||
public void testGetDependentInfo() { |
||||
Project project = getTestProject(); |
||||
Mockito.when(projectMapper.queryByName(project.getName())).thenReturn(project); |
||||
|
||||
ProcessDefinition processDefinition = getTestProcessDefinition(); |
||||
Mockito.when(processDefinitionMapper.queryByDefineName(project.getCode(), processDefinition.getName())).thenReturn(processDefinition); |
||||
|
||||
TaskDefinition taskDefinition = getTestTaskDefinition(); |
||||
Mockito.when(taskDefinitionMapper.queryByName(project.getCode(), processDefinition.getCode(), taskDefinition.getName())).thenReturn(taskDefinition); |
||||
|
||||
Map<String, Object> result = pythonGateway.getDependentInfo(project.getName(), processDefinition.getName(), taskDefinition.getName()); |
||||
Assert.assertEquals((long) result.get("taskDefinitionCode"), taskDefinition.getCode()); |
||||
} |
||||
|
||||
private Project getTestProject() { |
||||
Project project = new Project(); |
||||
project.setName("ut-project"); |
||||
project.setUserId(111); |
||||
project.setCode(1L); |
||||
project.setCreateTime(new Date()); |
||||
project.setUpdateTime(new Date()); |
||||
return project; |
||||
} |
||||
|
||||
private ProcessDefinition getTestProcessDefinition() { |
||||
ProcessDefinition processDefinition = new ProcessDefinition(); |
||||
processDefinition.setCode(1L); |
||||
processDefinition.setName("ut-process-definition"); |
||||
processDefinition.setProjectCode(1L); |
||||
processDefinition.setUserId(111); |
||||
processDefinition.setUpdateTime(new Date()); |
||||
processDefinition.setCreateTime(new Date()); |
||||
return processDefinition; |
||||
} |
||||
|
||||
private TaskDefinition getTestTaskDefinition() { |
||||
TaskDefinition taskDefinition = new TaskDefinition(); |
||||
taskDefinition.setCode(888888L); |
||||
taskDefinition.setName("ut-task-definition"); |
||||
taskDefinition.setProjectCode(1L); |
||||
taskDefinition.setTaskType("SHELL"); |
||||
taskDefinition.setUserId(111); |
||||
taskDefinition.setResourceIds("1"); |
||||
taskDefinition.setWorkerGroup("default"); |
||||
taskDefinition.setEnvironmentCode(1L); |
||||
taskDefinition.setVersion(1); |
||||
taskDefinition.setCreateTime(new Date()); |
||||
taskDefinition.setUpdateTime(new Date()); |
||||
return taskDefinition; |
||||
} |
||||
|
||||
} |
||||
Loading…
Reference in new issue