package com.zy.asrs.task;
|
|
import com.baomidou.mybatisplus.core.conditions.query.QueryWrapper;
|
import com.zy.asrs.entity.WrkMast;
|
import com.zy.asrs.service.LocMastService;
|
import com.zy.asrs.service.WrkAnalysisService;
|
import com.zy.asrs.service.WrkMastLogService;
|
import com.zy.asrs.service.WrkMastService;
|
import com.zy.asrs.utils.NotifyUtils;
|
import com.zy.core.enums.WrkIoType;
|
import com.zy.core.enums.WrkStsType;
|
import org.junit.jupiter.api.Test;
|
|
import java.util.List;
|
|
import static org.mockito.ArgumentMatchers.any;
|
import static org.mockito.Mockito.mock;
|
import static org.mockito.Mockito.verify;
|
import static org.mockito.Mockito.when;
|
|
class WrkMastSchedulerCrnMoveTest {
|
|
@Test
|
void executeCrnMove_archivesAndRemovesCompletedCrnMoveTask() {
|
WrkMastService wrkMastService = mock(WrkMastService.class);
|
WrkMastLogService wrkMastLogService = mock(WrkMastLogService.class);
|
WrkAnalysisService wrkAnalysisService = mock(WrkAnalysisService.class);
|
LocMastService locMastService = mock(LocMastService.class);
|
NotifyUtils notifyUtils = mock(NotifyUtils.class);
|
|
WrkMastScheduler scheduler = new WrkMastScheduler(
|
wrkMastService,
|
wrkMastLogService,
|
wrkAnalysisService,
|
locMastService,
|
notifyUtils
|
);
|
|
WrkMast wrkMast = new WrkMast();
|
wrkMast.setWrkNo(30001);
|
wrkMast.setIoType(WrkIoType.CRN_MOVE.id);
|
wrkMast.setWrkSts(WrkStsType.COMPLETE_CRN_MOVE.sts);
|
|
when(wrkMastService.list(any(QueryWrapper.class))).thenReturn(List.of(wrkMast));
|
when(wrkMastLogService.save(30001)).thenReturn(true);
|
when(wrkMastService.removeById(30001)).thenReturn(true);
|
|
scheduler.executeCrnMove();
|
|
verify(wrkMastLogService).save(30001);
|
verify(wrkAnalysisService).finishTask(any(WrkMast.class), any());
|
verify(wrkMastService).removeById(30001);
|
}
|
}
|
