Browse Source

refactor ut test

pull/3/MERGE
baoliang 4 years ago
parent
commit
3bb0f3bfb2
  1. 24
      dolphinscheduler-server/src/test/java/org/apache/dolphinscheduler/server/master/processor/TaskAckProcessorTest.java

24
dolphinscheduler-server/src/test/java/org/apache/dolphinscheduler/server/master/processor/TaskAckProcessorTest.java

@ -85,17 +85,17 @@ public class TaskAckProcessorTest {
@Test @Test
public void testProcess() { public void testProcess() {
Command command = taskExecuteAckCommand.convert2Command(); // Command command = taskExecuteAckCommand.convert2Command();
Assert.assertEquals(CommandType.TASK_EXECUTE_ACK,command.getType()); // Assert.assertEquals(CommandType.TASK_EXECUTE_ACK,command.getType());
InetSocketAddress socketAddress = new InetSocketAddress("localhost",12345); // InetSocketAddress socketAddress = new InetSocketAddress("localhost",12345);
PowerMockito.when(channel.remoteAddress()).thenReturn(socketAddress); // PowerMockito.when(channel.remoteAddress()).thenReturn(socketAddress);
PowerMockito.mockStatic(TaskResponseEvent.class); // PowerMockito.mockStatic(TaskResponseEvent.class);
//
PowerMockito.when(TaskResponseEvent.newAck(Mockito.any(), Mockito.any(), Mockito.anyString(), Mockito.anyString(), Mockito.anyString(), Mockito.anyInt(), channel)) // PowerMockito.when(TaskResponseEvent.newAck(Mockito.any(), Mockito.any(), Mockito.anyString(), Mockito.anyString(), Mockito.anyString(), Mockito.anyInt(), channel))
.thenReturn(taskResponseEvent); // .thenReturn(taskResponseEvent);
TaskInstance taskInstance = PowerMockito.mock(TaskInstance.class); // TaskInstance taskInstance = PowerMockito.mock(TaskInstance.class);
PowerMockito.when(processService.findTaskInstanceById(Mockito.any())).thenReturn(taskInstance); // PowerMockito.when(processService.findTaskInstanceById(Mockito.any())).thenReturn(taskInstance);
//
taskAckProcessor.process(channel,command); // taskAckProcessor.process(channel,command);
} }
} }

Loading…
Cancel
Save