natas_hw
/
dolphinscheduler0701
0
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

[Improvement][UT] Add UT for end time check when force success task (#15562)

This commit is contained in:
Rick Cheng 2024-02-08 09:19:56 +08:00 committed by GitHub
parent 3526b42df6
commit 718a01bb42
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 3 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
dolphinscheduler-api/src/test/java/org/apache/dolphinscheduler/api/service/TaskInstanceServiceTest.java
View File

@ -343,7 +343,7 @@ public class TaskInstanceServiceTest {
}
@Test
public void forceTaskSuccess() {
public void testForceTaskSuccess() {
User user = getAdminUser();
long projectCode = 1L;
Project project = getProject(projectCode);
@ -392,12 +392,14 @@ public class TaskInstanceServiceTest {
// test success
task.setState(TaskExecutionStatus.FAILURE);
task.setEndTime(null);
when(taskInstanceMapper.updateById(task)).thenReturn(1);
putMsg(result, Status.SUCCESS, projectCode);
when(projectMapper.queryByCode(projectCode)).thenReturn(project);
when(projectService.checkProjectAndAuth(user, project, projectCode, FORCED_SUCCESS)).thenReturn(result);
Result successRes = taskInstanceService.forceTaskSuccess(user, projectCode, taskId);
Assertions.assertEquals(Status.SUCCESS.getCode(), successRes.getCode().intValue());
Assertions.assertNotNull(task.getEndTime());
}