完成支付模块的商户管理开发以及单元测试

2025-07-20 22:15:07 +08:00 · 2021-11-03 17:49:08 +08:00
parent aa77eb029f
commit 1a721ceb5f
25 changed files with 1258 additions and 0 deletions
--- a/yudao-admin-server/src/test/java/cn/iocoder/yudao/adminserver/modules/pay/merchant/service/PayMerchantServiceTest.java
+++ b/yudao-admin-server/src/test/java/cn/iocoder/yudao/adminserver/modules/pay/merchant/service/PayMerchantServiceTest.java
@ -0,0 +1,192 @@
+package cn.iocoder.yudao.adminserver.modules.pay.merchant.service;
+
+import cn.hutool.core.util.RandomUtil;
+import cn.iocoder.yudao.adminserver.BaseDbUnitTest;
+import cn.iocoder.yudao.adminserver.modules.pay.controller.merchant.vo.PayMerchantCreateReqVO;
+import cn.iocoder.yudao.adminserver.modules.pay.controller.merchant.vo.PayMerchantExportReqVO;
+import cn.iocoder.yudao.adminserver.modules.pay.controller.merchant.vo.PayMerchantPageReqVO;
+import cn.iocoder.yudao.adminserver.modules.pay.controller.merchant.vo.PayMerchantUpdateReqVO;
+import cn.iocoder.yudao.adminserver.modules.pay.dal.mysql.merchant.PayMerchantMapper;
+import cn.iocoder.yudao.adminserver.modules.pay.service.merchant.impl.PayMerchantServiceImpl;
+import cn.iocoder.yudao.coreservice.modules.pay.dal.dataobject.merchant.PayMerchantDO;
+import cn.iocoder.yudao.framework.common.enums.CommonStatusEnum;
+import cn.iocoder.yudao.framework.common.pojo.PageResult;
+import cn.iocoder.yudao.framework.common.util.object.ObjectUtils;
+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.coreservice.modules.pay.enums.PayErrorCodeCoreConstants.MERCHANT_NOT_EXISTS;
+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.common.util.date.DateUtils.buildTime;
+import static cn.iocoder.yudao.framework.test.core.util.RandomUtils.*;
+import static org.junit.jupiter.api.Assertions.*;
+
+/**
+* {@link PayMerchantServiceImpl} 的单元测试类
+*
+* @author 芋艿
+*/
+@Import(PayMerchantServiceImpl.class)
+public class PayMerchantServiceTest extends BaseDbUnitTest {
+
+    @Resource
+    private PayMerchantServiceImpl merchantService;
+
+    @Resource
+    private PayMerchantMapper merchantMapper;
+
+    @Test
+    public void testCreateMerchant_success() {
+        // 准备参数
+        PayMerchantCreateReqVO reqVO = randomPojo(PayMerchantCreateReqVO.class,o ->
+            o.setStatus(RandomUtil.randomEle(CommonStatusEnum.values()).getStatus()));
+
+        // 调用
+        Long merchantId = merchantService.createMerchant(reqVO);
+        // 断言
+        assertNotNull(merchantId);
+        // 校验记录的属性是否正确
+        PayMerchantDO merchant = merchantMapper.selectById(merchantId);
+        assertPojoEquals(reqVO, merchant,"no");
+    }
+
+    @Test
+    public void testUpdateMerchant_success() {
+        // mock 数据
+        PayMerchantDO dbMerchant = randomPojo(PayMerchantDO.class, o ->
+            o.setStatus(CommonStatusEnum.ENABLE.getStatus()));
+        merchantMapper.insert(dbMerchant);// @Sql: 先插入出一条存在的数据
+        // 准备参数
+        PayMerchantUpdateReqVO reqVO = randomPojo(PayMerchantUpdateReqVO.class, o -> {
+            o.setId(dbMerchant.getId()); // 设置更新的 ID
+            o.setStatus(CommonStatusEnum.DISABLE.getStatus());
+        });
+
+        // 调用
+        merchantService.updateMerchant(reqVO);
+        // 校验是否更新正确
+        PayMerchantDO merchant = merchantMapper.selectById(reqVO.getId()); // 获取最新的
+        assertPojoEquals(reqVO, merchant);
+    }
+
+    @Test
+    public void testUpdateMerchant_notExists() {
+        // 准备参数
+        PayMerchantUpdateReqVO reqVO = randomPojo(PayMerchantUpdateReqVO.class);
+
+        // 调用, 并断言异常
+        assertServiceException(() -> merchantService.updateMerchant(reqVO), MERCHANT_NOT_EXISTS);
+    }
+
+    @Test
+    public void testDeleteMerchant_success() {
+        // mock 数据
+        PayMerchantDO dbMerchant = randomPojo(PayMerchantDO.class,
+                o-> o.setStatus(CommonStatusEnum.ENABLE.getStatus()));
+        merchantMapper.insert(dbMerchant);// @Sql: 先插入出一条存在的数据
+        // 准备参数
+        Long id = dbMerchant.getId();
+
+        // 调用
+        merchantService.deleteMerchant(id);
+       // 校验数据不存在了
+       assertNull(merchantMapper.selectById(id));
+    }
+
+    @Test
+    public void testDeleteMerchant_notExists() {
+        // 准备参数
+        Long id = randomLongId();
+
+        // 调用, 并断言异常
+        assertServiceException(() -> merchantService.deleteMerchant(id), MERCHANT_NOT_EXISTS);
+    }
+
+    @Test
+    public void testGetMerchantPage() {
+       // mock 数据
+       PayMerchantDO dbMerchant = randomPojo(PayMerchantDO.class, o -> { // 等会查询到
+           o.setNo("M1008611");
+           o.setName("灿哥的杂货铺");
+           o.setShortName("灿灿子");
+           o.setStatus(CommonStatusEnum.ENABLE.getStatus());
+           o.setRemark("灿哥的杂货铺");
+           o.setCreateTime(buildTime(2021,11,3));
+       });
+       merchantMapper.insert(dbMerchant);
+       // 测试 no 不匹配
+       merchantMapper.insert(ObjectUtils.clone(dbMerchant, o -> o.setNo("M200000")));
+       // 测试 name 不匹配
+       merchantMapper.insert(ObjectUtils.clone(dbMerchant, o -> o.setName("斌哥的杂货铺")));
+       // 测试 shortName 不匹配
+       merchantMapper.insert(ObjectUtils.clone(dbMerchant, o -> o.setShortName("斌斌子")));
+       // 测试 status 不匹配
+       merchantMapper.insert(ObjectUtils.clone(dbMerchant, o -> o.setStatus(CommonStatusEnum.DISABLE.getStatus())));
+       // 测试 remark 不匹配
+       merchantMapper.insert(ObjectUtils.clone(dbMerchant, o -> o.setRemark("斌哥的杂货铺")));
+       // 测试 createTime 不匹配
+       merchantMapper.insert(ObjectUtils.clone(dbMerchant, o -> o.setCreateTime(buildTime(2022,12,4))));
+       // 准备参数
+       PayMerchantPageReqVO reqVO = new PayMerchantPageReqVO();
+       reqVO.setNo("M1008611");
+       reqVO.setName("灿哥的杂货铺");
+       reqVO.setShortName("灿灿子");
+       reqVO.setStatus(CommonStatusEnum.ENABLE.getStatus());
+       reqVO.setRemark("灿哥的杂货铺");
+       reqVO.setBeginCreateTime(buildTime(2021,11,2));
+       reqVO.setEndCreateTime(buildTime(2021,11,4));
+
+       // 调用
+       PageResult<PayMerchantDO> pageResult = merchantService.getMerchantPage(reqVO);
+       // 断言
+       assertEquals(1, pageResult.getTotal());
+       assertEquals(1, pageResult.getList().size());
+       assertPojoEquals(dbMerchant, pageResult.getList().get(0));
+    }
+
+    @Test
+    public void testGetMerchantList() {
+       // mock 数据
+       PayMerchantDO dbMerchant = randomPojo(PayMerchantDO.class, o -> { // 等会查询到
+           o.setNo("M1008611");
+           o.setName("灿哥的杂货铺");
+           o.setShortName("灿灿子");
+           o.setStatus(CommonStatusEnum.ENABLE.getStatus());
+           o.setRemark("灿哥的杂货铺");
+           o.setCreateTime(buildTime(2021,11,3));
+       });
+        merchantMapper.insert(dbMerchant);
+        // 测试 no 不匹配
+        merchantMapper.insert(ObjectUtils.clone(dbMerchant, o -> o.setNo("M200000")));
+        // 测试 name 不匹配
+        merchantMapper.insert(ObjectUtils.clone(dbMerchant, o -> o.setName("斌哥的杂货铺")));
+        // 测试 shortName 不匹配
+        merchantMapper.insert(ObjectUtils.clone(dbMerchant, o -> o.setShortName("斌斌子")));
+        // 测试 status 不匹配
+        merchantMapper.insert(ObjectUtils.clone(dbMerchant, o -> o.setStatus(CommonStatusEnum.DISABLE.getStatus())));
+        // 测试 remark 不匹配
+        merchantMapper.insert(ObjectUtils.clone(dbMerchant, o -> o.setRemark("斌哥的杂货铺")));
+        // 测试 createTime 不匹配
+        merchantMapper.insert(ObjectUtils.clone(dbMerchant, o -> o.setCreateTime(buildTime(2022,12,4))));
+       // 准备参数
+       PayMerchantExportReqVO reqVO = new PayMerchantExportReqVO();
+       reqVO.setNo("M1008611");
+       reqVO.setName("灿哥的杂货铺");
+       reqVO.setShortName("灿灿子");
+       reqVO.setStatus(CommonStatusEnum.ENABLE.getStatus());
+       reqVO.setRemark("灿哥的杂货铺");
+       reqVO.setBeginCreateTime(buildTime(2021,11,2));
+       reqVO.setEndCreateTime(buildTime(2021,11,4));
+
+       // 调用
+       List<PayMerchantDO> list = merchantService.getMerchantList(reqVO);
+       // 断言
+       assertEquals(1, list.size());
+       assertPojoEquals(dbMerchant, list.get(0));
+    }
+
+}
--- a/yudao-admin-server/src/test/resources/sql/clean.sql
+++ b/yudao-admin-server/src/test/resources/sql/clean.sql
@ -24,3 +24,6 @@ DELETE FROM "sys_sms_template";
 DELETE FROM "sys_sms_log";
 DELETE FROM "sys_error_code";
 DELETE FROM "sys_social_user";
+
+-- pay 开头的 DB
+DELETE FROM pay_merchant;
--- a/yudao-admin-server/src/test/resources/sql/create_tables.sql
+++ b/yudao-admin-server/src/test/resources/sql/create_tables.sql
@ -449,3 +449,18 @@ CREATE TABLE IF NOT EXISTS "sys_social_user" (
   "deleted" bit NOT NULL DEFAULT FALSE,
   PRIMARY KEY ("id")
 ) COMMENT '社交用户';
+
+CREATE TABLE IF NOT EXISTS "pay_merchant" (
+    "id" number NOT NULL GENERATED BY DEFAULT AS IDENTITY,
+    "no" varchar(32) NOT NULL,
+    "name" varchar(64) NOT NULL,
+    "short_name" varchar(64) NOT NULL,
+    "status" tinyint NOT NULL,
+    "remark" varchar(255) DEFAULT NULL,
+    "creator" varchar(64) DEFAULT '',
+    "create_time" datetime NOT NULL DEFAULT CURRENT_TIMESTAMP ,
+    "updater" varchar(64) DEFAULT '',
+    "update_time" datetime NOT NULL DEFAULT CURRENT_TIMESTAMP ON UPDATE CURRENT_TIMESTAMP,
+    "deleted" bit(1) NOT NULL DEFAULT FALSE,
+    PRIMARY KEY ("id")
+)  COMMENT '支付商户信息';