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

import cn.iocoder.yudao.framework.common.exception.ServiceException;
import cn.iocoder.yudao.framework.ip.core.utils.AreaUtils;
import cn.iocoder.yudao.framework.test.core.ut.BaseDbUnitTest;
import cn.iocoder.yudao.module.crm.api.customer.CrmCustomerApi;
import cn.iocoder.yudao.module.crm.api.customer.dto.CrmCustomerRespDTO;
import cn.iocoder.yudao.module.mes.controller.admin.trace.crossregion.vo.MesTraceCrossRegionHandleReqVO;
import cn.iocoder.yudao.module.mes.controller.admin.trace.crossregion.vo.MesTraceCrossRegionStatisticsReqVO;
import cn.iocoder.yudao.module.mes.controller.admin.trace.crossregion.vo.MesTraceCrossRegionStatisticsRespVO;
import cn.iocoder.yudao.module.mes.controller.admin.trace.crossregion.vo.MesTraceCrossRegionSummaryRespVO;
import cn.iocoder.yudao.module.mes.controller.admin.trace.crossregion.vo.MesTraceCrossRegionWarningPageReqVO;
import cn.iocoder.yudao.module.mes.controller.admin.trace.crossregion.vo.MesTraceCrossRegionWarningRespVO;
import cn.iocoder.yudao.module.mes.controller.admin.trace.crossregion.vo.MesTraceAntiConfigRespVO;
import cn.iocoder.yudao.module.mes.dal.dataobject.mdm.MesMdmItemDO;
import cn.iocoder.yudao.module.mes.dal.dataobject.pro.bagcode.MesProBagCodeDO;
import cn.iocoder.yudao.module.mes.dal.dataobject.pro.batch.MesProBatchDO;
import cn.iocoder.yudao.module.mes.dal.dataobject.trace.consumer.MesTraceConsumerScanEventDO;
import cn.iocoder.yudao.module.mes.dal.dataobject.trace.crossregion.MesTraceAntiConfigDO;
import cn.iocoder.yudao.module.mes.dal.dataobject.trace.crossregion.MesTraceCrossRegionWarningDO;
import cn.iocoder.yudao.module.mes.dal.dataobject.wm.productsales.MesWmProductSalesDO;
import cn.iocoder.yudao.module.mes.dal.mysql.trace.crossregion.MesTraceAntiConfigMapper;
import cn.iocoder.yudao.module.mes.dal.mysql.trace.crossregion.MesTraceCrossRegionWarningMapper;
import cn.iocoder.yudao.module.mes.dal.mysql.wm.productsales.MesWmProductSalesDetailMapper;
import cn.iocoder.yudao.module.mes.dal.mysql.wm.productsales.MesWmProductSalesMapper;
import cn.iocoder.yudao.module.mes.enums.trace.MesTraceWarningStatusEnum;
import cn.iocoder.yudao.module.mes.enums.trace.MesTraceWarningTypeEnum;
import cn.iocoder.yudao.module.mes.service.mdm.MesMdmItemService;
import cn.iocoder.yudao.module.mes.service.pro.bagcode.MesProBagCodeService;
import cn.iocoder.yudao.module.mes.service.pro.batch.MesProBatchService;
import cn.iocoder.yudao.module.mes.service.pro.pallet.MesProPalletService;
import cn.iocoder.yudao.module.system.api.user.AdminUserApi;
import jakarta.annotation.Resource;
import org.junit.jupiter.api.Test;
import org.springframework.context.annotation.Import;
import org.springframework.test.context.bean.override.mockito.MockitoBean;

import java.time.LocalDateTime;
import java.util.Collections;
import java.util.List;

import static cn.iocoder.yudao.module.mes.enums.ErrorCodeConstants.MES_TRACE_ANTI_CONFIG_VALUE_INVALID;
import static cn.iocoder.yudao.module.mes.enums.ErrorCodeConstants.MES_TRACE_CROSS_REGION_STATUS_INVALID;
import static org.junit.jupiter.api.Assertions.assertEquals;
import static org.junit.jupiter.api.Assertions.assertThrows;
import static org.junit.jupiter.api.Assertions.assertTrue;
import static org.mockito.ArgumentMatchers.any;
import static org.mockito.Mockito.verify;
import static org.mockito.Mockito.when;

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

    @Resource
    private MesTraceCrossRegionService crossRegionService;
    @Resource
    private MesTraceCrossRegionWarningMapper warningMapper;
    @Resource
    private MesTraceAntiConfigMapper antiConfigMapper;

    @MockitoBean
    private MesProBagCodeService bagCodeService;
    @MockitoBean
    private MesProPalletService palletService;
    @MockitoBean
    private MesProBatchService batchService;
    @MockitoBean
    private MesMdmItemService mdmItemService;
    @MockitoBean
    private CrmCustomerApi customerApi;
    @MockitoBean
    private AdminUserApi adminUserApi;
    @MockitoBean
    private MesWmProductSalesMapper productSalesMapper;
    @MockitoBean
    private MesWmProductSalesDetailMapper productSalesDetailMapper;

    // ==================== 窜货判定 ====================

    @Test
    public void testCheckAndCreateWarning_跨省触发() {
        // 配置省级粒度
        antiConfigMapper.insert(MesTraceAntiConfigDO.builder()
                .configKey("warning_level").configValue("PROVINCE").build());
        mockDealer("江苏省", "南京市");
        MesTraceConsumerScanEventDO event = buildEvent("BAG", "浙江省", "杭州市", true);

        crossRegionService.checkAndCreateWarning(event);

        List<MesTraceCrossRegionWarningDO> warnings = warningMapper.selectList();
        assertEquals(1, warnings.size());
        MesTraceCrossRegionWarningDO warning = warnings.get(0);
        assertEquals(MesTraceWarningTypeEnum.CROSS_PROVINCE.getType(), warning.getWarningType());
        assertEquals("江苏省", warning.getDealerProvince());
        assertEquals("浙江省", warning.getScanProvince());
        assertEquals("SO-001", warning.getSalesCode());
        assertEquals("华东经销商", warning.getClientName());
        assertEquals(event.getId(), warning.getScanEventId());
        assertEquals(MesTraceWarningStatusEnum.PENDING.getStatus(), warning.getStatus());
    }

    @Test
    public void testCheckAndCreateWarning_跨市触发() {
        // 市级粒度：扫码市 ≠ 经销商市 → CROSS_CITY
        antiConfigMapper.insert(MesTraceAntiConfigDO.builder()
                .configKey("warning_level").configValue("CITY").build());
        mockDealer("江苏省", "南京市");
        MesTraceConsumerScanEventDO event = buildEvent("BAG", "江苏省", "苏州市", true);

        crossRegionService.checkAndCreateWarning(event);

        List<MesTraceCrossRegionWarningDO> warnings = warningMapper.selectList();
        assertEquals(1, warnings.size());
        assertEquals(MesTraceWarningTypeEnum.CROSS_CITY.getType(), warnings.get(0).getWarningType());
        assertEquals("南京市", warnings.get(0).getDealerCity());
        assertEquals("苏州市", warnings.get(0).getScanCity());
    }

    @Test
    public void testCheckAndCreateWarning_同地区不预警() {
        antiConfigMapper.insert(MesTraceAntiConfigDO.builder()
                .configKey("warning_level").configValue("PROVINCE").build());
        mockDealer("江苏省", "南京市");
        MesTraceConsumerScanEventDO event = buildEvent("BAG", "江苏省", "南京市", true);

        crossRegionService.checkAndCreateWarning(event);

        assertEquals(0, warningMapper.selectCount());
    }

    @Test
    public void testCheckAndCreateWarning_扫码无地区容错() {
        antiConfigMapper.insert(MesTraceAntiConfigDO.builder()
                .configKey("warning_level").configValue("PROVINCE").build());
        mockDealer("江苏省", "南京市");
        MesTraceConsumerScanEventDO event = buildEvent("BAG", null, null, true);

        crossRegionService.checkAndCreateWarning(event);

        assertEquals(0, warningMapper.selectCount());
    }

    @Test
    public void testCheckAndCreateWarning_批次入口跳过() {
        antiConfigMapper.insert(MesTraceAntiConfigDO.builder()
                .configKey("warning_level").configValue("PROVINCE").build());
        mockDealer("江苏省", "南京市");
        MesTraceConsumerScanEventDO event = buildEvent("BATCH", "浙江省", "杭州市", true);

        crossRegionService.checkAndCreateWarning(event);

        assertEquals(0, warningMapper.selectCount());
    }

    @Test
    public void testCheckAndCreateWarning_袋码未出库跳过() {
        antiConfigMapper.insert(MesTraceAntiConfigDO.builder()
                .configKey("warning_level").configValue("PROVINCE").build());
        // 袋码未回写出库单，且明细表无记录 → 不预警
        when(bagCodeService.getBagCodeByCode("BAG-00000001")).thenReturn(
                MesProBagCodeDO.builder().id(1L).code("BAG-00000001").batchId(10L).build());
        MesTraceConsumerScanEventDO event = buildEvent("BAG", "浙江省", "杭州市", true);

        crossRegionService.checkAndCreateWarning(event);

        verify(productSalesDetailMapper).selectLatestByBagCodeId(1L);
        assertEquals(0, warningMapper.selectCount());
    }

    @Test
    public void testCheckAndCreateWarning_袋码未绑定批次跳过() {
        antiConfigMapper.insert(MesTraceAntiConfigDO.builder()
                .configKey("warning_level").configValue("PROVINCE").build());
        when(bagCodeService.getBagCodeByCode("BAG-00000001")).thenReturn(null);
        MesTraceConsumerScanEventDO event = buildEvent("BAG", "浙江省", "杭州市", true);

        crossRegionService.checkAndCreateWarning(event);

        assertEquals(0, warningMapper.selectCount());
    }

    // ==================== 状态流转 ====================

    @Test
    public void testHandleWarning_状态流转() {
        MesTraceCrossRegionWarningDO warning = buildWarning(MesTraceWarningStatusEnum.PENDING.getStatus());
        warningMapper.insert(warning);

        MesTraceCrossRegionHandleReqVO reqVO = new MesTraceCrossRegionHandleReqVO();
        reqVO.setId(warning.getId());
        reqVO.setStatus(MesTraceWarningStatusEnum.HANDLED.getStatus());
        reqVO.setHandleRemark("确认为物流中转");
        crossRegionService.handleWarning(reqVO);

        MesTraceCrossRegionWarningDO updated = warningMapper.selectById(warning.getId());
        assertEquals(MesTraceWarningStatusEnum.HANDLED.getStatus(), updated.getStatus());
        assertEquals("确认为物流中转", updated.getHandleRemark());
    }

    @Test
    public void testHandleWarning_已处理不可再处理() {
        MesTraceCrossRegionWarningDO warning = buildWarning(MesTraceWarningStatusEnum.HANDLED.getStatus());
        warningMapper.insert(warning);

        MesTraceCrossRegionHandleReqVO reqVO = new MesTraceCrossRegionHandleReqVO();
        reqVO.setId(warning.getId());
        reqVO.setStatus(MesTraceWarningStatusEnum.FALSE_ALARM.getStatus());
        ServiceException ex = assertThrows(ServiceException.class,
                () -> crossRegionService.handleWarning(reqVO));
        assertEquals(MES_TRACE_CROSS_REGION_STATUS_INVALID.getCode(), ex.getCode());
    }

    // ==================== 配置 ====================

    @Test
    public void testGetConfig_updateConfig() {
        antiConfigMapper.insert(MesTraceAntiConfigDO.builder()
                .configKey("warning_level").configValue("PROVINCE").build());

        MesTraceAntiConfigRespVO config = crossRegionService.getConfig();
        assertEquals("PROVINCE", config.getWarningLevel());

        crossRegionService.updateConfig("CITY");
        assertEquals("CITY", antiConfigMapper.selectByConfigKey("warning_level").getConfigValue());

        ServiceException ex = assertThrows(ServiceException.class,
                () -> crossRegionService.updateConfig("INVALID"));
        assertEquals(MES_TRACE_ANTI_CONFIG_VALUE_INVALID.getCode(), ex.getCode());
    }

    // ==================== 汇总与统计 ====================

    @Test
    public void testGetWarningSummary() {
        warningMapper.insert(buildWarning(MesTraceWarningStatusEnum.PENDING.getStatus())
                .setWarningType(MesTraceWarningTypeEnum.CROSS_PROVINCE.getType()));
        warningMapper.insert(buildWarning(MesTraceWarningStatusEnum.HANDLED.getStatus())
                .setWarningType(MesTraceWarningTypeEnum.CROSS_CITY.getType()));
        warningMapper.insert(buildWarning(MesTraceWarningStatusEnum.FALSE_ALARM.getStatus())
                .setWarningType(MesTraceWarningTypeEnum.CROSS_PROVINCE.getType()));

        MesTraceCrossRegionSummaryRespVO vo = crossRegionService.getWarningSummary(null, null);

        assertEquals(3L, vo.getTotal());
        assertEquals(1L, vo.getPending());
        assertEquals(1L, vo.getHandled());
        assertEquals(1L, vo.getFalseAlarm());
        assertEquals(2L, vo.getCrossProvince());
        assertEquals(1L, vo.getCrossCity());
    }

    @Test
    public void testGetWarningStatistics_分组统计() {
        warningMapper.insert(buildWarning(MesTraceWarningStatusEnum.PENDING.getStatus())
                .setWarningType(MesTraceWarningTypeEnum.CROSS_PROVINCE.getType()).setScanProvince("浙江省"));
        warningMapper.insert(buildWarning(MesTraceWarningStatusEnum.PENDING.getStatus())
                .setWarningType(MesTraceWarningTypeEnum.CROSS_PROVINCE.getType()).setScanProvince("浙江省"));
        warningMapper.insert(buildWarning(MesTraceWarningStatusEnum.PENDING.getStatus())
                .setWarningType(MesTraceWarningTypeEnum.CROSS_PROVINCE.getType()).setScanProvince("江苏省"));

        MesTraceCrossRegionStatisticsReqVO reqVO = new MesTraceCrossRegionStatisticsReqVO();
        reqVO.setGroupBy("scanProvince");
        List<MesTraceCrossRegionStatisticsRespVO> list = crossRegionService.getWarningStatistics(reqVO);

        assertEquals(2, list.size());
        assertEquals("浙江省", list.get(0).getName());
        assertEquals(2L, list.get(0).getCount());
        assertEquals("江苏省", list.get(1).getName());
        assertEquals(1L, list.get(1).getCount());
    }

    @Test
    public void testGetWarningStatistics_非法维度拒绝() {
        MesTraceCrossRegionStatisticsReqVO reqVO = new MesTraceCrossRegionStatisticsReqVO();
        reqVO.setGroupBy("item_code; DROP TABLE x");
        assertThrows(IllegalArgumentException.class,
                () -> crossRegionService.getWarningStatistics(reqVO));
    }

    // ==================== 分页 ====================

    @Test
    public void testGetWarningPage() {
        when(adminUserApi.getUserMap(any())).thenReturn(Collections.emptyMap());
        warningMapper.insert(buildWarning(MesTraceWarningStatusEnum.PENDING.getStatus())
                .setWarningType(MesTraceWarningTypeEnum.CROSS_PROVINCE.getType())
                .setScanProvince("浙江省").setDealerProvince("江苏省"));

        MesTraceCrossRegionWarningPageReqVO reqVO = new MesTraceCrossRegionWarningPageReqVO();
        reqVO.setStatus(MesTraceWarningStatusEnum.PENDING.getStatus());
        var page = crossRegionService.getWarningPage(reqVO);

        assertEquals(1L, page.getTotal());
        MesTraceCrossRegionWarningRespVO vo = page.getList().get(0);
        assertEquals(MesTraceWarningStatusEnum.PENDING.getName(), vo.getStatusName());
        assertEquals(MesTraceWarningTypeEnum.CROSS_PROVINCE.getName(), vo.getWarningTypeName());
        assertTrue(vo.getWarningNo() != null);
    }

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

    /**
     * 构建消费者扫码事件（id 固定，便于断言关联）
     */
    private MesTraceConsumerScanEventDO buildEvent(String sourceType, String province, String city, boolean success) {
        return MesTraceConsumerScanEventDO.builder()
                .id(1L)
                .normalizedCode("BAG-00000001")
                .sourceType(sourceType)
                .province(province)
                .city(city)
                .batchId(10L)
                .success(success)
                .eventTime(LocalDateTime.of(2026, 9, 1, 10, 0))
                .build();
    }

    /**
     * Mock 袋码 → 出库单 → 经销商（指定地区）链路
     */
    private void mockDealer(String dealerProvince, String dealerCity) {
        MesProBagCodeDO bag = MesProBagCodeDO.builder().id(1L).code("BAG-00000001")
                .batchId(10L).salesId(100L).build();
        when(bagCodeService.getBagCodeByCode("BAG-00000001")).thenReturn(bag);

        MesWmProductSalesDO sales = MesWmProductSalesDO.builder().id(100L).code("SO-001")
                .clientId(200L).clientName("华东经销商").build();
        when(productSalesMapper.selectById(100L)).thenReturn(sales);

        Integer areaId = AreaUtils.parseArea(dealerProvince + "/" + dealerCity).getId();
        CrmCustomerRespDTO customer = new CrmCustomerRespDTO();
        customer.setId(200L);
        customer.setName("华东经销商");
        customer.setAreaId(areaId);
        when(customerApi.getCustomer(200L)).thenReturn(customer);

        MesProBatchDO batch = MesProBatchDO.builder().id(10L).itemId(5L).build();
        when(batchService.getBatch(10L)).thenReturn(batch);

        MesMdmItemDO item = MesMdmItemDO.builder().id(5L).code("ITEM001").name("水稻种子").build();
        when(mdmItemService.getItem(5L)).thenReturn(item);
    }

    private MesTraceCrossRegionWarningDO buildWarning(Integer status) {
        return MesTraceCrossRegionWarningDO.builder()
                .warningNo("WARN20260901100000ABCDEF")
                .batchId(10L)
                .batchCode("PC20260901L001M001_001")
                .salesId(100L)
                .salesCode("SO-001")
                .clientId(200L)
                .clientName("华东经销商")
                .dealerProvince("江苏省")
                .dealerCity("南京市")
                .scanProvince("浙江省")
                .scanCity("杭州市")
                .status(status)
                .eventTime(LocalDateTime.of(2026, 9, 1, 10, 0))
                .build();
    }

}