package cn.iocoder.yudao.module.crm.service.quotation;
|
|
import cn.iocoder.yudao.module.crm.controller.admin.contract.vo.contract.CrmContractSaveReqVO;
|
import cn.iocoder.yudao.module.crm.dal.dataobject.contract.CrmContractDO;
|
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.mysql.quotation.CrmSaleQuotationMapper;
|
import cn.iocoder.yudao.module.crm.dal.mysql.quotation.CrmSaleQuotationProductMapper;
|
import cn.iocoder.yudao.module.crm.service.contract.CrmContractService;
|
import com.fasterxml.jackson.databind.ObjectMapper;
|
import org.junit.jupiter.api.Test;
|
import org.springframework.test.util.ReflectionTestUtils;
|
|
import java.math.BigDecimal;
|
import java.lang.reflect.Proxy;
|
import java.util.List;
|
import java.util.concurrent.atomic.AtomicReference;
|
|
import static org.junit.jupiter.api.Assertions.assertEquals;
|
|
class CrmSaleQuotationQuantityTest {
|
@Test
|
void contractRequestAcceptsFractionalQuantity() throws Exception {
|
var product = new ObjectMapper().readValue("{\"itemId\":1103,\"count\":22304.8}",
|
CrmContractSaveReqVO.Product.class);
|
assertEquals(new BigDecimal("22304.8"), product.getCount());
|
}
|
|
@Test
|
void conversionKeepsFractionalQuantity() {
|
var service = new CrmSaleQuotationServiceImpl();
|
var request = new AtomicReference<CrmContractSaveReqVO>();
|
var quote = new CrmSaleQuotationDO().setId(1L).setNo("BJ-1").setName("台账报价").setStatus(0);
|
var quotes = stub(CrmSaleQuotationMapper.class, (proxy, method, args) -> switch (method.getName()) {
|
case "selectById" -> quote;
|
case "updateById" -> 1;
|
default -> throw new AssertionError(method.getName());
|
});
|
var products = stub(CrmSaleQuotationProductMapper.class, (proxy, method, args) -> {
|
assertEquals("selectListByQuotationId", method.getName());
|
return List.of(new CrmSaleQuotationProductDO().setItemId(1103L).setCount(new BigDecimal("22304.8")));
|
});
|
var contracts = stub(CrmContractService.class, (proxy, method, args) -> switch (method.getName()) {
|
case "createContract" -> {
|
assertEquals(291L, args[1]);
|
request.set((CrmContractSaveReqVO) args[0]);
|
yield 2L;
|
}
|
case "getContract" -> new CrmContractDO().setNo("HT-2");
|
default -> throw new AssertionError(method.getName());
|
});
|
ReflectionTestUtils.setField(service, "saleQuotationMapper", quotes);
|
ReflectionTestUtils.setField(service, "saleQuotationProductMapper", products);
|
ReflectionTestUtils.setField(service, "contractService", contracts);
|
assertEquals(2L, service.convertToContract(1L, 291L));
|
assertEquals(new BigDecimal("22304.8"), request.get().getProducts().getFirst().getCount());
|
}
|
|
private static <T> T stub(Class<T> type, java.lang.reflect.InvocationHandler handler) {
|
return type.cast(Proxy.newProxyInstance(type.getClassLoader(), new Class<?>[]{type}, handler));
|
}
|
}
|
