add org worker record
chenshuichuan
2 years ago
parent
46464ae58a
commit
1f91bafe6a
@ -0,0 +1,95 @@
|
||||
package cn.iocoder.yudao.module.mes.controller.admin.orgworker;
|
||||
|
||||
import org.springframework.web.bind.annotation.*;
|
||||
import javax.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 javax.validation.constraints.*;
|
||||
import javax.validation.*;
|
||||
import javax.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.mes.controller.admin.orgworker.vo.*;
|
||||
import cn.iocoder.yudao.module.mes.dal.dataobject.orgworker.OrgWorkerDO;
|
||||
import cn.iocoder.yudao.module.mes.service.orgworker.OrgWorkerService;
|
||||
|
||||
@Tag(name = "管理后台 - 工位安排")
|
||||
@RestController
|
||||
@RequestMapping("/mes/org-worker")
|
||||
@Validated
|
||||
public class OrgWorkerController {
|
||||
|
||||
@Resource
|
||||
private OrgWorkerService orgWorkerService;
|
||||
|
||||
@PostMapping("/create")
|
||||
@Operation(summary = "创建工位安排")
|
||||
@PreAuthorize("@ss.hasPermission('mes:org-worker:create')")
|
||||
public CommonResult<Long> createOrgWorker(@Valid @RequestBody OrgWorkerSaveReqVO createReqVO) {
|
||||
return success(orgWorkerService.createOrgWorker(createReqVO));
|
||||
}
|
||||
|
||||
@PutMapping("/update")
|
||||
@Operation(summary = "更新工位安排")
|
||||
@PreAuthorize("@ss.hasPermission('mes:org-worker:update')")
|
||||
public CommonResult<Boolean> updateOrgWorker(@Valid @RequestBody OrgWorkerSaveReqVO updateReqVO) {
|
||||
orgWorkerService.updateOrgWorker(updateReqVO);
|
||||
return success(true);
|
||||
}
|
||||
|
||||
@DeleteMapping("/delete")
|
||||
@Operation(summary = "删除工位安排")
|
||||
@Parameter(name = "id", description = "编号", required = true)
|
||||
@PreAuthorize("@ss.hasPermission('mes:org-worker:delete')")
|
||||
public CommonResult<Boolean> deleteOrgWorker(@RequestParam("id") Long id) {
|
||||
orgWorkerService.deleteOrgWorker(id);
|
||||
return success(true);
|
||||
}
|
||||
|
||||
@GetMapping("/get")
|
||||
@Operation(summary = "获得工位安排")
|
||||
@Parameter(name = "id", description = "编号", required = true, example = "1024")
|
||||
@PreAuthorize("@ss.hasPermission('mes:org-worker:query')")
|
||||
public CommonResult<OrgWorkerRespVO> getOrgWorker(@RequestParam("id") Long id) {
|
||||
OrgWorkerDO orgWorker = orgWorkerService.getOrgWorker(id);
|
||||
return success(BeanUtils.toBean(orgWorker, OrgWorkerRespVO.class));
|
||||
}
|
||||
|
||||
@GetMapping("/page")
|
||||
@Operation(summary = "获得工位安排分页")
|
||||
@PreAuthorize("@ss.hasPermission('mes:org-worker:query')")
|
||||
public CommonResult<PageResult<OrgWorkerRespVO>> getOrgWorkerPage(@Valid OrgWorkerPageReqVO pageReqVO) {
|
||||
PageResult<OrgWorkerDO> pageResult = orgWorkerService.getOrgWorkerPage(pageReqVO);
|
||||
return success(BeanUtils.toBean(pageResult, OrgWorkerRespVO.class));
|
||||
}
|
||||
|
||||
@GetMapping("/export-excel")
|
||||
@Operation(summary = "导出工位安排 Excel")
|
||||
@PreAuthorize("@ss.hasPermission('mes:org-worker:export')")
|
||||
@ApiAccessLog(operateType = EXPORT)
|
||||
public void exportOrgWorkerExcel(@Valid OrgWorkerPageReqVO pageReqVO,
|
||||
HttpServletResponse response) throws IOException {
|
||||
pageReqVO.setPageSize(PageParam.PAGE_SIZE_NONE);
|
||||
List<OrgWorkerDO> list = orgWorkerService.getOrgWorkerPage(pageReqVO).getList();
|
||||
// 导出 Excel
|
||||
ExcelUtils.write(response, "工位安排.xls", "数据", OrgWorkerRespVO.class,
|
||||
BeanUtils.toBean(list, OrgWorkerRespVO.class));
|
||||
}
|
||||
|
||||
}
|
||||
@ -0,0 +1,38 @@
|
||||
package cn.iocoder.yudao.module.mes.controller.admin.orgworker.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 OrgWorkerPageReqVO extends PageParam {
|
||||
|
||||
@Schema(description = "工位id", example = "4727")
|
||||
private Long orgId;
|
||||
|
||||
@Schema(description = "工作日期")
|
||||
@DateTimeFormat(pattern = FORMAT_YEAR_MONTH_DAY_HOUR_MINUTE_SECOND)
|
||||
private LocalDateTime[] workDate;
|
||||
|
||||
@Schema(description = "班别", example = "2")
|
||||
private Integer groupType;
|
||||
|
||||
@Schema(description = "工人id", example = "2095")
|
||||
private Long workerId;
|
||||
|
||||
@Schema(description = "报工单id", example = "5496")
|
||||
private Long reportId;
|
||||
|
||||
@Schema(description = "创建时间")
|
||||
@DateTimeFormat(pattern = FORMAT_YEAR_MONTH_DAY_HOUR_MINUTE_SECOND)
|
||||
private LocalDateTime[] createTime;
|
||||
|
||||
}
|
||||
@ -0,0 +1,47 @@
|
||||
package cn.iocoder.yudao.module.mes.controller.admin.orgworker.vo;
|
||||
|
||||
import io.swagger.v3.oas.annotations.media.Schema;
|
||||
import lombok.*;
|
||||
import java.util.*;
|
||||
import java.util.*;
|
||||
import org.springframework.format.annotation.DateTimeFormat;
|
||||
import java.time.LocalDateTime;
|
||||
import com.alibaba.excel.annotation.*;
|
||||
import cn.iocoder.yudao.framework.excel.core.annotations.DictFormat;
|
||||
import cn.iocoder.yudao.framework.excel.core.convert.DictConvert;
|
||||
|
||||
@Schema(description = "管理后台 - 工位安排 Response VO")
|
||||
@Data
|
||||
@ExcelIgnoreUnannotated
|
||||
public class OrgWorkerRespVO {
|
||||
|
||||
@Schema(description = "id", requiredMode = Schema.RequiredMode.REQUIRED, example = "2158")
|
||||
@ExcelProperty("id")
|
||||
private Long id;
|
||||
|
||||
@Schema(description = "工位id", requiredMode = Schema.RequiredMode.REQUIRED, example = "4727")
|
||||
@ExcelProperty("工位id")
|
||||
private Long orgId;
|
||||
|
||||
@Schema(description = "工作日期", requiredMode = Schema.RequiredMode.REQUIRED)
|
||||
@ExcelProperty("工作日期")
|
||||
private LocalDateTime workDate;
|
||||
|
||||
@Schema(description = "班别", requiredMode = Schema.RequiredMode.REQUIRED, example = "2")
|
||||
@ExcelProperty(value = "班别", converter = DictConvert.class)
|
||||
@DictFormat("mes_group_type") // TODO 代码优化：建议设置到对应的 DictTypeConstants 枚举类中
|
||||
private Integer groupType;
|
||||
|
||||
@Schema(description = "工人id", requiredMode = Schema.RequiredMode.REQUIRED, example = "2095")
|
||||
@ExcelProperty("工人id")
|
||||
private Long workerId;
|
||||
|
||||
@Schema(description = "报工单id", example = "5496")
|
||||
@ExcelProperty("报工单id")
|
||||
private Long reportId;
|
||||
|
||||
@Schema(description = "创建时间", requiredMode = Schema.RequiredMode.REQUIRED)
|
||||
@ExcelProperty("创建时间")
|
||||
private LocalDateTime createTime;
|
||||
|
||||
}
|
||||
@ -0,0 +1,36 @@
|
||||
package cn.iocoder.yudao.module.mes.controller.admin.orgworker.vo;
|
||||
|
||||
import io.swagger.v3.oas.annotations.media.Schema;
|
||||
import lombok.*;
|
||||
import java.util.*;
|
||||
import javax.validation.constraints.*;
|
||||
import org.springframework.format.annotation.DateTimeFormat;
|
||||
import java.time.LocalDateTime;
|
||||
|
||||
@Schema(description = "管理后台 - 工位安排新增/修改 Request VO")
|
||||
@Data
|
||||
public class OrgWorkerSaveReqVO {
|
||||
|
||||
@Schema(description = "id", requiredMode = Schema.RequiredMode.REQUIRED, example = "2158")
|
||||
private Long id;
|
||||
|
||||
@Schema(description = "工位id", requiredMode = Schema.RequiredMode.REQUIRED, example = "4727")
|
||||
@NotNull(message = "工位id不能为空")
|
||||
private Long orgId;
|
||||
|
||||
@Schema(description = "工作日期", requiredMode = Schema.RequiredMode.REQUIRED)
|
||||
@NotNull(message = "工作日期不能为空")
|
||||
private LocalDateTime workDate;
|
||||
|
||||
@Schema(description = "班别", requiredMode = Schema.RequiredMode.REQUIRED, example = "2")
|
||||
@NotNull(message = "班别不能为空")
|
||||
private Integer groupType;
|
||||
|
||||
@Schema(description = "工人id", requiredMode = Schema.RequiredMode.REQUIRED, example = "2095")
|
||||
@NotNull(message = "工人id不能为空")
|
||||
private Long workerId;
|
||||
|
||||
@Schema(description = "报工单id", example = "5496")
|
||||
private Long reportId;
|
||||
|
||||
}
|
||||
@ -0,0 +1,54 @@
|
||||
package cn.iocoder.yudao.module.mes.dal.dataobject.orgworker;
|
||||
|
||||
import lombok.*;
|
||||
import java.util.*;
|
||||
import java.time.LocalDateTime;
|
||||
import java.time.LocalDateTime;
|
||||
import java.time.LocalDateTime;
|
||||
import com.baomidou.mybatisplus.annotation.*;
|
||||
import cn.iocoder.yudao.framework.mybatis.core.dataobject.BaseDO;
|
||||
|
||||
/**
|
||||
* 工位安排 DO
|
||||
*
|
||||
* @author 内蒙必硕
|
||||
*/
|
||||
@TableName("mes_org_worker")
|
||||
@KeySequence("mes_org_worker_seq") // 用于 Oracle、PostgreSQL、Kingbase、DB2、H2 数据库的主键自增。如果是 MySQL 等数据库，可不写。
|
||||
@Data
|
||||
@EqualsAndHashCode(callSuper = true)
|
||||
@ToString(callSuper = true)
|
||||
@Builder
|
||||
@NoArgsConstructor
|
||||
@AllArgsConstructor
|
||||
public class OrgWorkerDO extends BaseDO {
|
||||
|
||||
/**
|
||||
* id
|
||||
*/
|
||||
@TableId
|
||||
private Long id;
|
||||
/**
|
||||
* 工位id
|
||||
*/
|
||||
private Long orgId;
|
||||
/**
|
||||
* 工作日期
|
||||
*/
|
||||
private LocalDateTime workDate;
|
||||
/**
|
||||
* 班别
|
||||
*
|
||||
* 枚举 {@link TODO mes_group_type 对应的类}
|
||||
*/
|
||||
private Integer groupType;
|
||||
/**
|
||||
* 工人id
|
||||
*/
|
||||
private Long workerId;
|
||||
/**
|
||||
* 报工单id
|
||||
*/
|
||||
private Long reportId;
|
||||
|
||||
}
|
||||
@ -0,0 +1,31 @@
|
||||
package cn.iocoder.yudao.module.mes.dal.mysql.orgworker;
|
||||
|
||||
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.mes.dal.dataobject.orgworker.OrgWorkerDO;
|
||||
import org.apache.ibatis.annotations.Mapper;
|
||||
import cn.iocoder.yudao.module.mes.controller.admin.orgworker.vo.*;
|
||||
|
||||
/**
|
||||
* 工位安排 Mapper
|
||||
*
|
||||
* @author 内蒙必硕
|
||||
*/
|
||||
@Mapper
|
||||
public interface OrgWorkerMapper extends BaseMapperX<OrgWorkerDO> {
|
||||
|
||||
default PageResult<OrgWorkerDO> selectPage(OrgWorkerPageReqVO reqVO) {
|
||||
return selectPage(reqVO, new LambdaQueryWrapperX<OrgWorkerDO>()
|
||||
.eqIfPresent(OrgWorkerDO::getOrgId, reqVO.getOrgId())
|
||||
.betweenIfPresent(OrgWorkerDO::getWorkDate, reqVO.getWorkDate())
|
||||
.eqIfPresent(OrgWorkerDO::getGroupType, reqVO.getGroupType())
|
||||
.eqIfPresent(OrgWorkerDO::getWorkerId, reqVO.getWorkerId())
|
||||
.eqIfPresent(OrgWorkerDO::getReportId, reqVO.getReportId())
|
||||
.betweenIfPresent(OrgWorkerDO::getCreateTime, reqVO.getCreateTime())
|
||||
.orderByDesc(OrgWorkerDO::getId));
|
||||
}
|
||||
|
||||
}
|
||||
@ -0,0 +1,55 @@
|
||||
package cn.iocoder.yudao.module.mes.service.orgworker;
|
||||
|
||||
import java.util.*;
|
||||
import javax.validation.*;
|
||||
import cn.iocoder.yudao.module.mes.controller.admin.orgworker.vo.*;
|
||||
import cn.iocoder.yudao.module.mes.dal.dataobject.orgworker.OrgWorkerDO;
|
||||
import cn.iocoder.yudao.framework.common.pojo.PageResult;
|
||||
import cn.iocoder.yudao.framework.common.pojo.PageParam;
|
||||
|
||||
/**
|
||||
* 工位安排 Service 接口
|
||||
*
|
||||
* @author 内蒙必硕
|
||||
*/
|
||||
public interface OrgWorkerService {
|
||||
|
||||
/**
|
||||
* 创建工位安排
|
||||
*
|
||||
* @param createReqVO 创建信息
|
||||
* @return 编号
|
||||
*/
|
||||
Long createOrgWorker(@Valid OrgWorkerSaveReqVO createReqVO);
|
||||
|
||||
/**
|
||||
* 更新工位安排
|
||||
*
|
||||
* @param updateReqVO 更新信息
|
||||
*/
|
||||
void updateOrgWorker(@Valid OrgWorkerSaveReqVO updateReqVO);
|
||||
|
||||
/**
|
||||
* 删除工位安排
|
||||
*
|
||||
* @param id 编号
|
||||
*/
|
||||
void deleteOrgWorker(Long id);
|
||||
|
||||
/**
|
||||
* 获得工位安排
|
||||
*
|
||||
* @param id 编号
|
||||
* @return 工位安排
|
||||
*/
|
||||
OrgWorkerDO getOrgWorker(Long id);
|
||||
|
||||
/**
|
||||
* 获得工位安排分页
|
||||
*
|
||||
* @param pageReqVO 分页查询
|
||||
* @return 工位安排分页
|
||||
*/
|
||||
PageResult<OrgWorkerDO> getOrgWorkerPage(OrgWorkerPageReqVO pageReqVO);
|
||||
|
||||
}
|
||||
@ -0,0 +1,74 @@
|
||||
package cn.iocoder.yudao.module.mes.service.orgworker;
|
||||
|
||||
import org.springframework.stereotype.Service;
|
||||
import javax.annotation.Resource;
|
||||
import org.springframework.validation.annotation.Validated;
|
||||
import org.springframework.transaction.annotation.Transactional;
|
||||
|
||||
import java.util.*;
|
||||
import cn.iocoder.yudao.module.mes.controller.admin.orgworker.vo.*;
|
||||
import cn.iocoder.yudao.module.mes.dal.dataobject.orgworker.OrgWorkerDO;
|
||||
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.mes.dal.mysql.orgworker.OrgWorkerMapper;
|
||||
|
||||
import static cn.iocoder.yudao.framework.common.exception.util.ServiceExceptionUtil.exception;
|
||||
import static cn.iocoder.yudao.module.mes.enums.ErrorCodeConstants.*;
|
||||
|
||||
/**
|
||||
* 工位安排 Service 实现类
|
||||
*
|
||||
* @author 内蒙必硕
|
||||
*/
|
||||
@Service
|
||||
@Validated
|
||||
public class OrgWorkerServiceImpl implements OrgWorkerService {
|
||||
|
||||
@Resource
|
||||
private OrgWorkerMapper orgWorkerMapper;
|
||||
|
||||
@Override
|
||||
public Long createOrgWorker(OrgWorkerSaveReqVO createReqVO) {
|
||||
// 插入
|
||||
OrgWorkerDO orgWorker = BeanUtils.toBean(createReqVO, OrgWorkerDO.class);
|
||||
orgWorkerMapper.insert(orgWorker);
|
||||
// 返回
|
||||
return orgWorker.getId();
|
||||
}
|
||||
|
||||
@Override
|
||||
public void updateOrgWorker(OrgWorkerSaveReqVO updateReqVO) {
|
||||
// 校验存在
|
||||
validateOrgWorkerExists(updateReqVO.getId());
|
||||
// 更新
|
||||
OrgWorkerDO updateObj = BeanUtils.toBean(updateReqVO, OrgWorkerDO.class);
|
||||
orgWorkerMapper.updateById(updateObj);
|
||||
}
|
||||
|
||||
@Override
|
||||
public void deleteOrgWorker(Long id) {
|
||||
// 校验存在
|
||||
validateOrgWorkerExists(id);
|
||||
// 删除
|
||||
orgWorkerMapper.deleteById(id);
|
||||
}
|
||||
|
||||
private void validateOrgWorkerExists(Long id) {
|
||||
if (orgWorkerMapper.selectById(id) == null) {
|
||||
throw exception(ORG_WORKER_NOT_EXISTS);
|
||||
}
|
||||
}
|
||||
|
||||
@Override
|
||||
public OrgWorkerDO getOrgWorker(Long id) {
|
||||
return orgWorkerMapper.selectById(id);
|
||||
}
|
||||
|
||||
@Override
|
||||
public PageResult<OrgWorkerDO> getOrgWorkerPage(OrgWorkerPageReqVO pageReqVO) {
|
||||
return orgWorkerMapper.selectPage(pageReqVO);
|
||||
}
|
||||
|
||||
}
|
||||
@ -0,0 +1,150 @@
|
||||
package cn.iocoder.yudao.module.mes.service.orgworker;
|
||||
|
||||
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 cn.iocoder.yudao.framework.test.core.ut.BaseDbUnitTest;
|
||||
|
||||
import cn.iocoder.yudao.module.mes.controller.admin.orgworker.vo.*;
|
||||
import cn.iocoder.yudao.module.mes.dal.dataobject.orgworker.OrgWorkerDO;
|
||||
import cn.iocoder.yudao.module.mes.dal.mysql.orgworker.OrgWorkerMapper;
|
||||
import cn.iocoder.yudao.framework.common.pojo.PageResult;
|
||||
|
||||
import javax.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.mes.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 OrgWorkerServiceImpl} 的单元测试类
|
||||
*
|
||||
* @author 内蒙必硕
|
||||
*/
|
||||
@Import(OrgWorkerServiceImpl.class)
|
||||
public class OrgWorkerServiceImplTest extends BaseDbUnitTest {
|
||||
|
||||
@Resource
|
||||
private OrgWorkerServiceImpl orgWorkerService;
|
||||
|
||||
@Resource
|
||||
private OrgWorkerMapper orgWorkerMapper;
|
||||
|
||||
@Test
|
||||
public void testCreateOrgWorker_success() {
|
||||
// 准备参数
|
||||
OrgWorkerSaveReqVO createReqVO = randomPojo(OrgWorkerSaveReqVO.class).setId(null);
|
||||
|
||||
// 调用
|
||||
Long orgWorkerId = orgWorkerService.createOrgWorker(createReqVO);
|
||||
// 断言
|
||||
assertNotNull(orgWorkerId);
|
||||
// 校验记录的属性是否正确
|
||||
OrgWorkerDO orgWorker = orgWorkerMapper.selectById(orgWorkerId);
|
||||
assertPojoEquals(createReqVO, orgWorker, "id");
|
||||
}
|
||||
|
||||
@Test
|
||||
public void testUpdateOrgWorker_success() {
|
||||
// mock 数据
|
||||
OrgWorkerDO dbOrgWorker = randomPojo(OrgWorkerDO.class);
|
||||
orgWorkerMapper.insert(dbOrgWorker);// @Sql: 先插入出一条存在的数据
|
||||
// 准备参数
|
||||
OrgWorkerSaveReqVO updateReqVO = randomPojo(OrgWorkerSaveReqVO.class, o -> {
|
||||
o.setId(dbOrgWorker.getId()); // 设置更新的 ID
|
||||
});
|
||||
|
||||
// 调用
|
||||
orgWorkerService.updateOrgWorker(updateReqVO);
|
||||
// 校验是否更新正确
|
||||
OrgWorkerDO orgWorker = orgWorkerMapper.selectById(updateReqVO.getId()); // 获取最新的
|
||||
assertPojoEquals(updateReqVO, orgWorker);
|
||||
}
|
||||
|
||||
@Test
|
||||
public void testUpdateOrgWorker_notExists() {
|
||||
// 准备参数
|
||||
OrgWorkerSaveReqVO updateReqVO = randomPojo(OrgWorkerSaveReqVO.class);
|
||||
|
||||
// 调用, 并断言异常
|
||||
assertServiceException(() -> orgWorkerService.updateOrgWorker(updateReqVO), ORG_WORKER_NOT_EXISTS);
|
||||
}
|
||||
|
||||
@Test
|
||||
public void testDeleteOrgWorker_success() {
|
||||
// mock 数据
|
||||
OrgWorkerDO dbOrgWorker = randomPojo(OrgWorkerDO.class);
|
||||
orgWorkerMapper.insert(dbOrgWorker);// @Sql: 先插入出一条存在的数据
|
||||
// 准备参数
|
||||
Long id = dbOrgWorker.getId();
|
||||
|
||||
// 调用
|
||||
orgWorkerService.deleteOrgWorker(id);
|
||||
// 校验数据不存在了
|
||||
assertNull(orgWorkerMapper.selectById(id));
|
||||
}
|
||||
|
||||
@Test
|
||||
public void testDeleteOrgWorker_notExists() {
|
||||
// 准备参数
|
||||
Long id = randomLongId();
|
||||
|
||||
// 调用, 并断言异常
|
||||
assertServiceException(() -> orgWorkerService.deleteOrgWorker(id), ORG_WORKER_NOT_EXISTS);
|
||||
}
|
||||
|
||||
@Test
|
||||
@Disabled // TODO 请修改 null 为需要的值，然后删除 @Disabled 注解
|
||||
public void testGetOrgWorkerPage() {
|
||||
// mock 数据
|
||||
OrgWorkerDO dbOrgWorker = randomPojo(OrgWorkerDO.class, o -> { // 等会查询到
|
||||
o.setOrgId(null);
|
||||
o.setWorkDate(null);
|
||||
o.setGroupType(null);
|
||||
o.setWorkerId(null);
|
||||
o.setReportId(null);
|
||||
o.setCreateTime(null);
|
||||
});
|
||||
orgWorkerMapper.insert(dbOrgWorker);
|
||||
// 测试 orgId 不匹配
|
||||
orgWorkerMapper.insert(cloneIgnoreId(dbOrgWorker, o -> o.setOrgId(null)));
|
||||
// 测试 workDate 不匹配
|
||||
orgWorkerMapper.insert(cloneIgnoreId(dbOrgWorker, o -> o.setWorkDate(null)));
|
||||
// 测试 groupType 不匹配
|
||||
orgWorkerMapper.insert(cloneIgnoreId(dbOrgWorker, o -> o.setGroupType(null)));
|
||||
// 测试 workerId 不匹配
|
||||
orgWorkerMapper.insert(cloneIgnoreId(dbOrgWorker, o -> o.setWorkerId(null)));
|
||||
// 测试 reportId 不匹配
|
||||
orgWorkerMapper.insert(cloneIgnoreId(dbOrgWorker, o -> o.setReportId(null)));
|
||||
// 测试 createTime 不匹配
|
||||
orgWorkerMapper.insert(cloneIgnoreId(dbOrgWorker, o -> o.setCreateTime(null)));
|
||||
// 准备参数
|
||||
OrgWorkerPageReqVO reqVO = new OrgWorkerPageReqVO();
|
||||
reqVO.setOrgId(null);
|
||||
reqVO.setWorkDate(buildBetweenTime(2023, 2, 1, 2023, 2, 28));
|
||||
reqVO.setGroupType(null);
|
||||
reqVO.setWorkerId(null);
|
||||
reqVO.setReportId(null);
|
||||
reqVO.setCreateTime(buildBetweenTime(2023, 2, 1, 2023, 2, 28));
|
||||
|
||||
// 调用
|
||||
PageResult<OrgWorkerDO> pageResult = orgWorkerService.getOrgWorkerPage(reqVO);
|
||||
// 断言
|
||||
assertEquals(1, pageResult.getTotal());
|
||||
assertEquals(1, pageResult.getList().size());
|
||||
assertPojoEquals(dbOrgWorker, pageResult.getList().get(0));
|
||||
}
|
||||
|
||||
}
|
||||
Loading…
Reference in New Issue