@ -18,8 +18,8 @@ package org.apache.dolphinscheduler.api.service;
import org.apache.dolphinscheduler.api.enums.Status ;
import org.apache.dolphinscheduler.api.enums.Status ;
import org.apache.dolphinscheduler.api.utils.Result ;
import org.apache.dolphinscheduler.api.utils.Result ;
import org.apache.dolphinscheduler.dao.ProcessDao ;
import org.apache.dolphinscheduler.dao.entity.TaskInstance ;
import org.apache.dolphinscheduler.dao.entity.TaskInstance ;
import org.apache.dolphinscheduler.service.process.ProcessService ;
import org.junit.Assert ;
import org.junit.Assert ;
import org.junit.Test ;
import org.junit.Test ;
import org.junit.runner.RunWith ;
import org.junit.runner.RunWith ;
@ -40,14 +40,14 @@ public class LoggerServiceTest {
@InjectMocks
@InjectMocks
private LoggerService loggerService ;
private LoggerService loggerService ;
@Mock
@Mock
private ProcessDao processDao ;
private ProcessService processService ;
@Test
@Test
public void testQueryDataSourceList ( ) {
public void testQueryDataSourceList ( ) {
TaskInstance taskInstance = new TaskInstance ( ) ;
TaskInstance taskInstance = new TaskInstance ( ) ;
Mockito . when ( processDao . findTaskInstanceById ( 1 ) ) . thenReturn ( taskInstance ) ;
Mockito . when ( processService . findTaskInstanceById ( 1 ) ) . thenReturn ( taskInstance ) ;
Result result = loggerService . queryLog ( 2 , 1 , 1 ) ;
Result result = loggerService . queryLog ( 2 , 1 , 1 ) ;
//TASK_INSTANCE_NOT_FOUND
//TASK_INSTANCE_NOT_FOUND
Assert . assertEquals ( Status . TASK_INSTANCE_NOT_FOUND . getCode ( ) , result . getCode ( ) . intValue ( ) ) ;
Assert . assertEquals ( Status . TASK_INSTANCE_NOT_FOUND . getCode ( ) , result . getCode ( ) . intValue ( ) ) ;
@ -59,7 +59,7 @@ public class LoggerServiceTest {
//SUCCESS
//SUCCESS
taskInstance . setHost ( "127.0.0.1" ) ;
taskInstance . setHost ( "127.0.0.1" ) ;
taskInstance . setLogPath ( "/temp/log" ) ;
taskInstance . setLogPath ( "/temp/log" ) ;
Mockito . when ( processDao . findTaskInstanceById ( 1 ) ) . thenReturn ( taskInstance ) ;
Mockito . when ( processService . findTaskInstanceById ( 1 ) ) . thenReturn ( taskInstance ) ;
result = loggerService . queryLog ( 1 , 1 , 1 ) ;
result = loggerService . queryLog ( 1 , 1 , 1 ) ;
Assert . assertEquals ( Status . SUCCESS . getCode ( ) , result . getCode ( ) . intValue ( ) ) ;
Assert . assertEquals ( Status . SUCCESS . getCode ( ) , result . getCode ( ) . intValue ( ) ) ;
}
}
@ -68,7 +68,7 @@ public class LoggerServiceTest {
public void testGetLogBytes ( ) {
public void testGetLogBytes ( ) {
TaskInstance taskInstance = new TaskInstance ( ) ;
TaskInstance taskInstance = new TaskInstance ( ) ;
Mockito . when ( processDao . findTaskInstanceById ( 1 ) ) . thenReturn ( taskInstance ) ;
Mockito . when ( processService . findTaskInstanceById ( 1 ) ) . thenReturn ( taskInstance ) ;
//task instance is null
//task instance is null
try {
try {