From 9e7575e9cb2f4686362eb53a58ced9a585b4cd76 Mon Sep 17 00:00:00 2001 From: hhyykk Date: Wed, 17 Jul 2024 19:47:32 +0800 Subject: [PATCH] =?UTF-8?q?[feat]=20=E5=A2=9E=E5=8A=A0=E9=A1=B9=E7=9B=AE?= =?UTF-8?q?=E4=B8=8E=E7=94=A8=E6=88=B7=E7=9A=84=E5=85=B3=E8=81=94=E8=A1=A8?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- .../module/pms/enums/ErrorCodeConstants.java | 6 +- .../projectuser/ProjectUserController.java | 95 +++++++++++++ .../projectuser/vo/ProjectUserPageReqVO.java | 18 +++ .../projectuser/vo/ProjectUserRespVO.java | 13 ++ .../projectuser/vo/ProjectUserSaveReqVO.java | 12 ++ .../dataobject/projectuser/ProjectUserDO.java | 47 +++++++ .../mysql/projectuser/ProjectUserMapper.java | 25 ++++ .../projectuser/ProjectUserService.java | 55 ++++++++ .../projectuser/ProjectUserServiceImpl.java | 74 ++++++++++ .../mapper/projectuser/ProjectUserMapper.xml | 12 ++ .../ProjectUserServiceImplTest.java | 126 ++++++++++++++++++ 11 files changed, 482 insertions(+), 1 deletion(-) create mode 100644 yudao-module-pms/yudao-module-pms-biz/src/main/java/cn/iocoder/yudao/module/pms/controller/admin/projectuser/ProjectUserController.java create mode 100644 yudao-module-pms/yudao-module-pms-biz/src/main/java/cn/iocoder/yudao/module/pms/controller/admin/projectuser/vo/ProjectUserPageReqVO.java create mode 100644 yudao-module-pms/yudao-module-pms-biz/src/main/java/cn/iocoder/yudao/module/pms/controller/admin/projectuser/vo/ProjectUserRespVO.java create mode 100644 yudao-module-pms/yudao-module-pms-biz/src/main/java/cn/iocoder/yudao/module/pms/controller/admin/projectuser/vo/ProjectUserSaveReqVO.java create mode 100644 yudao-module-pms/yudao-module-pms-biz/src/main/java/cn/iocoder/yudao/module/pms/dal/dataobject/projectuser/ProjectUserDO.java create mode 100644 yudao-module-pms/yudao-module-pms-biz/src/main/java/cn/iocoder/yudao/module/pms/dal/mysql/projectuser/ProjectUserMapper.java create mode 100644 yudao-module-pms/yudao-module-pms-biz/src/main/java/cn/iocoder/yudao/module/pms/service/projectuser/ProjectUserService.java create mode 100644 yudao-module-pms/yudao-module-pms-biz/src/main/java/cn/iocoder/yudao/module/pms/service/projectuser/ProjectUserServiceImpl.java create mode 100644 yudao-module-pms/yudao-module-pms-biz/src/main/resources/mapper/projectuser/ProjectUserMapper.xml create mode 100644 yudao-module-pms/yudao-module-pms-biz/src/test/java/cn/iocoder/yudao/module/pms/service/projectuser/ProjectUserServiceImplTest.java diff --git a/yudao-module-pms/yudao-module-pms-api/src/main/java/cn/iocoder/yudao/module/pms/enums/ErrorCodeConstants.java b/yudao-module-pms/yudao-module-pms-api/src/main/java/cn/iocoder/yudao/module/pms/enums/ErrorCodeConstants.java index 4a5214047..e1dea1fc9 100644 --- a/yudao-module-pms/yudao-module-pms-api/src/main/java/cn/iocoder/yudao/module/pms/enums/ErrorCodeConstants.java +++ b/yudao-module-pms/yudao-module-pms-api/src/main/java/cn/iocoder/yudao/module/pms/enums/ErrorCodeConstants.java @@ -4,9 +4,13 @@ import cn.iocoder.yudao.framework.common.exception.ErrorCode; /** * @author hhyykk - * @description 错误码枚举类 1_021_000_000 段 + * @description 错误码枚举类 * @date 2024/7/3 */ public interface ErrorCodeConstants { + // ============ 项目信息 ========= ErrorCode PROJECT_NOT_EXISTS = new ErrorCode(1_021_000_000, "项目信息不存在"); + + // ============ 项目与人员关联关系 ========= + ErrorCode PROJECT_USER_NOT_EXISTS = new ErrorCode(1_022_000_000, "项目与人员关联关系不存在"); } diff --git a/yudao-module-pms/yudao-module-pms-biz/src/main/java/cn/iocoder/yudao/module/pms/controller/admin/projectuser/ProjectUserController.java b/yudao-module-pms/yudao-module-pms-biz/src/main/java/cn/iocoder/yudao/module/pms/controller/admin/projectuser/ProjectUserController.java new file mode 100644 index 000000000..a6ef69bc9 --- /dev/null +++ b/yudao-module-pms/yudao-module-pms-biz/src/main/java/cn/iocoder/yudao/module/pms/controller/admin/projectuser/ProjectUserController.java @@ -0,0 +1,95 @@ +package cn.iocoder.yudao.module.pms.controller.admin.projectuser; + +import org.springframework.web.bind.annotation.*; +import jakarta.annotation.Resource; +import org.springframework.validation.annotation.Validated; +import org.springframework.security.access.prepost.PreAuthorize; +import io.swagger.v3.oas.annotations.tags.Tag; +import io.swagger.v3.oas.annotations.Parameter; +import io.swagger.v3.oas.annotations.Operation; + +import jakarta.validation.constraints.*; +import jakarta.validation.*; +import jakarta.servlet.http.*; +import java.util.*; +import java.io.IOException; + +import cn.iocoder.yudao.framework.common.pojo.PageParam; +import cn.iocoder.yudao.framework.common.pojo.PageResult; +import cn.iocoder.yudao.framework.common.pojo.CommonResult; +import cn.iocoder.yudao.framework.common.util.object.BeanUtils; +import static cn.iocoder.yudao.framework.common.pojo.CommonResult.success; + +import cn.iocoder.yudao.framework.excel.core.util.ExcelUtils; + +import cn.iocoder.yudao.framework.apilog.core.annotation.ApiAccessLog; +import static cn.iocoder.yudao.framework.apilog.core.enums.OperateTypeEnum.*; + +import cn.iocoder.yudao.module.pms.controller.admin.projectuser.vo.*; +import cn.iocoder.yudao.module.pms.dal.dataobject.projectuser.ProjectUserDO; +import cn.iocoder.yudao.module.pms.service.projectuser.ProjectUserService; + +@Tag(name = "管理后台 - 项目与人员关联关系") +@RestController +@RequestMapping("/pms/project-user") +@Validated +public class ProjectUserController { + + @Resource + private ProjectUserService projectUserService; + + @PostMapping("/create") + @Operation(summary = "创建项目与人员关联关系") + @PreAuthorize("@ss.hasPermission('pms:project-user:create')") + public CommonResult createProjectUser(@Valid @RequestBody ProjectUserSaveReqVO createReqVO) { + return success(projectUserService.createProjectUser(createReqVO)); + } + + @PutMapping("/update") + @Operation(summary = "更新项目与人员关联关系") + @PreAuthorize("@ss.hasPermission('pms:project-user:update')") + public CommonResult updateProjectUser(@Valid @RequestBody ProjectUserSaveReqVO updateReqVO) { + projectUserService.updateProjectUser(updateReqVO); + return success(true); + } + + @DeleteMapping("/delete") + @Operation(summary = "删除项目与人员关联关系") + @Parameter(name = "id", description = "编号", required = true) + @PreAuthorize("@ss.hasPermission('pms:project-user:delete')") + public CommonResult deleteProjectUser(@RequestParam("id") Long id) { + projectUserService.deleteProjectUser(id); + return success(true); + } + + @GetMapping("/get") + @Operation(summary = "获得项目与人员关联关系") + @Parameter(name = "id", description = "编号", required = true, example = "1024") + @PreAuthorize("@ss.hasPermission('pms:project-user:query')") + public CommonResult getProjectUser(@RequestParam("id") Long id) { + ProjectUserDO projectUser = projectUserService.getProjectUser(id); + return success(BeanUtils.toBean(projectUser, ProjectUserRespVO.class)); + } + + @GetMapping("/page") + @Operation(summary = "获得项目与人员关联关系分页") + @PreAuthorize("@ss.hasPermission('pms:project-user:query')") + public CommonResult> getProjectUserPage(@Valid ProjectUserPageReqVO pageReqVO) { + PageResult pageResult = projectUserService.getProjectUserPage(pageReqVO); + return success(BeanUtils.toBean(pageResult, ProjectUserRespVO.class)); + } + + @GetMapping("/export-excel") + @Operation(summary = "导出项目与人员关联关系 Excel") + @PreAuthorize("@ss.hasPermission('pms:project-user:export')") + @ApiAccessLog(operateType = EXPORT) + public void exportProjectUserExcel(@Valid ProjectUserPageReqVO pageReqVO, + HttpServletResponse response) throws IOException { + pageReqVO.setPageSize(PageParam.PAGE_SIZE_NONE); + List list = projectUserService.getProjectUserPage(pageReqVO).getList(); + // 导出 Excel + ExcelUtils.write(response, "项目与人员关联关系.xls", "数据", ProjectUserRespVO.class, + BeanUtils.toBean(list, ProjectUserRespVO.class)); + } + +} \ No newline at end of file diff --git a/yudao-module-pms/yudao-module-pms-biz/src/main/java/cn/iocoder/yudao/module/pms/controller/admin/projectuser/vo/ProjectUserPageReqVO.java b/yudao-module-pms/yudao-module-pms-biz/src/main/java/cn/iocoder/yudao/module/pms/controller/admin/projectuser/vo/ProjectUserPageReqVO.java new file mode 100644 index 000000000..0b5b4b5ed --- /dev/null +++ b/yudao-module-pms/yudao-module-pms-biz/src/main/java/cn/iocoder/yudao/module/pms/controller/admin/projectuser/vo/ProjectUserPageReqVO.java @@ -0,0 +1,18 @@ +package cn.iocoder.yudao.module.pms.controller.admin.projectuser.vo; + +import lombok.*; +import java.util.*; +import io.swagger.v3.oas.annotations.media.Schema; +import cn.iocoder.yudao.framework.common.pojo.PageParam; +import org.springframework.format.annotation.DateTimeFormat; +import java.time.LocalDateTime; + +import static cn.iocoder.yudao.framework.common.util.date.DateUtils.FORMAT_YEAR_MONTH_DAY_HOUR_MINUTE_SECOND; + +@Schema(description = "管理后台 - 项目与人员关联关系分页 Request VO") +@Data +@EqualsAndHashCode(callSuper = true) +@ToString(callSuper = true) +public class ProjectUserPageReqVO extends PageParam { + +} \ No newline at end of file diff --git a/yudao-module-pms/yudao-module-pms-biz/src/main/java/cn/iocoder/yudao/module/pms/controller/admin/projectuser/vo/ProjectUserRespVO.java b/yudao-module-pms/yudao-module-pms-biz/src/main/java/cn/iocoder/yudao/module/pms/controller/admin/projectuser/vo/ProjectUserRespVO.java new file mode 100644 index 000000000..6c26e691b --- /dev/null +++ b/yudao-module-pms/yudao-module-pms-biz/src/main/java/cn/iocoder/yudao/module/pms/controller/admin/projectuser/vo/ProjectUserRespVO.java @@ -0,0 +1,13 @@ +package cn.iocoder.yudao.module.pms.controller.admin.projectuser.vo; + +import io.swagger.v3.oas.annotations.media.Schema; +import lombok.*; +import java.util.*; +import com.alibaba.excel.annotation.*; + +@Schema(description = "管理后台 - 项目与人员关联关系 Response VO") +@Data +@ExcelIgnoreUnannotated +public class ProjectUserRespVO { + +} \ No newline at end of file diff --git a/yudao-module-pms/yudao-module-pms-biz/src/main/java/cn/iocoder/yudao/module/pms/controller/admin/projectuser/vo/ProjectUserSaveReqVO.java b/yudao-module-pms/yudao-module-pms-biz/src/main/java/cn/iocoder/yudao/module/pms/controller/admin/projectuser/vo/ProjectUserSaveReqVO.java new file mode 100644 index 000000000..d9fc6a4ff --- /dev/null +++ b/yudao-module-pms/yudao-module-pms-biz/src/main/java/cn/iocoder/yudao/module/pms/controller/admin/projectuser/vo/ProjectUserSaveReqVO.java @@ -0,0 +1,12 @@ +package cn.iocoder.yudao.module.pms.controller.admin.projectuser.vo; + +import io.swagger.v3.oas.annotations.media.Schema; +import lombok.*; +import java.util.*; +import jakarta.validation.constraints.*; + +@Schema(description = "管理后台 - 项目与人员关联关系新增/修改 Request VO") +@Data +public class ProjectUserSaveReqVO { + +} \ No newline at end of file diff --git a/yudao-module-pms/yudao-module-pms-biz/src/main/java/cn/iocoder/yudao/module/pms/dal/dataobject/projectuser/ProjectUserDO.java b/yudao-module-pms/yudao-module-pms-biz/src/main/java/cn/iocoder/yudao/module/pms/dal/dataobject/projectuser/ProjectUserDO.java new file mode 100644 index 000000000..81de1ff5e --- /dev/null +++ b/yudao-module-pms/yudao-module-pms-biz/src/main/java/cn/iocoder/yudao/module/pms/dal/dataobject/projectuser/ProjectUserDO.java @@ -0,0 +1,47 @@ +package cn.iocoder.yudao.module.pms.dal.dataobject.projectuser; + +import lombok.*; +import java.util.*; +import java.time.LocalDateTime; +import java.time.LocalDateTime; +import com.baomidou.mybatisplus.annotation.*; +import cn.iocoder.yudao.framework.mybatis.core.dataobject.BaseDO; + +/** + * 项目与人员关联关系 DO + * + * @author hhyykk + */ +@TableName("pms_project_user") +@KeySequence("pms_project_user_seq") // 用于 Oracle、PostgreSQL、Kingbase、DB2、H2 数据库的主键自增。如果是 MySQL 等数据库，可不写。 +@Data +@EqualsAndHashCode(callSuper = true) +@ToString(callSuper = true) +@Builder +@NoArgsConstructor +@AllArgsConstructor +public class ProjectUserDO extends BaseDO { + + /** + * 主键 + */ + @TableId + private Long id; + /** + * 项目id + */ + private Long projectId; + /** + * 用户id + */ + private Long userId; + /** + * 更新者 + */ + private String updator; + /** + * 是否删除 + */ + private Boolean delete; + +} \ No newline at end of file diff --git a/yudao-module-pms/yudao-module-pms-biz/src/main/java/cn/iocoder/yudao/module/pms/dal/mysql/projectuser/ProjectUserMapper.java b/yudao-module-pms/yudao-module-pms-biz/src/main/java/cn/iocoder/yudao/module/pms/dal/mysql/projectuser/ProjectUserMapper.java new file mode 100644 index 000000000..26e200e40 --- /dev/null +++ b/yudao-module-pms/yudao-module-pms-biz/src/main/java/cn/iocoder/yudao/module/pms/dal/mysql/projectuser/ProjectUserMapper.java @@ -0,0 +1,25 @@ +package cn.iocoder.yudao.module.pms.dal.mysql.projectuser; + +import java.util.*; + +import cn.iocoder.yudao.framework.common.pojo.PageResult; +import cn.iocoder.yudao.framework.mybatis.core.query.LambdaQueryWrapperX; +import cn.iocoder.yudao.framework.mybatis.core.mapper.BaseMapperX; +import cn.iocoder.yudao.module.pms.dal.dataobject.projectuser.ProjectUserDO; +import org.apache.ibatis.annotations.Mapper; +import cn.iocoder.yudao.module.pms.controller.admin.projectuser.vo.*; + +/** + * 项目与人员关联关系 Mapper + * + * @author hhyykk + */ +@Mapper +public interface ProjectUserMapper extends BaseMapperX { + + default PageResult selectPage(ProjectUserPageReqVO reqVO) { + return selectPage(reqVO, new LambdaQueryWrapperX() + .orderByDesc(ProjectUserDO::getId)); + } + +} \ No newline at end of file diff --git a/yudao-module-pms/yudao-module-pms-biz/src/main/java/cn/iocoder/yudao/module/pms/service/projectuser/ProjectUserService.java b/yudao-module-pms/yudao-module-pms-biz/src/main/java/cn/iocoder/yudao/module/pms/service/projectuser/ProjectUserService.java new file mode 100644 index 000000000..67dd92d92 --- /dev/null +++ b/yudao-module-pms/yudao-module-pms-biz/src/main/java/cn/iocoder/yudao/module/pms/service/projectuser/ProjectUserService.java @@ -0,0 +1,55 @@ +package cn.iocoder.yudao.module.pms.service.projectuser; + +import java.util.*; +import jakarta.validation.*; +import cn.iocoder.yudao.module.pms.controller.admin.projectuser.vo.*; +import cn.iocoder.yudao.module.pms.dal.dataobject.projectuser.ProjectUserDO; +import cn.iocoder.yudao.framework.common.pojo.PageResult; +import cn.iocoder.yudao.framework.common.pojo.PageParam; + +/** + * 项目与人员关联关系 Service 接口 + * + * @author hhyykk + */ +public interface ProjectUserService { + + /** + * 创建项目与人员关联关系 + * + * @param createReqVO 创建信息 + * @return 编号 + */ + Long createProjectUser(@Valid ProjectUserSaveReqVO createReqVO); + + /** + * 更新项目与人员关联关系 + * + * @param updateReqVO 更新信息 + */ + void updateProjectUser(@Valid ProjectUserSaveReqVO updateReqVO); + + /** + * 删除项目与人员关联关系 + * + * @param id 编号 + */ + void deleteProjectUser(Long id); + + /** + * 获得项目与人员关联关系 + * + * @param id 编号 + * @return 项目与人员关联关系 + */ + ProjectUserDO getProjectUser(Long id); + + /** + * 获得项目与人员关联关系分页 + * + * @param pageReqVO 分页查询 + * @return 项目与人员关联关系分页 + */ + PageResult getProjectUserPage(ProjectUserPageReqVO pageReqVO); + +} \ No newline at end of file diff --git a/yudao-module-pms/yudao-module-pms-biz/src/main/java/cn/iocoder/yudao/module/pms/service/projectuser/ProjectUserServiceImpl.java b/yudao-module-pms/yudao-module-pms-biz/src/main/java/cn/iocoder/yudao/module/pms/service/projectuser/ProjectUserServiceImpl.java new file mode 100644 index 000000000..bd2f9a9d6 --- /dev/null +++ b/yudao-module-pms/yudao-module-pms-biz/src/main/java/cn/iocoder/yudao/module/pms/service/projectuser/ProjectUserServiceImpl.java @@ -0,0 +1,74 @@ +package cn.iocoder.yudao.module.pms.service.projectuser; + +import org.springframework.stereotype.Service; +import jakarta.annotation.Resource; +import org.springframework.validation.annotation.Validated; +import org.springframework.transaction.annotation.Transactional; + +import java.util.*; +import cn.iocoder.yudao.module.pms.controller.admin.projectuser.vo.*; +import cn.iocoder.yudao.module.pms.dal.dataobject.projectuser.ProjectUserDO; +import cn.iocoder.yudao.framework.common.pojo.PageResult; +import cn.iocoder.yudao.framework.common.pojo.PageParam; +import cn.iocoder.yudao.framework.common.util.object.BeanUtils; + +import cn.iocoder.yudao.module.pms.dal.mysql.projectuser.ProjectUserMapper; + +import static cn.iocoder.yudao.framework.common.exception.util.ServiceExceptionUtil.exception; +import static cn.iocoder.yudao.module.pms.enums.ErrorCodeConstants.*; + +/** + * 项目与人员关联关系 Service 实现类 + * + * @author hhyykk + */ +@Service +@Validated +public class ProjectUserServiceImpl implements ProjectUserService { + + @Resource + private ProjectUserMapper projectUserMapper; + + @Override + public Long createProjectUser(ProjectUserSaveReqVO createReqVO) { + // 插入 + ProjectUserDO projectUser = BeanUtils.toBean(createReqVO, ProjectUserDO.class); + projectUserMapper.insert(projectUser); + // 返回 + return projectUser.getId(); + } + + @Override + public void updateProjectUser(ProjectUserSaveReqVO updateReqVO) { + // 校验存在 + validateProjectUserExists(updateReqVO.getId()); + // 更新 + ProjectUserDO updateObj = BeanUtils.toBean(updateReqVO, ProjectUserDO.class); + projectUserMapper.updateById(updateObj); + } + + @Override + public void deleteProjectUser(Long id) { + // 校验存在 + validateProjectUserExists(id); + // 删除 + projectUserMapper.deleteById(id); + } + + private void validateProjectUserExists(Long id) { + if (projectUserMapper.selectById(id) == null) { + throw exception(PROJECT_USER_NOT_EXISTS); + } + } + + @Override + public ProjectUserDO getProjectUser(Long id) { + return projectUserMapper.selectById(id); + } + + @Override + public PageResult getProjectUserPage(ProjectUserPageReqVO pageReqVO) { + return projectUserMapper.selectPage(pageReqVO); + } + +} \ No newline at end of file diff --git a/yudao-module-pms/yudao-module-pms-biz/src/main/resources/mapper/projectuser/ProjectUserMapper.xml b/yudao-module-pms/yudao-module-pms-biz/src/main/resources/mapper/projectuser/ProjectUserMapper.xml new file mode 100644 index 000000000..920087431 --- /dev/null +++ b/yudao-module-pms/yudao-module-pms-biz/src/main/resources/mapper/projectuser/ProjectUserMapper.xml @@ -0,0 +1,12 @@ + + + + + + + \ No newline at end of file diff --git a/yudao-module-pms/yudao-module-pms-biz/src/test/java/cn/iocoder/yudao/module/pms/service/projectuser/ProjectUserServiceImplTest.java b/yudao-module-pms/yudao-module-pms-biz/src/test/java/cn/iocoder/yudao/module/pms/service/projectuser/ProjectUserServiceImplTest.java new file mode 100644 index 000000000..5ec4e40c2 --- /dev/null +++ b/yudao-module-pms/yudao-module-pms-biz/src/test/java/cn/iocoder/yudao/module/pms/service/projectuser/ProjectUserServiceImplTest.java @@ -0,0 +1,126 @@ +package cn.iocoder.yudao.module.pms.service.projectuser; + +import org.junit.jupiter.api.Disabled; +import org.junit.jupiter.api.Test; +import org.springframework.boot.test.mock.mockito.MockBean; + +import jakarta.annotation.Resource; + +import cn.iocoder.yudao.framework.test.core.ut.BaseDbUnitTest; + +import cn.iocoder.yudao.module.pms.controller.admin.projectuser.vo.*; +import cn.iocoder.yudao.module.pms.dal.dataobject.projectuser.ProjectUserDO; +import cn.iocoder.yudao.module.pms.dal.mysql.projectuser.ProjectUserMapper; +import cn.iocoder.yudao.framework.common.pojo.PageResult; + +import jakarta.annotation.Resource; +import org.springframework.context.annotation.Import; +import java.util.*; +import java.time.LocalDateTime; + +import static cn.hutool.core.util.RandomUtil.*; +import static cn.iocoder.yudao.module.pms.enums.ErrorCodeConstants.*; +import static cn.iocoder.yudao.framework.test.core.util.AssertUtils.*; +import static cn.iocoder.yudao.framework.test.core.util.RandomUtils.*; +import static cn.iocoder.yudao.framework.common.util.date.LocalDateTimeUtils.*; +import static cn.iocoder.yudao.framework.common.util.object.ObjectUtils.*; +import static cn.iocoder.yudao.framework.common.util.date.DateUtils.*; +import static org.junit.jupiter.api.Assertions.*; +import static org.mockito.Mockito.*; + +/** + * {@link ProjectUserServiceImpl} 的单元测试类 + * + * @author hhyykk + */ +@Import(ProjectUserServiceImpl.class) +public class ProjectUserServiceImplTest extends BaseDbUnitTest { + + @Resource + private ProjectUserServiceImpl projectUserService; + + @Resource + private ProjectUserMapper projectUserMapper; + + @Test + public void testCreateProjectUser_success() { + // 准备参数 + ProjectUserSaveReqVO createReqVO = randomPojo(ProjectUserSaveReqVO.class).setId(null); + + // 调用 + Long projectUserId = projectUserService.createProjectUser(createReqVO); + // 断言 + assertNotNull(projectUserId); + // 校验记录的属性是否正确 + ProjectUserDO projectUser = projectUserMapper.selectById(projectUserId); + assertPojoEquals(createReqVO, projectUser, "id"); + } + + @Test + public void testUpdateProjectUser_success() { + // mock 数据 + ProjectUserDO dbProjectUser = randomPojo(ProjectUserDO.class); + projectUserMapper.insert(dbProjectUser);// @Sql: 先插入出一条存在的数据 + // 准备参数 + ProjectUserSaveReqVO updateReqVO = randomPojo(ProjectUserSaveReqVO.class, o -> { + o.setId(dbProjectUser.getId()); // 设置更新的 ID + }); + + // 调用 + projectUserService.updateProjectUser(updateReqVO); + // 校验是否更新正确 + ProjectUserDO projectUser = projectUserMapper.selectById(updateReqVO.getId()); // 获取最新的 + assertPojoEquals(updateReqVO, projectUser); + } + + @Test + public void testUpdateProjectUser_notExists() { + // 准备参数 + ProjectUserSaveReqVO updateReqVO = randomPojo(ProjectUserSaveReqVO.class); + + // 调用, 并断言异常 + assertServiceException(() -> projectUserService.updateProjectUser(updateReqVO), PROJECT_USER_NOT_EXISTS); + } + + @Test + public void testDeleteProjectUser_success() { + // mock 数据 + ProjectUserDO dbProjectUser = randomPojo(ProjectUserDO.class); + projectUserMapper.insert(dbProjectUser);// @Sql: 先插入出一条存在的数据 + // 准备参数 + Long id = dbProjectUser.getId(); + + // 调用 + projectUserService.deleteProjectUser(id); + // 校验数据不存在了 + assertNull(projectUserMapper.selectById(id)); + } + + @Test + public void testDeleteProjectUser_notExists() { + // 准备参数 + Long id = randomLongId(); + + // 调用, 并断言异常 + assertServiceException(() -> projectUserService.deleteProjectUser(id), PROJECT_USER_NOT_EXISTS); + } + + @Test + @Disabled // TODO 请修改 null 为需要的值，然后删除 @Disabled 注解 + public void testGetProjectUserPage() { + // mock 数据 + ProjectUserDO dbProjectUser = randomPojo(ProjectUserDO.class, o -> { // 等会查询到 + }); + projectUserMapper.insert(dbProjectUser); + // 准备参数 + ProjectUserPageReqVO reqVO = new ProjectUserPageReqVO(); + + // 调用 + PageResult pageResult = projectUserService.getProjectUserPage(reqVO); + // 断言 + assertEquals(1, pageResult.getTotal()); + assertEquals(1, pageResult.getList().size()); + assertPojoEquals(dbProjectUser, pageResult.getList().get(0)); + } + +} \ No newline at end of file