package cn.iocoder.yudao.module.mes.service.trace.integrity;

import cn.iocoder.yudao.framework.common.exception.ServiceException;
import cn.iocoder.yudao.framework.test.core.ut.BaseDbUnitTest;
import cn.iocoder.yudao.module.mes.controller.admin.trace.integrity.vo.MesTraceIntegrityVerifyRespVO;
import cn.iocoder.yudao.module.mes.dal.dataobject.trace.consumer.MesTraceConsumerScanEventDO;
import cn.iocoder.yudao.module.mes.dal.dataobject.trace.integrity.MesTraceIntegrityRecordDO;
import cn.iocoder.yudao.module.mes.dal.mysql.trace.consumer.MesTraceConsumerScanEventMapper;
import cn.iocoder.yudao.module.mes.dal.mysql.trace.integrity.MesTraceIntegrityRecordMapper;
import cn.iocoder.yudao.module.mes.enums.trace.MesTraceIntegrityBizTypeEnum;
import jakarta.annotation.Resource;
import org.junit.jupiter.api.Test;
import org.springframework.context.annotation.Import;

import java.time.LocalDateTime;

import static cn.iocoder.yudao.module.mes.enums.ErrorCodeConstants.MES_TRACE_INTEGRITY_BIZ_TYPE_INVALID;
import static cn.iocoder.yudao.module.mes.enums.ErrorCodeConstants.MES_TRACE_INTEGRITY_RECORD_NOT_EXISTS;
import static org.junit.jupiter.api.Assertions.assertEquals;
import static org.junit.jupiter.api.Assertions.assertFalse;
import static org.junit.jupiter.api.Assertions.assertNotNull;
import static org.junit.jupiter.api.Assertions.assertThrows;
import static org.junit.jupiter.api.Assertions.assertTrue;

/**
 * {@link MesTraceIntegrityServiceImpl} 的单元测试
 *
 * @author 超级管理员
 */
@Import(MesTraceIntegrityServiceImpl.class)
public class MesTraceIntegrityServiceImplTest extends BaseDbUnitTest {

    @Resource
    private MesTraceIntegrityService integrityService;
    @Resource
    private MesTraceIntegrityRecordMapper integrityRecordMapper;
    @Resource
    private MesTraceConsumerScanEventMapper scanEventMapper;

    @Test
    public void testRecordScanIntegrity_生成摘要并入库() {
        MesTraceConsumerScanEventDO event = insertEvent("江苏省", "南京市");
        integrityService.recordScanIntegrity(event);

        MesTraceIntegrityRecordDO record = integrityRecordMapper.selectByBiz(
                MesTraceIntegrityBizTypeEnum.SCAN.getType(), event.getId());
        assertNotNull(record);
        assertEquals(64, record.getContentHash().length());
        assertEquals(event.getNormalizedCode(), record.getBizCode());
    }

    @Test
    public void testVerify_数据完整通过() {
        MesTraceConsumerScanEventDO event = insertEvent("江苏省", "南京市");
        integrityService.recordScanIntegrity(event);

        MesTraceIntegrityVerifyRespVO vo = integrityService.verify(
                MesTraceIntegrityBizTypeEnum.SCAN.getType(), event.getId());

        assertTrue(vo.getMatched());
        assertEquals(vo.getStoredHash(), vo.getCurrentHash());
        assertEquals(event.getNormalizedCode(), vo.getBizCode());
    }

    @Test
    public void testVerify_篡改检测失败() {
        MesTraceConsumerScanEventDO event = insertEvent("江苏省", "南京市");
        integrityService.recordScanIntegrity(event);
        // 模拟事后篡改扫码地区
        scanEventMapper.updateById(new MesTraceConsumerScanEventDO().setId(event.getId()).setProvince("浙江省"));

        MesTraceIntegrityVerifyRespVO vo = integrityService.verify(
                MesTraceIntegrityBizTypeEnum.SCAN.getType(), event.getId());

        assertFalse(vo.getMatched());
        assertEquals("数据不一致，可能存在篡改", vo.getMessage());
    }

    @Test
    public void testVerify_覆盖更新不产生重复记录() {
        MesTraceConsumerScanEventDO event = insertEvent("江苏省", "南京市");
        integrityService.recordScanIntegrity(event);
        integrityService.recordScanIntegrity(event);

        assertEquals(1L, integrityRecordMapper.selectCount());
        MesTraceIntegrityRecordDO record = integrityRecordMapper.selectByBiz(
                MesTraceIntegrityBizTypeEnum.SCAN.getType(), event.getId());
        assertEquals(64, record.getContentHash().length());
    }

    @Test
    public void testVerify_无摘要记录异常() {
        ServiceException ex = assertThrows(ServiceException.class,
                () -> integrityService.verify(MesTraceIntegrityBizTypeEnum.SCAN.getType(), 99999L));
        assertEquals(MES_TRACE_INTEGRITY_RECORD_NOT_EXISTS.getCode(), ex.getCode());
    }

    @Test
    public void testVerify_非法业务类型() {
        ServiceException ex = assertThrows(ServiceException.class,
                () -> integrityService.verify("XXX", 1L));
        assertEquals(MES_TRACE_INTEGRITY_BIZ_TYPE_INVALID.getCode(), ex.getCode());
    }

    @Test
    public void testRecordScanIntegrity_null事件容错() {
        integrityService.recordScanIntegrity(null);
        integrityService.recordScanIntegrity(MesTraceConsumerScanEventDO.builder().id(null).build());
        assertEquals(0L, integrityRecordMapper.selectCount());
    }

    // ==================== 辅助方法 ====================

    private MesTraceConsumerScanEventDO insertEvent(String province, String city) {
        MesTraceConsumerScanEventDO event = MesTraceConsumerScanEventDO.builder()
                .traceCode("BAG-00000001")
                .normalizedCode("BAG-00000001")
                .sourceType("BAG")
                .province(province)
                .city(city)
                .batchId(10L)
                .success(true)
                .eventTime(LocalDateTime.of(2026, 9, 1, 10, 0))
                .build();
        scanEventMapper.insert(event);
        return event;
    }

}