package cn.iocoder.yudao.module.crm.service.quotation;
|
|
import cn.iocoder.yudao.framework.common.exception.ServiceException;
|
import cn.iocoder.yudao.module.crm.dal.dataobject.quotation.CrmSaleQuotationDO;
|
import cn.iocoder.yudao.module.crm.dal.dataobject.quotation.CrmSaleQuotationProductDO;
|
import cn.iocoder.yudao.module.crm.dal.dataobject.quotation.CrmSaleQuotationRevisionDO;
|
import cn.iocoder.yudao.module.crm.dal.mysql.quotation.CrmSaleQuotationMapper;
|
import cn.iocoder.yudao.module.crm.dal.mysql.quotation.CrmSaleQuotationProductMapper;
|
import cn.iocoder.yudao.module.crm.dal.mysql.quotation.CrmSaleQuotationRevisionMapper;
|
import cn.iocoder.yudao.module.mdm.api.item.MdmItemApi;
|
import org.junit.jupiter.api.Test;
|
import org.springframework.test.util.ReflectionTestUtils;
|
|
import java.lang.reflect.Proxy;
|
import java.math.BigDecimal;
|
import java.util.ArrayList;
|
import java.util.List;
|
import java.util.Map;
|
|
import static cn.iocoder.yudao.module.crm.enums.ErrorCodeConstants.SALE_QUOTATION_REVISION_EXISTS;
|
import static org.junit.jupiter.api.Assertions.assertEquals;
|
import static org.junit.jupiter.api.Assertions.assertNotNull;
|
import static org.junit.jupiter.api.Assertions.assertThrows;
|
|
class CrmSaleQuotationRevisionTest {
|
|
private final CrmSaleQuotationServiceImpl service = new CrmSaleQuotationServiceImpl();
|
private final List<CrmSaleQuotationRevisionDO> inserted = new ArrayList<>();
|
|
private void prepare(long existingRevisionCount) {
|
CrmSaleQuotationRevisionMapper revisionMapper = stub(CrmSaleQuotationRevisionMapper.class,
|
(proxy, method, args) -> switch (method.getName()) {
|
case "selectByQuotationIdAndRevisionNo" ->
|
(Integer) args[1] <= existingRevisionCount ? new CrmSaleQuotationRevisionDO() : null;
|
case "insert" -> {
|
inserted.add((CrmSaleQuotationRevisionDO) args[0]);
|
yield 1;
|
}
|
default -> throw new AssertionError(method.getName());
|
});
|
CrmSaleQuotationProductMapper productMapper = stub(CrmSaleQuotationProductMapper.class,
|
(proxy, method, args) -> {
|
if ("selectListByQuotationId".equals(method.getName())) {
|
return List.of(new CrmSaleQuotationProductDO().setItemId(1103L)
|
.setCount(new BigDecimal("2"))
|
.setQuotationPrice(new BigDecimal("100")));
|
}
|
throw new AssertionError(method.getName());
|
});
|
CrmSaleQuotationMapper quoteMapper = stub(CrmSaleQuotationMapper.class,
|
(proxy, method, args) -> 1);
|
MdmItemApi itemApi = stub(MdmItemApi.class, (proxy, method, args) -> Map.of());
|
ReflectionTestUtils.setField(service, "revisionMapper", revisionMapper);
|
ReflectionTestUtils.setField(service, "saleQuotationProductMapper", productMapper);
|
ReflectionTestUtils.setField(service, "saleQuotationMapper", quoteMapper);
|
ReflectionTestUtils.setField(service, "mdmItemApi", itemApi);
|
}
|
|
@Test
|
void eachSubmitIncrementsRevision() {
|
prepare(0);
|
CrmSaleQuotationDO quote = new CrmSaleQuotationDO().setId(1L).setNo("BJ1").setRevisionNo(0);
|
|
ReflectionTestUtils.invokeMethod(service, "recordRevision", quote, 291L);
|
ReflectionTestUtils.invokeMethod(service, "recordRevision", quote, 291L);
|
|
assertEquals(2, inserted.size());
|
assertEquals(1, inserted.get(0).getRevisionNo());
|
assertEquals(2, inserted.get(1).getRevisionNo());
|
assertEquals(2, quote.getRevisionNo());
|
assertNotNull(inserted.get(0).getItemsSnapshot());
|
assertEquals(1, inserted.get(0).getStatus());
|
}
|
|
@Test
|
void duplicateRevisionRejected() {
|
prepare(1);
|
CrmSaleQuotationDO quote = new CrmSaleQuotationDO().setId(1L).setNo("BJ1").setRevisionNo(0);
|
|
ServiceException ex = assertThrows(ServiceException.class,
|
() -> ReflectionTestUtils.invokeMethod(service, "recordRevision", quote, 291L));
|
assertEquals(SALE_QUOTATION_REVISION_EXISTS.getCode(), ex.getCode());
|
}
|
|
private static <T> T stub(Class<T> type, java.lang.reflect.InvocationHandler handler) {
|
return type.cast(Proxy.newProxyInstance(type.getClassLoader(), new Class<?>[]{type}, handler));
|
}
|
}
|
