mirror of
				https://gitee.com/hhyykk/ipms-sjy.git
				synced 2025-11-04 12:18:42 +08:00 
			
		
		
		
	@@ -0,0 +1,264 @@
 | 
			
		||||
package cn.iocoder.yudao.module.member.service.level;
 | 
			
		||||
 | 
			
		||||
import cn.iocoder.yudao.framework.common.enums.CommonStatusEnum;
 | 
			
		||||
import cn.iocoder.yudao.framework.common.pojo.PageResult;
 | 
			
		||||
import cn.iocoder.yudao.framework.common.util.collection.ArrayUtils;
 | 
			
		||||
import cn.iocoder.yudao.framework.test.core.ut.BaseDbUnitTest;
 | 
			
		||||
import cn.iocoder.yudao.module.member.controller.admin.level.vo.MemberLevelCreateReqVO;
 | 
			
		||||
import cn.iocoder.yudao.module.member.controller.admin.level.vo.MemberLevelPageReqVO;
 | 
			
		||||
import cn.iocoder.yudao.module.member.controller.admin.level.vo.MemberLevelUpdateReqVO;
 | 
			
		||||
import cn.iocoder.yudao.module.member.dal.dataobject.level.MemberLevelDO;
 | 
			
		||||
import cn.iocoder.yudao.module.member.dal.mysql.level.MemberLevelMapper;
 | 
			
		||||
import org.junit.jupiter.api.Test;
 | 
			
		||||
import org.springframework.boot.test.mock.mockito.MockBean;
 | 
			
		||||
import org.springframework.context.annotation.Import;
 | 
			
		||||
 | 
			
		||||
import javax.annotation.Resource;
 | 
			
		||||
import java.util.List;
 | 
			
		||||
import java.util.function.Consumer;
 | 
			
		||||
 | 
			
		||||
import static cn.hutool.core.util.RandomUtil.randomInt;
 | 
			
		||||
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.*;
 | 
			
		||||
import static cn.iocoder.yudao.module.member.enums.ErrorCodeConstants.*;
 | 
			
		||||
import static org.junit.jupiter.api.Assertions.*;
 | 
			
		||||
 | 
			
		||||
/**
 | 
			
		||||
 * {@link MemberLevelServiceImpl} 的单元测试类
 | 
			
		||||
 *
 | 
			
		||||
 * @author owen
 | 
			
		||||
 */
 | 
			
		||||
@Import(MemberLevelServiceImpl.class)
 | 
			
		||||
public class MemberLevelServiceImplTest extends BaseDbUnitTest {
 | 
			
		||||
 | 
			
		||||
    @Resource
 | 
			
		||||
    private MemberLevelServiceImpl levelService;
 | 
			
		||||
 | 
			
		||||
    @Resource
 | 
			
		||||
    private MemberLevelMapper levelMapper;
 | 
			
		||||
 | 
			
		||||
    @MockBean
 | 
			
		||||
    private MemberLevelLogService memberLevelLogService;
 | 
			
		||||
    @MockBean
 | 
			
		||||
    private MemberExperienceLogService memberExperienceLogService;
 | 
			
		||||
 | 
			
		||||
    @Test
 | 
			
		||||
    public void testCreateLevel_success() {
 | 
			
		||||
        // 准备参数
 | 
			
		||||
        MemberLevelCreateReqVO reqVO = randomPojo(MemberLevelCreateReqVO.class, o -> {
 | 
			
		||||
            o.setDiscount(randomInt());
 | 
			
		||||
            o.setIcon(randomURL());
 | 
			
		||||
            o.setBackgroundUrl(randomURL());
 | 
			
		||||
        });
 | 
			
		||||
 | 
			
		||||
        // 调用
 | 
			
		||||
        Long levelId = levelService.createLevel(reqVO);
 | 
			
		||||
        // 断言
 | 
			
		||||
        assertNotNull(levelId);
 | 
			
		||||
        // 校验记录的属性是否正确
 | 
			
		||||
        MemberLevelDO level = levelMapper.selectById(levelId);
 | 
			
		||||
        assertPojoEquals(reqVO, level);
 | 
			
		||||
    }
 | 
			
		||||
 | 
			
		||||
    @Test
 | 
			
		||||
    public void testUpdateLevel_success() {
 | 
			
		||||
        // mock 数据
 | 
			
		||||
        MemberLevelDO dbLevel = randomPojo(MemberLevelDO.class);
 | 
			
		||||
        levelMapper.insert(dbLevel);// @Sql: 先插入出一条存在的数据
 | 
			
		||||
        // 准备参数
 | 
			
		||||
        MemberLevelUpdateReqVO reqVO = randomPojo(MemberLevelUpdateReqVO.class, o -> {
 | 
			
		||||
            o.setId(dbLevel.getId()); // 设置更新的 ID
 | 
			
		||||
            //以下要保持一致
 | 
			
		||||
            o.setName(dbLevel.getName());
 | 
			
		||||
            o.setLevel(dbLevel.getLevel());
 | 
			
		||||
            o.setExperience(dbLevel.getExperience());
 | 
			
		||||
            //以下是要修改的字段
 | 
			
		||||
            o.setDiscount(randomInt());
 | 
			
		||||
            o.setIcon(randomURL());
 | 
			
		||||
            o.setBackgroundUrl(randomURL());
 | 
			
		||||
        });
 | 
			
		||||
 | 
			
		||||
        // 调用
 | 
			
		||||
        levelService.updateLevel(reqVO);
 | 
			
		||||
        // 校验是否更新正确
 | 
			
		||||
        MemberLevelDO level = levelMapper.selectById(reqVO.getId()); // 获取最新的
 | 
			
		||||
        assertPojoEquals(reqVO, level);
 | 
			
		||||
    }
 | 
			
		||||
 | 
			
		||||
    @Test
 | 
			
		||||
    public void testUpdateLevel_notExists() {
 | 
			
		||||
        // 准备参数
 | 
			
		||||
        MemberLevelUpdateReqVO reqVO = randomPojo(MemberLevelUpdateReqVO.class);
 | 
			
		||||
 | 
			
		||||
        // 调用, 并断言异常
 | 
			
		||||
        assertServiceException(() -> levelService.updateLevel(reqVO), LEVEL_NOT_EXISTS);
 | 
			
		||||
    }
 | 
			
		||||
 | 
			
		||||
    @Test
 | 
			
		||||
    public void testDeleteLevel_success() {
 | 
			
		||||
        // mock 数据
 | 
			
		||||
        MemberLevelDO dbLevel = randomPojo(MemberLevelDO.class);
 | 
			
		||||
        levelMapper.insert(dbLevel);// @Sql: 先插入出一条存在的数据
 | 
			
		||||
        // 准备参数
 | 
			
		||||
        Long id = dbLevel.getId();
 | 
			
		||||
 | 
			
		||||
        // 调用
 | 
			
		||||
        levelService.deleteLevel(id);
 | 
			
		||||
        // 校验数据不存在了
 | 
			
		||||
        assertNull(levelMapper.selectById(id));
 | 
			
		||||
    }
 | 
			
		||||
 | 
			
		||||
    @Test
 | 
			
		||||
    public void testDeleteLevel_notExists() {
 | 
			
		||||
        // 准备参数
 | 
			
		||||
        Long id = randomLongId();
 | 
			
		||||
 | 
			
		||||
        // 调用, 并断言异常
 | 
			
		||||
        assertServiceException(() -> levelService.deleteLevel(id), LEVEL_NOT_EXISTS);
 | 
			
		||||
    }
 | 
			
		||||
 | 
			
		||||
    @Test
 | 
			
		||||
    public void testGetLevelPage() {
 | 
			
		||||
        // mock 数据
 | 
			
		||||
        MemberLevelDO dbLevel = randomPojo(MemberLevelDO.class, o -> { // 等会查询到
 | 
			
		||||
            o.setName("黄金会员");
 | 
			
		||||
            o.setStatus(1);
 | 
			
		||||
        });
 | 
			
		||||
        levelMapper.insert(dbLevel);
 | 
			
		||||
        // 测试 name 不匹配
 | 
			
		||||
        levelMapper.insert(cloneIgnoreId(dbLevel, o -> o.setName("")));
 | 
			
		||||
        // 测试 status 不匹配
 | 
			
		||||
        levelMapper.insert(cloneIgnoreId(dbLevel, o -> o.setStatus(0)));
 | 
			
		||||
        // 准备参数
 | 
			
		||||
        MemberLevelPageReqVO reqVO = new MemberLevelPageReqVO();
 | 
			
		||||
        reqVO.setName("黄金会员");
 | 
			
		||||
        reqVO.setStatus(1);
 | 
			
		||||
 | 
			
		||||
        // 调用
 | 
			
		||||
        PageResult<MemberLevelDO> pageResult = levelService.getLevelPage(reqVO);
 | 
			
		||||
        // 断言
 | 
			
		||||
        assertEquals(1, pageResult.getTotal());
 | 
			
		||||
        assertEquals(1, pageResult.getList().size());
 | 
			
		||||
        assertPojoEquals(dbLevel, pageResult.getList().get(0));
 | 
			
		||||
    }
 | 
			
		||||
 | 
			
		||||
    @Test
 | 
			
		||||
    public void testCreateLevel_nameUnique() {
 | 
			
		||||
        // 准备参数
 | 
			
		||||
        String name = randomString();
 | 
			
		||||
 | 
			
		||||
        // mock 数据
 | 
			
		||||
        levelMapper.insert(randomLevelDO(o -> o.setName(name)));
 | 
			
		||||
 | 
			
		||||
        // 调用,校验异常
 | 
			
		||||
        List<MemberLevelDO> list = levelMapper.selectList();
 | 
			
		||||
        assertServiceException(() -> levelService.validateNameUnique(list, null, name), LEVEL_NAME_EXISTS, name);
 | 
			
		||||
    }
 | 
			
		||||
 | 
			
		||||
    @Test
 | 
			
		||||
    public void testUpdateLevel_nameUnique() {
 | 
			
		||||
        // 准备参数
 | 
			
		||||
        Long id = randomLongId();
 | 
			
		||||
        String name = randomString();
 | 
			
		||||
 | 
			
		||||
        // mock 数据
 | 
			
		||||
        levelMapper.insert(randomLevelDO(o -> o.setName(name)));
 | 
			
		||||
 | 
			
		||||
        // 调用,校验异常
 | 
			
		||||
        List<MemberLevelDO> list = levelMapper.selectList();
 | 
			
		||||
        assertServiceException(() -> levelService.validateNameUnique(list, id, name), LEVEL_NAME_EXISTS, name);
 | 
			
		||||
    }
 | 
			
		||||
 | 
			
		||||
    @Test
 | 
			
		||||
    public void testCreateLevel_levelUnique() {
 | 
			
		||||
        // 准备参数
 | 
			
		||||
        Integer level = randomInteger();
 | 
			
		||||
        String name = randomString();
 | 
			
		||||
 | 
			
		||||
        // mock 数据
 | 
			
		||||
        levelMapper.insert(randomLevelDO(o -> {
 | 
			
		||||
            o.setLevel(level);
 | 
			
		||||
            o.setName(name);
 | 
			
		||||
        }));
 | 
			
		||||
 | 
			
		||||
        // 调用,校验异常
 | 
			
		||||
        List<MemberLevelDO> list = levelMapper.selectList();
 | 
			
		||||
        assertServiceException(() -> levelService.validateLevelUnique(list, null, level), LEVEL_VALUE_EXISTS, level, name);
 | 
			
		||||
    }
 | 
			
		||||
 | 
			
		||||
    @Test
 | 
			
		||||
    public void testUpdateLevel_levelUnique() {
 | 
			
		||||
        // 准备参数
 | 
			
		||||
        Long id = randomLongId();
 | 
			
		||||
        Integer level = randomInteger();
 | 
			
		||||
        String name = randomString();
 | 
			
		||||
 | 
			
		||||
        // mock 数据
 | 
			
		||||
        levelMapper.insert(randomLevelDO(o -> {
 | 
			
		||||
            o.setLevel(level);
 | 
			
		||||
            o.setName(name);
 | 
			
		||||
        }));
 | 
			
		||||
 | 
			
		||||
        // 调用,校验异常
 | 
			
		||||
        List<MemberLevelDO> list = levelMapper.selectList();
 | 
			
		||||
        assertServiceException(() -> levelService.validateLevelUnique(list, id, level), LEVEL_VALUE_EXISTS, level, name);
 | 
			
		||||
    }
 | 
			
		||||
 | 
			
		||||
    @Test
 | 
			
		||||
    public void testCreateLevel_experienceOutRange() {
 | 
			
		||||
        // 准备参数
 | 
			
		||||
        int level = 10;
 | 
			
		||||
        int experience = 10;
 | 
			
		||||
        String name = randomString();
 | 
			
		||||
 | 
			
		||||
        // mock 数据
 | 
			
		||||
        levelMapper.insert(randomLevelDO(o -> {
 | 
			
		||||
            o.setLevel(level);
 | 
			
		||||
            o.setExperience(experience);
 | 
			
		||||
            o.setName(name);
 | 
			
		||||
        }));
 | 
			
		||||
        List<MemberLevelDO> list = levelMapper.selectList();
 | 
			
		||||
 | 
			
		||||
        // 调用,校验异常
 | 
			
		||||
        assertServiceException(() -> levelService.validateExperienceOutRange(list, null, level + 1, experience - 1), LEVEL_EXPERIENCE_MIN, name, level);
 | 
			
		||||
        // 调用,校验异常
 | 
			
		||||
        assertServiceException(() -> levelService.validateExperienceOutRange(list, null, level - 1, experience + 1), LEVEL_EXPERIENCE_MAX, name, level);
 | 
			
		||||
    }
 | 
			
		||||
 | 
			
		||||
    @Test
 | 
			
		||||
    public void testUpdateLevel_experienceOutRange() {
 | 
			
		||||
        // 准备参数
 | 
			
		||||
        int level = 10;
 | 
			
		||||
        int experience = 10;
 | 
			
		||||
        Long id = randomLongId();
 | 
			
		||||
        String name = randomString();
 | 
			
		||||
 | 
			
		||||
        // mock 数据
 | 
			
		||||
        levelMapper.insert(randomLevelDO(o -> {
 | 
			
		||||
            o.setLevel(level);
 | 
			
		||||
            o.setExperience(experience);
 | 
			
		||||
            o.setName(name);
 | 
			
		||||
        }));
 | 
			
		||||
        List<MemberLevelDO> list = levelMapper.selectList();
 | 
			
		||||
 | 
			
		||||
        // 调用,校验异常
 | 
			
		||||
        assertServiceException(() -> levelService.validateExperienceOutRange(list, id, level + 1, experience - 1), LEVEL_EXPERIENCE_MIN, name, level);
 | 
			
		||||
        // 调用,校验异常
 | 
			
		||||
        assertServiceException(() -> levelService.validateExperienceOutRange(list, id, level - 1, experience + 1), LEVEL_EXPERIENCE_MAX, name, level);
 | 
			
		||||
    }
 | 
			
		||||
 | 
			
		||||
    // ========== 随机对象 ==========
 | 
			
		||||
 | 
			
		||||
    @SafeVarargs
 | 
			
		||||
    private static MemberLevelDO randomLevelDO(Consumer<MemberLevelDO>... consumers) {
 | 
			
		||||
        Consumer<MemberLevelDO> consumer = (o) -> {
 | 
			
		||||
            o.setStatus(CommonStatusEnum.ENABLE.getStatus());
 | 
			
		||||
            o.setDiscount(randomInt(0, 100));
 | 
			
		||||
            o.setIcon(randomURL());
 | 
			
		||||
            o.setBackgroundUrl(randomURL());
 | 
			
		||||
        };
 | 
			
		||||
        return randomPojo(MemberLevelDO.class, ArrayUtils.append(consumer, consumers));
 | 
			
		||||
    }
 | 
			
		||||
}
 | 
			
		||||
@@ -1,3 +1,4 @@
 | 
			
		||||
DELETE FROM "member_user";
 | 
			
		||||
DELETE FROM "member_address";
 | 
			
		||||
DELETE FROM "member_tag";
 | 
			
		||||
DELETE FROM "member_tag";
 | 
			
		||||
DELETE FROM "member_level";
 | 
			
		||||
@@ -44,4 +44,23 @@ CREATE TABLE IF NOT EXISTS "member_tag"
 | 
			
		||||
    "deleted"     bit      NOT NULL DEFAULT FALSE,
 | 
			
		||||
    "tenant_id"   bigint   NOT NULL default '0',
 | 
			
		||||
    PRIMARY KEY ("id")
 | 
			
		||||
) COMMENT '会员标签';
 | 
			
		||||
) COMMENT '会员标签';
 | 
			
		||||
 | 
			
		||||
CREATE TABLE IF NOT EXISTS "member_level"
 | 
			
		||||
(
 | 
			
		||||
    "id"             bigint   NOT NULL GENERATED BY DEFAULT AS IDENTITY,
 | 
			
		||||
    "name"           varchar  NOT NULL,
 | 
			
		||||
    "experience"     int      NOT NULL,
 | 
			
		||||
    "level"          int      NOT NULL,
 | 
			
		||||
    "discount"       int      NOT NULL,
 | 
			
		||||
    "icon"           varchar  NOT NULL,
 | 
			
		||||
    "background_url" varchar  NOT NULL,
 | 
			
		||||
    "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 default '0',
 | 
			
		||||
    "status"         int      NOT NULL,
 | 
			
		||||
    PRIMARY KEY ("id")
 | 
			
		||||
) COMMENT '会员等级';
 | 
			
		||||
		Reference in New Issue
	
	Block a user