|
@@ -0,0 +1,151 @@
|
|
|
+package com.citu.module.menduner.system.service.projectexp;
|
|
|
+
|
|
|
+import org.junit.jupiter.api.Disabled;
|
|
|
+import org.junit.jupiter.api.Test;
|
|
|
+import org.springframework.boot.test.mock.mockito.MockBean;
|
|
|
+
|
|
|
+import javax.annotation.Resource;
|
|
|
+
|
|
|
+import com.citu.framework.test.core.ut.BaseDbUnitTest;
|
|
|
+
|
|
|
+import com.citu.module.menduner.system.controller.admin.projectexp.vo.*;
|
|
|
+import com.citu.module.menduner.system.dal.dataobject.projectexp.ProjectExpDO;
|
|
|
+import com.citu.module.menduner.system.dal.mysql.projectexp.ProjectExpMapper;
|
|
|
+import com.citu.framework.common.pojo.PageResult;
|
|
|
+
|
|
|
+import javax.annotation.Resource;
|
|
|
+import org.springframework.context.annotation.Import;
|
|
|
+
|
|
|
+import java.time.LocalDate;
|
|
|
+import java.util.*;
|
|
|
+import java.time.LocalDateTime;
|
|
|
+
|
|
|
+import static cn.hutool.core.util.RandomUtil.*;
|
|
|
+import static com.citu.module.menduner.system.enums.ErrorCodeConstants.*;
|
|
|
+import static com.citu.framework.test.core.util.AssertUtils.*;
|
|
|
+import static com.citu.framework.test.core.util.RandomUtils.*;
|
|
|
+import static com.citu.framework.common.util.date.LocalDateTimeUtils.*;
|
|
|
+import static com.citu.framework.common.util.object.ObjectUtils.*;
|
|
|
+import static com.citu.framework.common.util.date.DateUtils.*;
|
|
|
+import static org.junit.jupiter.api.Assertions.*;
|
|
|
+import static org.mockito.Mockito.*;
|
|
|
+
|
|
|
+/**
|
|
|
+ * {@link ProjectExpServiceImpl} 的单元测试类
|
|
|
+ *
|
|
|
+ * @author Rayson
|
|
|
+ */
|
|
|
+@Import(ProjectExpServiceImpl.class)
|
|
|
+public class ProjectExpServiceImplTest extends BaseDbUnitTest {
|
|
|
+
|
|
|
+ @Resource
|
|
|
+ private ProjectExpServiceImpl projectExpService;
|
|
|
+
|
|
|
+ @Resource
|
|
|
+ private ProjectExpMapper projectExpMapper;
|
|
|
+
|
|
|
+ @Test
|
|
|
+ public void testCreateProjectExp_success() {
|
|
|
+ // 准备参数
|
|
|
+ ProjectExpSaveReqVO createReqVO = randomPojo(ProjectExpSaveReqVO.class).setId(null);
|
|
|
+
|
|
|
+ // 调用
|
|
|
+ Long projectExpId = projectExpService.createProjectExp(createReqVO);
|
|
|
+ // 断言
|
|
|
+ assertNotNull(projectExpId);
|
|
|
+ // 校验记录的属性是否正确
|
|
|
+ ProjectExpDO projectExp = projectExpMapper.selectById(projectExpId);
|
|
|
+ assertPojoEquals(createReqVO, projectExp, "id");
|
|
|
+ }
|
|
|
+
|
|
|
+ @Test
|
|
|
+ public void testUpdateProjectExp_success() {
|
|
|
+ // mock 数据
|
|
|
+ ProjectExpDO dbProjectExp = randomPojo(ProjectExpDO.class);
|
|
|
+ projectExpMapper.insert(dbProjectExp);// @Sql: 先插入出一条存在的数据
|
|
|
+ // 准备参数
|
|
|
+ ProjectExpSaveReqVO updateReqVO = randomPojo(ProjectExpSaveReqVO.class, o -> {
|
|
|
+ o.setId(dbProjectExp.getId()); // 设置更新的 ID
|
|
|
+ });
|
|
|
+
|
|
|
+ // 调用
|
|
|
+ projectExpService.updateProjectExp(updateReqVO);
|
|
|
+ // 校验是否更新正确
|
|
|
+ ProjectExpDO projectExp = projectExpMapper.selectById(updateReqVO.getId()); // 获取最新的
|
|
|
+ assertPojoEquals(updateReqVO, projectExp);
|
|
|
+ }
|
|
|
+
|
|
|
+ @Test
|
|
|
+ public void testUpdateProjectExp_notExists() {
|
|
|
+ // 准备参数
|
|
|
+ ProjectExpSaveReqVO updateReqVO = randomPojo(ProjectExpSaveReqVO.class);
|
|
|
+
|
|
|
+ // 调用, 并断言异常
|
|
|
+ assertServiceException(() -> projectExpService.updateProjectExp(updateReqVO), PROJECT_EXP_NOT_EXISTS);
|
|
|
+ }
|
|
|
+
|
|
|
+ @Test
|
|
|
+ public void testDeleteProjectExp_success() {
|
|
|
+ // mock 数据
|
|
|
+ ProjectExpDO dbProjectExp = randomPojo(ProjectExpDO.class);
|
|
|
+ projectExpMapper.insert(dbProjectExp);// @Sql: 先插入出一条存在的数据
|
|
|
+ // 准备参数
|
|
|
+ Long id = dbProjectExp.getId();
|
|
|
+
|
|
|
+ // 调用
|
|
|
+ projectExpService.deleteProjectExp(id);
|
|
|
+ // 校验数据不存在了
|
|
|
+ assertNull(projectExpMapper.selectById(id));
|
|
|
+ }
|
|
|
+
|
|
|
+ @Test
|
|
|
+ public void testDeleteProjectExp_notExists() {
|
|
|
+ // 准备参数
|
|
|
+ Long id = randomLongId();
|
|
|
+
|
|
|
+ // 调用, 并断言异常
|
|
|
+ assertServiceException(() -> projectExpService.deleteProjectExp(id), PROJECT_EXP_NOT_EXISTS);
|
|
|
+ }
|
|
|
+
|
|
|
+ @Test
|
|
|
+ @Disabled // TODO 请修改 null 为需要的值,然后删除 @Disabled 注解
|
|
|
+ public void testGetProjectExpPage() {
|
|
|
+ // mock 数据
|
|
|
+ ProjectExpDO dbProjectExp = randomPojo(ProjectExpDO.class, o -> { // 等会查询到
|
|
|
+ o.setUserId(null);
|
|
|
+ o.setName(null);
|
|
|
+ o.setStartTime(null);
|
|
|
+ o.setEndTime(null);
|
|
|
+ o.setCreateTime(null);
|
|
|
+ });
|
|
|
+ projectExpMapper.insert(dbProjectExp);
|
|
|
+ // 测试 userId 不匹配
|
|
|
+ projectExpMapper.insert(cloneIgnoreId(dbProjectExp, o -> o.setUserId(null)));
|
|
|
+ // 测试 name 不匹配
|
|
|
+ projectExpMapper.insert(cloneIgnoreId(dbProjectExp, o -> o.setName(null)));
|
|
|
+ // 测试 startTime 不匹配
|
|
|
+ projectExpMapper.insert(cloneIgnoreId(dbProjectExp, o -> o.setStartTime(null)));
|
|
|
+ // 测试 endTime 不匹配
|
|
|
+ projectExpMapper.insert(cloneIgnoreId(dbProjectExp, o -> o.setEndTime(null)));
|
|
|
+ // 测试 createTime 不匹配
|
|
|
+ projectExpMapper.insert(cloneIgnoreId(dbProjectExp, o -> o.setCreateTime(null)));
|
|
|
+ // 准备参数
|
|
|
+ ProjectExpPageReqVO reqVO = new ProjectExpPageReqVO();
|
|
|
+ reqVO.setUserId(null);
|
|
|
+ reqVO.setName(null);
|
|
|
+ reqVO.setStartTime(buildBetweenDate(2023, 2, 1, 2023, 2, 28));
|
|
|
+ reqVO.setEndTime(buildBetweenDate(2023, 2, 1, 2023, 2, 28));
|
|
|
+ reqVO.setCreateTime(buildBetweenTime(2023, 2, 1, 2023, 2, 28));
|
|
|
+
|
|
|
+ // 调用
|
|
|
+ PageResult<ProjectExpDO> pageResult = projectExpService.getProjectExpPage(reqVO);
|
|
|
+ // 断言
|
|
|
+ assertEquals(1, pageResult.getTotal());
|
|
|
+ assertEquals(1, pageResult.getList().size());
|
|
|
+ assertPojoEquals(dbProjectExp, pageResult.getList().get(0));
|
|
|
+ }
|
|
|
+ public static LocalDate[] buildBetweenDate(int year1, int mouth1, int day1,
|
|
|
+ int year2, int mouth2, int day2) {
|
|
|
+ return new LocalDate[]{LocalDate.of(year1, mouth1, day1), LocalDate.of(year2, mouth2, day2)};
|
|
|
+ }
|
|
|
+}
|