feat: 客户表的 crud

2025-08-02 04:14:06 +08:00 · 2023-10-25 00:25:00 +08:00
parent 55f9e0131c
commit 04613af0cd
20 changed files with 989 additions and 7 deletions
--- a/yudao-module-crm/yudao-module-crm-biz/src/test/java/cn/iocoder/yudao/module/crm/service/customer/CrmCustomerServiceImplTest.java
+++ b/yudao-module-crm/yudao-module-crm-biz/src/test/java/cn/iocoder/yudao/module/crm/service/customer/CrmCustomerServiceImplTest.java
@@ -0,0 +1,170 @@
+package cn.iocoder.yudao.module.crm.service.customer;
+
+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.customer.vo.CrmCustomerCreateReqVO;
+import cn.iocoder.yudao.module.crm.controller.admin.customer.vo.CrmCustomerExportReqVO;
+import cn.iocoder.yudao.module.crm.controller.admin.customer.vo.CrmCustomerPageReqVO;
+import cn.iocoder.yudao.module.crm.controller.admin.customer.vo.CrmCustomerUpdateReqVO;
+import cn.iocoder.yudao.module.crm.dal.dataobject.customer.CrmCustomerDO;
+import cn.iocoder.yudao.module.crm.dal.mysql.customer.CrmCustomerMapper;
+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.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.CUSTOMER_NOT_EXISTS;
+import static org.junit.jupiter.api.Assertions.*;
+
+/**
+ * {@link CrmCustomerServiceImpl} 的单元测试类
+ *
+ * @author Wanwan
+ */
+@Import(CrmCustomerServiceImpl.class)
+public class CrmCustomerServiceImplTest extends BaseDbUnitTest {
+
+    @Resource
+    private CrmCustomerServiceImpl customerService;
+
+    @Resource
+    private CrmCustomerMapper customerMapper;
+
+    @Test
+    public void testCreateCustomer_success() {
+        // 准备参数
+        CrmCustomerCreateReqVO reqVO = randomPojo(CrmCustomerCreateReqVO.class);
+
+        // 调用
+        Long customerId = customerService.createCustomer(reqVO);
+        // 断言
+        assertNotNull(customerId);
+        // 校验记录的属性是否正确
+        CrmCustomerDO customer = customerMapper.selectById(customerId);
+        assertPojoEquals(reqVO, customer);
+    }
+
+    @Test
+    public void testUpdateCustomer_success() {
+        // mock 数据
+        CrmCustomerDO dbCustomer = randomPojo(CrmCustomerDO.class);
+        customerMapper.insert(dbCustomer);// @Sql: 先插入出一条存在的数据
+        // 准备参数
+        CrmCustomerUpdateReqVO reqVO = randomPojo(CrmCustomerUpdateReqVO.class, o -> {
+            o.setId(dbCustomer.getId()); // 设置更新的 ID
+        });
+
+        // 调用
+        customerService.updateCustomer(reqVO);
+        // 校验是否更新正确
+        CrmCustomerDO customer = customerMapper.selectById(reqVO.getId()); // 获取最新的
+        assertPojoEquals(reqVO, customer);
+    }
+
+    @Test
+    public void testUpdateCustomer_notExists() {
+        // 准备参数
+        CrmCustomerUpdateReqVO reqVO = randomPojo(CrmCustomerUpdateReqVO.class);
+
+        // 调用, 并断言异常
+        assertServiceException(() -> customerService.updateCustomer(reqVO), CUSTOMER_NOT_EXISTS);
+    }
+
+    @Test
+    public void testDeleteCustomer_success() {
+        // mock 数据
+        CrmCustomerDO dbCustomer = randomPojo(CrmCustomerDO.class);
+        customerMapper.insert(dbCustomer);// @Sql: 先插入出一条存在的数据
+        // 准备参数
+        Long id = dbCustomer.getId();
+
+        // 调用
+        customerService.deleteCustomer(id);
+        // 校验数据不存在了
+        assertNull(customerMapper.selectById(id));
+    }
+
+    @Test
+    public void testDeleteCustomer_notExists() {
+        // 准备参数
+        Long id = randomLongId();
+
+        // 调用, 并断言异常
+        assertServiceException(() -> customerService.deleteCustomer(id), CUSTOMER_NOT_EXISTS);
+    }
+
+    @Test
+    @Disabled  // TODO 请修改 null 为需要的值，然后删除 @Disabled 注解
+    public void testGetCustomerPage() {
+        // mock 数据
+        CrmCustomerDO dbCustomer = randomPojo(CrmCustomerDO.class, o -> { // 等会查询到
+            o.setName(null);
+            o.setMobile(null);
+            o.setTelephone(null);
+            o.setWebsite(null);
+        });
+        customerMapper.insert(dbCustomer);
+        // 测试 name 不匹配
+        customerMapper.insert(cloneIgnoreId(dbCustomer, o -> o.setName(null)));
+        // 测试 mobile 不匹配
+        customerMapper.insert(cloneIgnoreId(dbCustomer, o -> o.setMobile(null)));
+        // 测试 telephone 不匹配
+        customerMapper.insert(cloneIgnoreId(dbCustomer, o -> o.setTelephone(null)));
+        // 测试 website 不匹配
+        customerMapper.insert(cloneIgnoreId(dbCustomer, o -> o.setWebsite(null)));
+        // 准备参数
+        CrmCustomerPageReqVO reqVO = new CrmCustomerPageReqVO();
+        reqVO.setName(null);
+        reqVO.setMobile(null);
+        reqVO.setTelephone(null);
+        reqVO.setWebsite(null);
+
+        // 调用
+        PageResult<CrmCustomerDO> pageResult = customerService.getCustomerPage(reqVO);
+        // 断言
+        assertEquals(1, pageResult.getTotal());
+        assertEquals(1, pageResult.getList().size());
+        assertPojoEquals(dbCustomer, pageResult.getList().get(0));
+    }
+
+    @Test
+    @Disabled  // TODO 请修改 null 为需要的值，然后删除 @Disabled 注解
+    public void testGetCustomerList() {
+        // mock 数据
+        CrmCustomerDO dbCustomer = randomPojo(CrmCustomerDO.class, o -> { // 等会查询到
+            o.setName(null);
+            o.setMobile(null);
+            o.setTelephone(null);
+            o.setWebsite(null);
+        });
+        customerMapper.insert(dbCustomer);
+        // 测试 name 不匹配
+        customerMapper.insert(cloneIgnoreId(dbCustomer, o -> o.setName(null)));
+        // 测试 mobile 不匹配
+        customerMapper.insert(cloneIgnoreId(dbCustomer, o -> o.setMobile(null)));
+        // 测试 telephone 不匹配
+        customerMapper.insert(cloneIgnoreId(dbCustomer, o -> o.setTelephone(null)));
+        // 测试 website 不匹配
+        customerMapper.insert(cloneIgnoreId(dbCustomer, o -> o.setWebsite(null)));
+        // 准备参数
+        CrmCustomerExportReqVO reqVO = new CrmCustomerExportReqVO();
+        reqVO.setName(null);
+        reqVO.setMobile(null);
+        reqVO.setTelephone(null);
+        reqVO.setWebsite(null);
+
+        // 调用
+        List<CrmCustomerDO> list = customerService.getCustomerList(reqVO);
+        // 断言
+        assertEquals(1, list.size());
+        assertPojoEquals(dbCustomer, list.get(0));
+    }
+
+}
--- a/yudao-module-crm/yudao-module-crm-biz/src/test/resources/sql/clean.sql
+++ b/yudao-module-crm/yudao-module-crm-biz/src/test/resources/sql/clean.sql
@@ -5,3 +5,5 @@ DELETE FROM "crm_clue";
 DELETE FROM "crm_receivable";

 DELETE FROM "crm_receivable_plan";
+
+DELETE FROM "crm_customer";
--- a/yudao-module-crm/yudao-module-crm-biz/src/test/resources/sql/create_tables.sql
+++ b/yudao-module-crm/yudao-module-crm-biz/src/test/resources/sql/create_tables.sql
@@ -98,3 +98,30 @@ CREATE TABLE IF NOT EXISTS "crm_receivable_plan" (
     PRIMARY KEY ("id")
 ) COMMENT '回款计划';

+CREATE TABLE IF NOT EXISTS "crm_customer" (
+  "id" bigint NOT NULL GENERATED BY DEFAULT AS IDENTITY,
+  "name" varchar,
+  "follow_up_status" bit NOT NULL,
+  "lock_status" bit NOT NULL,
+  "deal_status" bit NOT NULL,
+  "mobile" varchar,
+  "telephone" varchar,
+  "website" varchar,
+  "remark" varchar,
+  "owner_user_id" bigint,
+  "ro_user_ids" varchar,
+  "rw_user_ids" varchar,
+  "area_id" bigint,
+  "detail_address" varchar,
+  "longitude" varchar,
+  "latitude" varchar,
+  "contact_last_time" varchar,
+  "contact_next_time" varchar,
+  "creator" varchar DEFAULT '',
+  "create_time" datetime NOT NULL DEFAULT CURRENT_TIMESTAMP,
+  "updater" varchar DEFAULT '',
+  "update_time" datetime NOT NULL DEFAULT CURRENT_TIMESTAMP ON UPDATE CURRENT_TIMESTAMP,
+  "deleted" bit NOT NULL DEFAULT FALSE,
+  "tenant_id" bigint NOT NULL,
+  PRIMARY KEY ("id")
+) COMMENT '客户表';