From 545173faa31e477f9487e98611f1951cf43efa2b Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?=E7=99=BE=E5=B2=81?= Date: Mon, 27 Sep 2021 16:08:49 +0800 Subject: [PATCH] add mock getTaskLogName() for UT test pass (#6369) --- .../apache/dolphinscheduler/plugin/task/tis/TISTaskTest.java | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/dolphinscheduler-task-plugin/dolphinscheduler-task-tis/src/test/java/org/apache/dolphinscheduler/plugin/task/tis/TISTaskTest.java b/dolphinscheduler-task-plugin/dolphinscheduler-task-tis/src/test/java/org/apache/dolphinscheduler/plugin/task/tis/TISTaskTest.java index 814f401db9..adbeb36042 100644 --- a/dolphinscheduler-task-plugin/dolphinscheduler-task-tis/src/test/java/org/apache/dolphinscheduler/plugin/task/tis/TISTaskTest.java +++ b/dolphinscheduler-task-plugin/dolphinscheduler-task-tis/src/test/java/org/apache/dolphinscheduler/plugin/task/tis/TISTaskTest.java @@ -56,6 +56,7 @@ public class TISTaskTest { String taskParams = "{\"targetJobName\":\"mysql_elastic\"}"; taskExecutionContext = Mockito.mock(TaskRequest.class); + Mockito.when(taskExecutionContext.getTaskLogName()).thenReturn("tislogger"); Mockito.when(taskExecutionContext.getTaskParams()).thenReturn(taskParams); Mockito.when(taskExecutionContext.getExecutePath()).thenReturn("/tmp"); Mockito.when(taskExecutionContext.getTaskAppId()).thenReturn(UUID.randomUUID().toString()); @@ -68,8 +69,7 @@ public class TISTaskTest { Map gloabParams = Collections.singletonMap(TISTask.KEY_POOL_VAR_TIS_HOST, "127.0.0.1:8080"); Mockito.when(taskExecutionContext.getDefinedParams()).thenReturn(gloabParams); - tisTask = PowerMockito.spy(new TISTask(taskExecutionContext)); - //tisTask = new TISTask(taskExecutionContext); + tisTask = new TISTask(taskExecutionContext); tisTask.init(); }