package cn.iocoder.yudao.module.mes.service.md.item;
|
|
import cn.iocoder.yudao.framework.common.pojo.PageResult;
|
import cn.iocoder.yudao.module.mes.controller.admin.md.item.vo.MesMdItemPageReqVO;
|
import cn.iocoder.yudao.module.mes.dal.dataobject.md.item.MesMdItemDO;
|
import cn.iocoder.yudao.module.mes.dal.dataobject.md.item.MesMdItemTypeDO;
|
import cn.iocoder.yudao.module.mes.dal.mysql.md.item.MesMdItemMapper;
|
import org.junit.jupiter.api.Test;
|
import org.junit.jupiter.api.extension.ExtendWith;
|
import org.mockito.InjectMocks;
|
import org.mockito.Mock;
|
import org.mockito.junit.jupiter.MockitoExtension;
|
|
import java.util.List;
|
import java.util.Set;
|
|
import static org.junit.jupiter.api.Assertions.assertSame;
|
import static org.mockito.Mockito.verify;
|
import static org.mockito.Mockito.when;
|
|
@ExtendWith(MockitoExtension.class)
|
class MesMdItemServiceImplTest {
|
|
@InjectMocks
|
private MesMdItemServiceImpl itemService;
|
|
@Mock
|
private MesMdItemMapper itemMapper;
|
@Mock
|
private MesMdItemTypeService itemTypeService;
|
|
@Test
|
void getItemPage_excludesEquipmentAndItsChildren() {
|
MesMdItemPageReqVO reqVO = new MesMdItemPageReqVO();
|
reqVO.setExcludeItemTypeCode("EQUIPMENT");
|
MesMdItemTypeDO equipment = new MesMdItemTypeDO().setId(100L).setCode("EQUIPMENT");
|
MesMdItemTypeDO equipmentChild = new MesMdItemTypeDO().setId(101L).setParentId(100L);
|
PageResult<MesMdItemDO> expected = new PageResult<>(List.of(), 0L);
|
when(itemTypeService.getItemTypeByCode("EQUIPMENT")).thenReturn(equipment);
|
when(itemTypeService.getItemTypeChildrenList(100L)).thenReturn(List.of(equipmentChild));
|
when(itemMapper.selectPage(reqVO, null, Set.of(100L, 101L))).thenReturn(expected);
|
|
PageResult<MesMdItemDO> result = itemService.getItemPage(reqVO);
|
|
assertSame(expected, result);
|
verify(itemMapper).selectPage(reqVO, null, Set.of(100L, 101L));
|
}
|
|
}
|
