完善 LoginLogServiceImplTest 单元测试

YunaiV
1 parent 9ca79391
Showing 2 changed files with 47 additions and 79 deletions
yudao-framework/yudao-spring-boot-starter-test/src/main/java/cn/iocoder/yudao/framework/test/core/util/RandomUtils.java
yudao-module-system/yudao-module-system-biz/src/test/java/cn/iocoder/yudao/module/system/service/logger/LoginLogServiceImplTest.java
@@ -47,7 +47,7 @@ public class RandomUtils {
             }
             // 如果是 type、status 结尾的字段，返回 tinyint 范围
             if (StrUtil.endWithAnyIgnoreCase(attributeMetadata.getAttributeName(),
-                    "type", "status", "category", "scope")) {
+                    "type", "status", "category", "scope", "result")) {
                 return RandomUtil.randomInt(0, TINYINT_MAX + 1);
             }
             return RandomUtil.randomInt();
 package cn.iocoder.yudao.module.system.service.logger;
-import cn.hutool.core.util.RandomUtil;
-import cn.iocoder.yudao.framework.common.enums.UserTypeEnum;
 import cn.iocoder.yudao.framework.common.pojo.PageResult;
-import cn.iocoder.yudao.framework.common.util.monitor.TracerUtils;
-import cn.iocoder.yudao.framework.common.util.object.ObjectUtils;
 import cn.iocoder.yudao.framework.test.core.ut.BaseDbUnitTest;
-import cn.iocoder.yudao.framework.test.core.util.RandomUtils;
 import cn.iocoder.yudao.module.system.api.logger.dto.LoginLogCreateReqDTO;
 import cn.iocoder.yudao.module.system.controller.admin.logger.vo.loginlog.LoginLogExportReqVO;
 import cn.iocoder.yudao.module.system.controller.admin.logger.vo.loginlog.LoginLogPageReqVO;
 import cn.iocoder.yudao.module.system.dal.dataobject.logger.LoginLogDO;
 import cn.iocoder.yudao.module.system.dal.mysql.logger.LoginLogMapper;
-import cn.iocoder.yudao.module.system.enums.logger.LoginLogTypeEnum;
 import org.junit.jupiter.api.Test;
 import org.springframework.context.annotation.Import;
 import javax.annotation.Resource;
-import java.time.LocalDateTime;
 import java.util.List;
-import static cn.hutool.core.util.RandomUtil.randomEle;
+import static cn.iocoder.yudao.framework.common.util.date.LocalDateTimeUtils.buildBetweenTime;
 import static cn.iocoder.yudao.framework.common.util.date.LocalDateTimeUtils.buildTime;
+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.module.system.enums.logger.LoginResultEnum.*;
+import static cn.iocoder.yudao.framework.test.core.util.RandomUtils.randomPojo;
+import static cn.iocoder.yudao.module.system.enums.logger.LoginResultEnum.CAPTCHA_CODE_ERROR;
+import static cn.iocoder.yudao.module.system.enums.logger.LoginResultEnum.SUCCESS;
 import static org.junit.jupiter.api.Assertions.assertEquals;
 @Import(LoginLogServiceImpl.class)
@@ -37,42 +33,31 @@ public class LoginLogServiceImplTest extends BaseDbUnitTest {
     @Test
     public void testGetLoginLogPage() {
-        // 构造测试数据
-        // 登录成功的
-        LoginLogDO loginLogDO = RandomUtils.randomPojo(LoginLogDO.class, logDO -> {
-            logDO.setLogType(RandomUtil.randomEle(cn.iocoder.yudao.module.system.enums.logger.LoginLogTypeEnum.values()).getType());
-            logDO.setTraceId(TracerUtils.getTraceId());
-            logDO.setUserType(RandomUtil.randomEle(UserTypeEnum.values()).getValue());
-
-            logDO.setUserIp("192.168.199.16");
-            logDO.setUsername("wangkai");
-            logDO.setCreateTime(buildTime(2021, 3, 6));
-            logDO.setResult(SUCCESS.getResult());
+        // mock 数据
+        LoginLogDO loginLogDO = randomPojo(LoginLogDO.class, o -> {
+            o.setUserIp("192.168.199.16");
+            o.setUsername("wang");
+            o.setResult(SUCCESS.getResult());
+            o.setCreateTime(buildTime(2021, 3, 6));
         });
         loginLogMapper.insert(loginLogDO);
-
-        // 下面几个都是不匹配的数据
-        // 登录失败的
-        loginLogMapper.insert(ObjectUtils.cloneIgnoreId(loginLogDO, logDO -> logDO.setResult(CAPTCHA_CODE_ERROR.getResult())));
-        // 不同ip段的
-        loginLogMapper.insert(ObjectUtils.cloneIgnoreId(loginLogDO, logDO -> logDO.setUserIp("192.168.128.18")));
-        // 不同username
-        loginLogMapper.insert(ObjectUtils.cloneIgnoreId(loginLogDO, logDO -> logDO.setUsername("yunai")));
-        // 构造一个早期时间 2021-02-06 00:00:00
-        loginLogMapper.insert(ObjectUtils.cloneIgnoreId(loginLogDO, logDO -> logDO.setCreateTime(buildTime(2021, 2, 6))));
-
-
+        // 测试 status 不匹配
+        loginLogMapper.insert(cloneIgnoreId(loginLogDO, o -> o.setResult(CAPTCHA_CODE_ERROR.getResult())));
+        // 测试 ip 不匹配
+        loginLogMapper.insert(cloneIgnoreId(loginLogDO, o -> o.setUserIp("192.168.128.18")));
+        // 测试 username 不匹配
+        loginLogMapper.insert(cloneIgnoreId(loginLogDO, o -> o.setUsername("yunai")));
+        // 测试 createTime 不匹配
+        loginLogMapper.insert(cloneIgnoreId(loginLogDO, o -> o.setCreateTime(buildTime(2021, 2, 6))));
         // 构造调用参数
         LoginLogPageReqVO reqVO = new LoginLogPageReqVO();
-        reqVO.setUsername("wangkai");
+        reqVO.setUsername("wang");
         reqVO.setUserIp("192.168.199");
         reqVO.setStatus(true);
-        reqVO.setCreateTime((new LocalDateTime[]{buildTime(2021, 3, 5),
-                buildTime(2021, 3, 7)}));
+        reqVO.setCreateTime(buildBetweenTime(2021, 3, 5, 2021, 3, 7));
-        // 调用service方法
+        // 调用
         PageResult<LoginLogDO> pageResult = loginLogService.getLoginLogPage(reqVO);
-
         // 断言，只查到了一条符合条件的
         assertEquals(1, pageResult.getTotal());
         assertEquals(1, pageResult.getList().size());
@@ -81,62 +66,45 @@ public class LoginLogServiceImplTest extends BaseDbUnitTest {
     @Test
     public void testGetLoginLogList() {
-        // 构造测试数据
-
-        // 登录成功的
-        LoginLogDO loginLogDO = RandomUtils.randomPojo(LoginLogDO.class, logDO -> {
-            logDO.setLogType(RandomUtil.randomEle(cn.iocoder.yudao.module.system.enums.logger.LoginLogTypeEnum.values()).getType());
-            logDO.setTraceId(TracerUtils.getTraceId());
-            logDO.setUserType(RandomUtil.randomEle(UserTypeEnum.values()).getValue());
-
-            logDO.setUserIp("192.168.111.16");
-            logDO.setUsername("wangxiaokai");
-            logDO.setCreateTime(buildTime(2021, 3, 6));
-            logDO.setResult(SUCCESS.getResult());
+        // mock 数据
+        LoginLogDO loginLogDO = randomPojo(LoginLogDO.class, o -> {
+            o.setUserIp("192.168.199.16");
+            o.setUsername("wang");
+            o.setResult(SUCCESS.getResult());
+            o.setCreateTime(buildTime(2021, 3, 6));
         });
         loginLogMapper.insert(loginLogDO);
-
-        // 下面几个都是不匹配的数据
-        // 登录失败的
-        loginLogMapper.insert(ObjectUtils.cloneIgnoreId(loginLogDO, logDO -> logDO.setResult(CAPTCHA_CODE_ERROR.getResult())));
-        // 不同ip段的
-        loginLogMapper.insert(ObjectUtils.cloneIgnoreId(loginLogDO, logDO -> logDO.setUserIp("192.168.128.18")));
-        // 不同username
-        loginLogMapper.insert(ObjectUtils.cloneIgnoreId(loginLogDO, logDO -> logDO.setUsername("yunai")));
-        // 构造一个早期时间 2021-02-06 00:00:00
-        loginLogMapper.insert(ObjectUtils.cloneIgnoreId(loginLogDO, logDO -> logDO.setCreateTime(buildTime(2021, 2, 6))));
-
+        // 测试 status 不匹配
+        loginLogMapper.insert(cloneIgnoreId(loginLogDO, o -> o.setResult(CAPTCHA_CODE_ERROR.getResult())));
+        // 测试 ip 不匹配
+        loginLogMapper.insert(cloneIgnoreId(loginLogDO, o -> o.setUserIp("192.168.128.18")));
+        // 测试 username 不匹配
+        loginLogMapper.insert(cloneIgnoreId(loginLogDO, o -> o.setUsername("yunai")));
+        // 测试 createTime 不匹配
+        loginLogMapper.insert(cloneIgnoreId(loginLogDO, o -> o.setCreateTime(buildTime(2021, 2, 6))));
         // 构造调用参数
         LoginLogExportReqVO reqVO = new LoginLogExportReqVO();
-        reqVO.setUsername("wangxiaokai");
-        reqVO.setUserIp("192.168.111");
+        reqVO.setUsername("wang");
+        reqVO.setUserIp("192.168.199");
         reqVO.setStatus(true);
-        reqVO.setCreateTime((new LocalDateTime[]{buildTime(2021, 3, 5),
-                buildTime(2021, 3, 7)}));
+        reqVO.setCreateTime(buildBetweenTime(2021, 3, 5, 2021, 3, 7));
         // 调用service方法
-        List<LoginLogDO> loginLogList = loginLogService.getLoginLogList(reqVO);
-
+        List<LoginLogDO> list = loginLogService.getLoginLogList(reqVO);
         // 断言
-        assertEquals(1, loginLogList.size());
-        assertPojoEquals(loginLogDO, loginLogList.get(0));
+        assertEquals(1, list.size());
+        assertPojoEquals(loginLogDO, list.get(0));
     }
     @Test
     public void testCreateLoginLog() {
-        LoginLogCreateReqDTO reqDTO = RandomUtils.randomPojo(LoginLogCreateReqDTO.class, vo -> {
-            // 指定随机的范围,避免超出范围入库失败
-            vo.setUserType(randomEle(UserTypeEnum.values()).getValue());
-            vo.setLogType(randomEle(LoginLogTypeEnum.values()).getType());
-            vo.setResult(randomEle(values()).getResult());
-            vo.setTraceId(TracerUtils.getTraceId());
-        });
+        LoginLogCreateReqDTO reqDTO = randomPojo(LoginLogCreateReqDTO.class);
         // 调用
         loginLogService.createLoginLog(reqDTO);
-        // 断言，忽略基本字段
-        LoginLogDO sysLoginLogDO = loginLogMapper.selectOne(null);
-        assertPojoEquals(reqDTO, sysLoginLogDO);
+        // 断言
+        LoginLogDO loginLogDO = loginLogMapper.selectOne(null);
+        assertPojoEquals(reqDTO, loginLogDO);
     }
 }