|
@@ -0,0 +1,195 @@
|
|
|
+package cn.iocoder.yudao.module.crm.service.contract;
|
|
|
+
|
|
|
+import cn.iocoder.yudao.framework.common.pojo.PageResult;
|
|
|
+import cn.iocoder.yudao.framework.test.core.ut.BaseDbUnitTest;
|
|
|
+import cn.iocoder.yudao.module.crm.controller.admin.contract.vo.ContractCreateReqVO;
|
|
|
+import cn.iocoder.yudao.module.crm.controller.admin.contract.vo.ContractExportReqVO;
|
|
|
+import cn.iocoder.yudao.module.crm.controller.admin.contract.vo.ContractPageReqVO;
|
|
|
+import cn.iocoder.yudao.module.crm.controller.admin.contract.vo.ContractUpdateReqVO;
|
|
|
+import cn.iocoder.yudao.module.crm.dal.dataobject.contract.ContractDO;
|
|
|
+import cn.iocoder.yudao.module.crm.dal.mysql.contract.ContractMapper;
|
|
|
+import org.junit.jupiter.api.Disabled;
|
|
|
+import org.junit.jupiter.api.Test;
|
|
|
+import org.springframework.context.annotation.Import;
|
|
|
+
|
|
|
+import javax.annotation.Resource;
|
|
|
+import java.util.List;
|
|
|
+
|
|
|
+import static cn.iocoder.yudao.framework.common.util.date.LocalDateTimeUtils.buildBetweenTime;
|
|
|
+import static cn.iocoder.yudao.framework.common.util.object.ObjectUtils.cloneIgnoreId;
|
|
|
+import static cn.iocoder.yudao.framework.test.core.util.AssertUtils.assertPojoEquals;
|
|
|
+import static cn.iocoder.yudao.framework.test.core.util.AssertUtils.assertServiceException;
|
|
|
+import static cn.iocoder.yudao.framework.test.core.util.RandomUtils.randomLongId;
|
|
|
+import static cn.iocoder.yudao.framework.test.core.util.RandomUtils.randomPojo;
|
|
|
+import static cn.iocoder.yudao.module.crm.enums.ErrorCodeConstants.CONTRACT_NOT_EXISTS;
|
|
|
+import static org.junit.jupiter.api.Assertions.*;
|
|
|
+
|
|
|
+/**
|
|
|
+ * {@link ContractServiceImpl} 的单元测试类
|
|
|
+ *
|
|
|
+ * @author dhb52
|
|
|
+ */
|
|
|
+@Import(ContractServiceImpl.class)
|
|
|
+public class ContractServiceImplTest extends BaseDbUnitTest {
|
|
|
+
|
|
|
+ @Resource
|
|
|
+ private ContractServiceImpl contractService;
|
|
|
+
|
|
|
+ @Resource
|
|
|
+ private ContractMapper contractMapper;
|
|
|
+
|
|
|
+ @Test
|
|
|
+ public void testCreateContract_success() {
|
|
|
+ // 准备参数
|
|
|
+ ContractCreateReqVO reqVO = randomPojo(ContractCreateReqVO.class);
|
|
|
+
|
|
|
+ // 调用
|
|
|
+ Long contractId = contractService.createContract(reqVO);
|
|
|
+ // 断言
|
|
|
+ assertNotNull(contractId);
|
|
|
+ // 校验记录的属性是否正确
|
|
|
+ ContractDO contract = contractMapper.selectById(contractId);
|
|
|
+ assertPojoEquals(reqVO, contract);
|
|
|
+ }
|
|
|
+
|
|
|
+ @Test
|
|
|
+ public void testUpdateContract_success() {
|
|
|
+ // mock 数据
|
|
|
+ ContractDO dbContract = randomPojo(ContractDO.class);
|
|
|
+ contractMapper.insert(dbContract);// @Sql: 先插入出一条存在的数据
|
|
|
+ // 准备参数
|
|
|
+ ContractUpdateReqVO reqVO = randomPojo(ContractUpdateReqVO.class, o -> {
|
|
|
+ o.setId(dbContract.getId()); // 设置更新的 ID
|
|
|
+ });
|
|
|
+
|
|
|
+ // 调用
|
|
|
+ contractService.updateContract(reqVO);
|
|
|
+ // 校验是否更新正确
|
|
|
+ ContractDO contract = contractMapper.selectById(reqVO.getId()); // 获取最新的
|
|
|
+ assertPojoEquals(reqVO, contract);
|
|
|
+ }
|
|
|
+
|
|
|
+ @Test
|
|
|
+ public void testUpdateContract_notExists() {
|
|
|
+ // 准备参数
|
|
|
+ ContractUpdateReqVO reqVO = randomPojo(ContractUpdateReqVO.class);
|
|
|
+
|
|
|
+ // 调用, 并断言异常
|
|
|
+ assertServiceException(() -> contractService.updateContract(reqVO), CONTRACT_NOT_EXISTS);
|
|
|
+ }
|
|
|
+
|
|
|
+ @Test
|
|
|
+ public void testDeleteContract_success() {
|
|
|
+ // mock 数据
|
|
|
+ ContractDO dbContract = randomPojo(ContractDO.class);
|
|
|
+ contractMapper.insert(dbContract);// @Sql: 先插入出一条存在的数据
|
|
|
+ // 准备参数
|
|
|
+ Long id = dbContract.getId();
|
|
|
+
|
|
|
+ // 调用
|
|
|
+ contractService.deleteContract(id);
|
|
|
+ // 校验数据不存在了
|
|
|
+ assertNull(contractMapper.selectById(id));
|
|
|
+ }
|
|
|
+
|
|
|
+ @Test
|
|
|
+ public void testDeleteContract_notExists() {
|
|
|
+ // 准备参数
|
|
|
+ Long id = randomLongId();
|
|
|
+
|
|
|
+ // 调用, 并断言异常
|
|
|
+ assertServiceException(() -> contractService.deleteContract(id), CONTRACT_NOT_EXISTS);
|
|
|
+ }
|
|
|
+
|
|
|
+ @Test
|
|
|
+ @Disabled // TODO 请修改 null 为需要的值,然后删除 @Disabled 注解
|
|
|
+ public void testGetContractPage() {
|
|
|
+ // mock 数据
|
|
|
+ ContractDO dbContract = randomPojo(ContractDO.class, o -> { // 等会查询到
|
|
|
+ o.setName(null);
|
|
|
+ o.setCustomerId(null);
|
|
|
+ o.setBusinessId(null);
|
|
|
+ o.setOrderDate(null);
|
|
|
+ o.setNo(null);
|
|
|
+ o.setDiscountPercent(null);
|
|
|
+ o.setProductPrice(null);
|
|
|
+ });
|
|
|
+ contractMapper.insert(dbContract);
|
|
|
+ // 测试 name 不匹配
|
|
|
+ contractMapper.insert(cloneIgnoreId(dbContract, o -> o.setName(null)));
|
|
|
+ // 测试 customerId 不匹配
|
|
|
+ contractMapper.insert(cloneIgnoreId(dbContract, o -> o.setCustomerId(null)));
|
|
|
+ // 测试 businessId 不匹配
|
|
|
+ contractMapper.insert(cloneIgnoreId(dbContract, o -> o.setBusinessId(null)));
|
|
|
+ // 测试 orderDate 不匹配
|
|
|
+ contractMapper.insert(cloneIgnoreId(dbContract, o -> o.setOrderDate(null)));
|
|
|
+ // 测试 no 不匹配
|
|
|
+ contractMapper.insert(cloneIgnoreId(dbContract, o -> o.setNo(null)));
|
|
|
+ // 测试 discountPercent 不匹配
|
|
|
+ contractMapper.insert(cloneIgnoreId(dbContract, o -> o.setDiscountPercent(null)));
|
|
|
+ // 测试 productPrice 不匹配
|
|
|
+ contractMapper.insert(cloneIgnoreId(dbContract, o -> o.setProductPrice(null)));
|
|
|
+ // 准备参数
|
|
|
+ ContractPageReqVO reqVO = new ContractPageReqVO();
|
|
|
+ reqVO.setName(null);
|
|
|
+ reqVO.setCustomerId(null);
|
|
|
+ reqVO.setBusinessId(null);
|
|
|
+ reqVO.setOrderDate(buildBetweenTime(2023, 2, 1, 2023, 2, 28));
|
|
|
+ reqVO.setNo(null);
|
|
|
+ reqVO.setDiscountPercent(null);
|
|
|
+ reqVO.setProductPrice(null);
|
|
|
+
|
|
|
+ // 调用
|
|
|
+ PageResult<ContractDO> pageResult = contractService.getContractPage(reqVO);
|
|
|
+ // 断言
|
|
|
+ assertEquals(1, pageResult.getTotal());
|
|
|
+ assertEquals(1, pageResult.getList().size());
|
|
|
+ assertPojoEquals(dbContract, pageResult.getList().get(0));
|
|
|
+ }
|
|
|
+
|
|
|
+ @Test
|
|
|
+ @Disabled // TODO 请修改 null 为需要的值,然后删除 @Disabled 注解
|
|
|
+ public void testGetContractList() {
|
|
|
+ // mock 数据
|
|
|
+ ContractDO dbContract = randomPojo(ContractDO.class, o -> { // 等会查询到
|
|
|
+ o.setName("合同名称");
|
|
|
+ o.setCustomerId(null);
|
|
|
+ o.setBusinessId(null);
|
|
|
+ o.setOrderDate(null);
|
|
|
+ o.setNo(null);
|
|
|
+ o.setDiscountPercent(null);
|
|
|
+ o.setProductPrice(null);
|
|
|
+ });
|
|
|
+ contractMapper.insert(dbContract);
|
|
|
+ // 测试 name 不匹配
|
|
|
+ contractMapper.insert(cloneIgnoreId(dbContract, o -> o.setName(null)));
|
|
|
+ // 测试 customerId 不匹配
|
|
|
+ contractMapper.insert(cloneIgnoreId(dbContract, o -> o.setCustomerId(null)));
|
|
|
+ // 测试 businessId 不匹配
|
|
|
+ contractMapper.insert(cloneIgnoreId(dbContract, o -> o.setBusinessId(null)));
|
|
|
+ // 测试 orderDate 不匹配
|
|
|
+ contractMapper.insert(cloneIgnoreId(dbContract, o -> o.setOrderDate(null)));
|
|
|
+ // 测试 no 不匹配
|
|
|
+ contractMapper.insert(cloneIgnoreId(dbContract, o -> o.setNo(null)));
|
|
|
+ // 测试 discountPercent 不匹配
|
|
|
+ contractMapper.insert(cloneIgnoreId(dbContract, o -> o.setDiscountPercent(null)));
|
|
|
+ // 测试 productPrice 不匹配
|
|
|
+ contractMapper.insert(cloneIgnoreId(dbContract, o -> o.setProductPrice(null)));
|
|
|
+ // 准备参数
|
|
|
+ ContractExportReqVO reqVO = new ContractExportReqVO();
|
|
|
+ reqVO.setName(null);
|
|
|
+ reqVO.setCustomerId(null);
|
|
|
+ reqVO.setBusinessId(null);
|
|
|
+ reqVO.setOrderDate(buildBetweenTime(2023, 2, 1, 2023, 2, 28));
|
|
|
+ reqVO.setNo(null);
|
|
|
+ reqVO.setDiscountPercent(null);
|
|
|
+ reqVO.setProductPrice(null);
|
|
|
+
|
|
|
+ // 调用
|
|
|
+ List<ContractDO> list = contractService.getContractList(reqVO);
|
|
|
+ // 断言
|
|
|
+ assertEquals(1, list.size());
|
|
|
+ assertPojoEquals(dbContract, list.get(0));
|
|
|
+ }
|
|
|
+
|
|
|
+}
|