Update ShellTaskTest.java

4 years ago · c086b03b8d
1 changed files with 2 additions and 0 deletions
--- a/dolphinscheduler-server/src/test/java/org/apache/dolphinscheduler/server/worker/task/shell/ShellTaskTest.java
+++ b/dolphinscheduler-server/src/test/java/org/apache/dolphinscheduler/server/worker/task/shell/ShellTaskTest.java
@ -99,6 +99,8 @@ public class ShellTaskTest {
    public void testComplementData() throws Exception {
        shellTask = new ShellTask(taskExecutionContext, logger);
        shellTask.init();
+        shellCommandExecutor.isSuccessOfYarnState(new ArrayList<>());
+        shellCommandExecutor.isSuccessOfYarnState(null);
        PowerMockito.when(shellCommandExecutor.run(anyString())).thenReturn(commandExecuteResult);
        shellTask.handle();
    }