package cn.iocoder.yudao.module.mes.service.trace.consumer;
|
|
import cn.iocoder.yudao.framework.test.core.ut.BaseDbUnitTest;
|
import cn.iocoder.yudao.module.mes.controller.admin.trace.consumer.vo.MesTraceConsumerScanDistributionReqVO;
|
import cn.iocoder.yudao.module.mes.controller.admin.trace.consumer.vo.MesTraceConsumerScanDistributionRespVO;
|
import cn.iocoder.yudao.module.mes.controller.admin.trace.consumer.vo.MesTraceConsumerScanSummaryRespVO;
|
import cn.iocoder.yudao.module.mes.controller.admin.trace.consumer.vo.MesTraceConsumerScanTrendRespVO;
|
import cn.iocoder.yudao.module.mes.dal.dataobject.trace.consumer.MesTraceConsumerScanEventDO;
|
import cn.iocoder.yudao.module.mes.dal.mysql.trace.consumer.MesTraceConsumerScanEventMapper;
|
import jakarta.annotation.Resource;
|
import org.junit.jupiter.api.Test;
|
import org.springframework.context.annotation.Import;
|
|
import java.time.LocalDateTime;
|
import java.util.List;
|
|
import static org.junit.jupiter.api.Assertions.assertEquals;
|
import static org.junit.jupiter.api.Assertions.assertNotNull;
|
import static org.junit.jupiter.api.Assertions.assertThrows;
|
|
/**
|
* {@link MesTraceConsumerScanStatisticsServiceImpl} 的单元测试
|
*
|
* @author 超级管理员
|
*/
|
@Import(MesTraceConsumerScanStatisticsServiceImpl.class)
|
public class MesTraceConsumerScanStatisticsServiceImplTest extends BaseDbUnitTest {
|
|
@Resource
|
private MesTraceConsumerScanStatisticsService statisticsService;
|
|
@Resource
|
private MesTraceConsumerScanEventMapper eventMapper;
|
|
private MesTraceConsumerScanEventDO buildEvent(String traceCode, String sourceType, String channel,
|
String province, Long batchId, boolean success,
|
LocalDateTime eventTime) {
|
return MesTraceConsumerScanEventDO.builder()
|
.traceCode(traceCode).normalizedCode(traceCode).sourceType(sourceType)
|
.channel(channel).province(province).batchId(batchId).success(success)
|
.eventTime(eventTime).build();
|
}
|
|
@Test
|
public void testGetSummary_统计口径() {
|
LocalDateTime t1 = LocalDateTime.of(2026, 9, 1, 10, 0);
|
LocalDateTime t2 = LocalDateTime.of(2026, 9, 2, 11, 0);
|
eventMapper.insert(buildEvent("B1", "BAG", "微信", "江苏省", 100L, true, t1));
|
eventMapper.insert(buildEvent("B2", "BAG", "浏览器", "江苏省", 100L, true, t2));
|
eventMapper.insert(buildEvent("B3", "BATCH", null, null, null, false, t2));
|
|
MesTraceConsumerScanSummaryRespVO vo = statisticsService.getSummary(
|
LocalDateTime.of(2026, 9, 1, 0, 0), LocalDateTime.of(2026, 9, 3, 0, 0));
|
|
assertNotNull(vo);
|
assertEquals(3L, vo.getTotal());
|
assertEquals(2L, vo.getSuccess());
|
assertEquals(1L, vo.getFailed());
|
assertEquals(1L, vo.getUniqueBatchCount()); // 仅 batchId 非空且去重
|
}
|
|
@Test
|
public void testGetTrend_补齐无数据日期() {
|
// 9 月 1 日两条、9 月 3 日一条；9 月 2 日无数据应补 0
|
eventMapper.insert(buildEvent("B1", "BAG", "微信", null, 100L, true,
|
LocalDateTime.of(2026, 9, 1, 9, 0)));
|
eventMapper.insert(buildEvent("B2", "BATCH", null, null, null, false,
|
LocalDateTime.of(2026, 9, 1, 10, 0)));
|
eventMapper.insert(buildEvent("B3", "BAG", "浏览器", null, 100L, true,
|
LocalDateTime.of(2026, 9, 3, 9, 0)));
|
|
List<MesTraceConsumerScanTrendRespVO> list = statisticsService.getTrend(
|
LocalDateTime.of(2026, 9, 1, 0, 0), LocalDateTime.of(2026, 9, 4, 0, 0));
|
|
assertNotNull(list);
|
assertEquals(3, list.size());
|
assertEquals("2026-09-01", list.get(0).getDate());
|
assertEquals(2L, list.get(0).getTotal());
|
assertEquals("2026-09-02", list.get(1).getDate());
|
assertEquals(0L, list.get(1).getTotal()); // 补 0
|
assertEquals("2026-09-03", list.get(2).getDate());
|
assertEquals(1L, list.get(2).getTotal());
|
}
|
|
@Test
|
public void testGetDistribution_bySourceType() {
|
eventMapper.insert(buildEvent("B1", "BAG", null, null, 100L, true,
|
LocalDateTime.of(2026, 9, 1, 9, 0)));
|
eventMapper.insert(buildEvent("B2", "PALLET", null, null, 100L, true,
|
LocalDateTime.of(2026, 9, 1, 10, 0)));
|
eventMapper.insert(buildEvent("B3", "PALLET", null, null, null, false,
|
LocalDateTime.of(2026, 9, 1, 11, 0)));
|
|
MesTraceConsumerScanDistributionReqVO reqVO = new MesTraceConsumerScanDistributionReqVO();
|
reqVO.setGroupBy("sourceType");
|
List<MesTraceConsumerScanDistributionRespVO> list = statisticsService.getDistribution(reqVO);
|
|
assertNotNull(list);
|
// PALLET 2 条 > BAG 1 条（按 count 倒序）
|
assertEquals("PALLET", list.get(0).getName());
|
assertEquals(2L, list.get(0).getCount());
|
assertEquals("BAG", list.get(1).getName());
|
assertEquals(1L, list.get(1).getCount());
|
}
|
|
@Test
|
public void testGetDistribution_nullKey映射未知() {
|
// 解析失败的事件 sourceType 为 null，不得因 groupBy 空键抛 NPE，应聚合为"未知"
|
eventMapper.insert(buildEvent("B1", "BAG", null, null, 100L, true,
|
LocalDateTime.of(2026, 9, 1, 9, 0)));
|
eventMapper.insert(buildEvent("B2", null, null, null, null, false,
|
LocalDateTime.of(2026, 9, 1, 10, 0)));
|
eventMapper.insert(buildEvent("B3", null, null, null, null, false,
|
LocalDateTime.of(2026, 9, 1, 11, 0)));
|
|
MesTraceConsumerScanDistributionReqVO reqVO = new MesTraceConsumerScanDistributionReqVO();
|
reqVO.setGroupBy("sourceType");
|
List<MesTraceConsumerScanDistributionRespVO> list = statisticsService.getDistribution(reqVO);
|
|
assertNotNull(list);
|
// 按 count 倒序：未知(2) 在 BAG(1) 之前
|
assertEquals(2, list.size());
|
assertEquals("未知", list.get(0).getName());
|
assertEquals(2L, list.get(0).getCount());
|
assertEquals("BAG", list.get(1).getName());
|
assertEquals(1L, list.get(1).getCount());
|
}
|
|
@Test
|
public void testGetDistribution_success显示名称() {
|
eventMapper.insert(buildEvent("B1", "BAG", null, null, 100L, true,
|
LocalDateTime.of(2026, 9, 1, 9, 0)));
|
eventMapper.insert(buildEvent("B2", "BAG", null, null, null, false,
|
LocalDateTime.of(2026, 9, 1, 10, 0)));
|
|
MesTraceConsumerScanDistributionReqVO reqVO = new MesTraceConsumerScanDistributionReqVO();
|
reqVO.setGroupBy("success");
|
List<MesTraceConsumerScanDistributionRespVO> list = statisticsService.getDistribution(reqVO);
|
|
assertNotNull(list);
|
assertEquals(2, list.size());
|
// 数量相同时按分组名升序："0"(失败) 在 "1"(成功) 之前
|
assertEquals("失败", list.get(0).getName());
|
assertEquals("成功", list.get(1).getName());
|
}
|
|
@Test
|
public void testGetDistribution_非法维度拒绝() {
|
MesTraceConsumerScanDistributionReqVO reqVO = new MesTraceConsumerScanDistributionReqVO();
|
reqVO.setGroupBy("item_code; DROP TABLE x");
|
assertThrows(IllegalArgumentException.class, () -> statisticsService.getDistribution(reqVO));
|
}
|
|
}
|
