Browse Source
* rebase dev * change task state to dispatch if dispatch success * update task host when dispatch * add dispatch task event * test * check tenant after enable auto create * handle dispatch state Co-authored-by: caishunfeng <534328519@qq.com>3.0.0/version-upgrade
caishunfeng
3 years ago
committed by
GitHub
35 changed files with 873 additions and 585 deletions
@ -0,0 +1,91 @@
|
||||
/* |
||||
* 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.processor; |
||||
|
||||
import org.apache.dolphinscheduler.remote.command.TaskExecuteRunningCommand; |
||||
import org.apache.dolphinscheduler.server.master.processor.queue.TaskEvent; |
||||
import org.apache.dolphinscheduler.server.master.processor.queue.TaskEventService; |
||||
import org.apache.dolphinscheduler.service.bean.SpringApplicationContext; |
||||
import org.apache.dolphinscheduler.service.process.ProcessService; |
||||
|
||||
import java.util.Date; |
||||
|
||||
import org.junit.Before; |
||||
import org.junit.Test; |
||||
import org.junit.runner.RunWith; |
||||
import org.powermock.api.mockito.PowerMockito; |
||||
import org.powermock.core.classloader.annotations.PrepareForTest; |
||||
import org.powermock.modules.junit4.PowerMockRunner; |
||||
|
||||
import io.netty.channel.Channel; |
||||
|
||||
/** |
||||
* task ack processor test |
||||
*/ |
||||
@RunWith(PowerMockRunner.class) |
||||
@PrepareForTest({SpringApplicationContext.class, TaskEvent.class}) |
||||
public class TaskAckProcessorTest { |
||||
|
||||
private TaskExecuteRunningProcessor taskExecuteRunningProcessor; |
||||
private TaskEventService taskEventService; |
||||
private ProcessService processService; |
||||
private TaskExecuteRunningCommand taskExecuteRunningCommand; |
||||
private TaskEvent taskResponseEvent; |
||||
private Channel channel; |
||||
|
||||
@Before |
||||
public void before() { |
||||
PowerMockito.mockStatic(SpringApplicationContext.class); |
||||
|
||||
taskEventService = PowerMockito.mock(TaskEventService.class); |
||||
PowerMockito.when(SpringApplicationContext.getBean(TaskEventService.class)).thenReturn(taskEventService); |
||||
|
||||
processService = PowerMockito.mock(ProcessService.class); |
||||
PowerMockito.when(SpringApplicationContext.getBean(ProcessService.class)).thenReturn(processService); |
||||
|
||||
taskExecuteRunningProcessor = new TaskExecuteRunningProcessor(); |
||||
|
||||
channel = PowerMockito.mock(Channel.class); |
||||
taskResponseEvent = PowerMockito.mock(TaskEvent.class); |
||||
|
||||
taskExecuteRunningCommand = new TaskExecuteRunningCommand(); |
||||
taskExecuteRunningCommand.setStatus(1); |
||||
taskExecuteRunningCommand.setExecutePath("/dolphinscheduler/worker"); |
||||
taskExecuteRunningCommand.setHost("localhost"); |
||||
taskExecuteRunningCommand.setLogPath("/temp/worker.log"); |
||||
taskExecuteRunningCommand.setStartTime(new Date()); |
||||
taskExecuteRunningCommand.setTaskInstanceId(1); |
||||
taskExecuteRunningCommand.setProcessInstanceId(1); |
||||
} |
||||
|
||||
@Test |
||||
public void testProcess() { |
||||
// Command command = taskExecuteAckCommand.convert2Command();
|
||||
// Assert.assertEquals(CommandType.TASK_EXECUTE_ACK,command.getType());
|
||||
// InetSocketAddress socketAddress = new InetSocketAddress("localhost",12345);
|
||||
// PowerMockito.when(channel.remoteAddress()).thenReturn(socketAddress);
|
||||
// PowerMockito.mockStatic(TaskResponseEvent.class);
|
||||
//
|
||||
// PowerMockito.when(TaskResponseEvent.newAck(Mockito.any(), Mockito.any(), Mockito.anyString(), Mockito.anyString(), Mockito.anyString(), Mockito.anyInt(), channel))
|
||||
// .thenReturn(taskResponseEvent);
|
||||
// TaskInstance taskInstance = PowerMockito.mock(TaskInstance.class);
|
||||
// PowerMockito.when(processService.findTaskInstanceById(Mockito.any())).thenReturn(taskInstance);
|
||||
//
|
||||
// taskAckProcessor.process(channel,command);
|
||||
} |
||||
} |
||||
Loading…
Reference in new issue