|
@@ -23,14 +23,19 @@ import static org.springframework.test.web.servlet.request.MockMvcRequestBuilder
|
|
|
import static org.springframework.test.web.servlet.result.MockMvcResultMatchers.content;
|
|
|
import static org.springframework.test.web.servlet.result.MockMvcResultMatchers.status;
|
|
|
|
|
|
+import java.util.Map;
|
|
|
+
|
|
|
import org.apache.dolphinscheduler.api.enums.Status;
|
|
|
import org.apache.dolphinscheduler.api.service.SchedulerService;
|
|
|
+import org.apache.dolphinscheduler.api.utils.PageInfo;
|
|
|
import org.apache.dolphinscheduler.api.utils.Result;
|
|
|
+import org.apache.dolphinscheduler.common.Constants;
|
|
|
import org.apache.dolphinscheduler.common.enums.FailureStrategy;
|
|
|
import org.apache.dolphinscheduler.common.enums.Priority;
|
|
|
import org.apache.dolphinscheduler.common.enums.ReleaseState;
|
|
|
import org.apache.dolphinscheduler.common.enums.WarningType;
|
|
|
import org.apache.dolphinscheduler.common.utils.JSONUtils;
|
|
|
+import org.apache.dolphinscheduler.dao.entity.Resource;
|
|
|
import org.apache.dolphinscheduler.dao.entity.User;
|
|
|
import org.junit.Assert;
|
|
|
import org.junit.Test;
|
|
@@ -68,7 +73,7 @@ public class SchedulerControllerTest extends AbstractControllerTest {
|
|
|
|
|
|
Mockito.when(schedulerService.insertSchedule(isA(User.class), isA(Long.class), isA(Long.class),
|
|
|
isA(String.class), isA(WarningType.class), isA(int.class), isA(FailureStrategy.class),
|
|
|
- isA(Priority.class), isA(String.class))).thenReturn(successResult());
|
|
|
+ isA(Priority.class), isA(String.class))).thenReturn(success());
|
|
|
|
|
|
MvcResult mvcResult = mockMvc.perform(post("/projects/{projectCode}/schedule/create",123)
|
|
|
.header(SESSION_ID, sessionId)
|
|
@@ -97,7 +102,7 @@ public class SchedulerControllerTest extends AbstractControllerTest {
|
|
|
|
|
|
Mockito.when(schedulerService.updateSchedule(isA(User.class), isA(Long.class), isA(Integer.class),
|
|
|
isA(String.class), isA(WarningType.class), isA(Integer.class), isA(FailureStrategy.class),
|
|
|
- isA(Priority.class), isA(String.class))).thenReturn(successResult());
|
|
|
+ isA(Priority.class), isA(String.class))).thenReturn(success());
|
|
|
|
|
|
MvcResult mvcResult = mockMvc.perform(post("/projects/{projectCode}/schedule/update",123)
|
|
|
.header(SESSION_ID, sessionId)
|
|
@@ -117,7 +122,7 @@ public class SchedulerControllerTest extends AbstractControllerTest {
|
|
|
paramsMap.add("id","37");
|
|
|
|
|
|
Mockito.when(schedulerService.setScheduleState(isA(User.class), isA(Long.class), isA(Integer.class),
|
|
|
- isA(ReleaseState.class))).thenReturn(successResult());
|
|
|
+ isA(ReleaseState.class))).thenReturn(success());
|
|
|
|
|
|
MvcResult mvcResult = mockMvc.perform(post("/projects/{projectCode}/schedule/online",123)
|
|
|
.header(SESSION_ID, sessionId)
|
|
@@ -137,7 +142,7 @@ public class SchedulerControllerTest extends AbstractControllerTest {
|
|
|
paramsMap.add("id","28");
|
|
|
|
|
|
Mockito.when(schedulerService.setScheduleState(isA(User.class), isA(Long.class), isA(Integer.class),
|
|
|
- isA(ReleaseState.class))).thenReturn(successResult());
|
|
|
+ isA(ReleaseState.class))).thenReturn(success());
|
|
|
|
|
|
MvcResult mvcResult = mockMvc.perform(post("/projects/{projectCode}/schedule/offline",123)
|
|
|
.header(SESSION_ID, sessionId)
|
|
@@ -151,16 +156,22 @@ public class SchedulerControllerTest extends AbstractControllerTest {
|
|
|
logger.info(mvcResult.getResponse().getContentAsString());
|
|
|
}
|
|
|
|
|
|
-
|
|
|
@Test
|
|
|
public void testQueryScheduleListPaging() throws Exception {
|
|
|
MultiValueMap<String, String> paramsMap = new LinkedMultiValueMap<>();
|
|
|
- paramsMap.add("processDefinitionId","40");
|
|
|
+ paramsMap.add("processDefinitionCode","40");
|
|
|
paramsMap.add("searchVal","test");
|
|
|
paramsMap.add("pageNo","1");
|
|
|
paramsMap.add("pageSize","30");
|
|
|
|
|
|
- MvcResult mvcResult = mockMvc.perform(get("/projects/{projectName}/schedule/list-paging","cxc_1113")
|
|
|
+ Map<String, Object> mockResult = success();
|
|
|
+ PageInfo<Resource> pageInfo = new PageInfo<>(1, 10);
|
|
|
+ mockResult.put(Constants.DATA_LIST, pageInfo);
|
|
|
+
|
|
|
+ Mockito.when(schedulerService.querySchedule(isA(User.class), isA(Long.class), isA(Long.class),
|
|
|
+ isA(String.class), isA(Integer.class), isA(Integer.class))).thenReturn(mockResult);
|
|
|
+
|
|
|
+ MvcResult mvcResult = mockMvc.perform(get("/projects/{projectCode}/schedule/list-paging",123)
|
|
|
.header(SESSION_ID, sessionId)
|
|
|
.params(paramsMap))
|
|
|
.andExpect(status().isOk())
|