package cn.iocoder.yudao.module.crm.service.invoice;
|
|
import cn.iocoder.yudao.framework.common.exception.ServiceException;
|
import cn.iocoder.yudao.module.crm.dal.dataobject.contract.CrmContractProductDO;
|
import cn.iocoder.yudao.module.crm.service.contract.CrmContractService;
|
import cn.iocoder.yudao.module.mdm.api.item.MdmItemApi;
|
import cn.iocoder.yudao.module.mdm.api.item.dto.MdmItemRespDTO;
|
import cn.iocoder.yudao.module.mdm.enums.MdmItemTypeEnum;
|
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.Map;
|
import java.util.Set;
|
|
import static org.junit.jupiter.api.Assertions.assertDoesNotThrow;
|
import static org.junit.jupiter.api.Assertions.assertEquals;
|
import static org.junit.jupiter.api.Assertions.assertThrows;
|
import static org.mockito.Mockito.when;
|
|
@ExtendWith(MockitoExtension.class)
|
class CrmInvoiceServiceImplTest {
|
|
@Mock
|
private CrmContractService contractService;
|
@Mock
|
private MdmItemApi mdmItemApi;
|
@InjectMocks
|
private CrmInvoiceServiceImpl invoiceService;
|
|
@Test
|
void validateContractItemsInvoiceable_rejectsMaterial() {
|
mockContractItem(MdmItemTypeEnum.MATERIAL);
|
|
ServiceException exception = assertThrows(ServiceException.class,
|
() -> invoiceService.validateContractItemsInvoiceable(1L));
|
|
assertEquals("物料不允许开票", exception.getMessage());
|
}
|
|
@Test
|
void validateContractItemsInvoiceable_allowsProduct() {
|
mockContractItem(MdmItemTypeEnum.PRODUCT);
|
|
assertDoesNotThrow(() -> invoiceService.validateContractItemsInvoiceable(1L));
|
}
|
|
private void mockContractItem(MdmItemTypeEnum itemType) {
|
CrmContractProductDO contractProduct = CrmContractProductDO.builder().itemId(10L).build();
|
MdmItemRespDTO item = new MdmItemRespDTO();
|
item.setId(10L);
|
item.setItemType(itemType.getType());
|
when(contractService.getContractProductListByContractId(1L)).thenReturn(List.of(contractProduct));
|
when(mdmItemApi.getItemMap(Set.of(10L))).thenReturn(Map.of(10L, item));
|
}
|
|
}
|
