package cn.iocoder.yudao.module.mes.service.pd.knowledge;

import cn.iocoder.yudao.module.mes.dal.dataobject.pd.knowledge.KnowledgeDO;
import cn.iocoder.yudao.module.mes.dal.dataobject.pd.knowledge.KnowledgeStandardDO;
import cn.iocoder.yudao.module.mes.dal.mysql.pd.knowledge.KnowledgeMapper;
import cn.iocoder.yudao.module.mes.dal.mysql.pd.knowledge.KnowledgeStandardMapper;
import cn.iocoder.yudao.module.mes.dal.dataobject.md.item.MesMdProductBomDO;
import cn.iocoder.yudao.module.mes.service.md.item.MesMdItemService;
import cn.iocoder.yudao.module.mes.service.md.item.MesMdProductBomService;
import com.baomidou.mybatisplus.core.conditions.Wrapper;
import org.junit.jupiter.api.Test;
import org.springframework.test.util.ReflectionTestUtils;
import java.math.BigDecimal;
import java.util.List;
import java.util.Set;
import static org.junit.jupiter.api.Assertions.assertEquals;
import static org.mockito.ArgumentMatchers.any;
import static org.mockito.Mockito.*;

class KnowledgeSearchServiceTest {
    @Test
    void missingBomReferenceIsNotRecommended() {
        var knowledgeMapper = mock(KnowledgeMapper.class);
        var standardMapper = mock(KnowledgeStandardMapper.class);
        var knowledge = new KnowledgeDO(); knowledge.setId(1L); knowledge.setBomId(501L); knowledge.setMarket("ALL");
        when(knowledgeMapper.selectList(any(Wrapper.class))).thenReturn(List.of(knowledge));
        when(standardMapper.selectList(any(Wrapper.class))).thenReturn(List.of());
        var service = new KnowledgeSearchService();
        ReflectionTestUtils.setField(service, "knowledgeMapper", knowledgeMapper);
        ReflectionTestUtils.setField(service, "standardMapper", standardMapper);
        assertEquals(List.of(), service.match(new KnowledgeMatch.Request(null, null, null, null, Set.of())));
    }

    @Test
    void productBomMatchesByComponentRatherThanBomLineId() {
        var knowledgeMapper = mock(KnowledgeMapper.class);
        var standardMapper = mock(KnowledgeStandardMapper.class);
        var itemService = mock(MesMdItemService.class);
        var bomService = mock(MesMdProductBomService.class);
        var knowledge = new KnowledgeDO(); knowledge.setId(1L); knowledge.setProductItemId(42L);
        knowledge.setBomId(70L); knowledge.setMarket("ALL");
        var candidate = new MesMdProductBomDO().setId(502L).setItemId(42L).setBomItemId(70L).setStatus(0);
        when(knowledgeMapper.selectList(any(Wrapper.class))).thenReturn(List.of(knowledge));
        when(standardMapper.selectList(any(Wrapper.class))).thenReturn(List.of());
        when(bomService.getProductBomListByItemId(42L)).thenReturn(List.of(candidate));
        var service = new KnowledgeSearchService();
        ReflectionTestUtils.setField(service, "knowledgeMapper", knowledgeMapper);
        ReflectionTestUtils.setField(service, "standardMapper", standardMapper);
        ReflectionTestUtils.setField(service, "itemService", itemService);
        ReflectionTestUtils.setField(service, "bomService", bomService);
        assertEquals(List.of(1L), service.match(new KnowledgeMatch.Request(42L, null, null, null, Set.of()))
                .stream().map(KnowledgeDO::getId).toList());
        org.junit.jupiter.api.Assertions.assertThrows(RuntimeException.class,
                () -> service.match(new KnowledgeMatch.Request(42L, 71L, null, null, Set.of())));
        candidate.setBomItemId(71L);
        assertEquals(List.of(), service.match(new KnowledgeMatch.Request(42L, null, null, null, Set.of())));
    }

    @Test
    void joinsStandardsToTheRightKnowledgeRecord() {
        var knowledgeMapper = mock(KnowledgeMapper.class);
        var standardMapper = mock(KnowledgeStandardMapper.class);
        var first = new KnowledgeDO(); first.setId(1L); first.setMarket("ALL");
        var second = new KnowledgeDO(); second.setId(2L); second.setMarket("ALL");
        var standard = new KnowledgeStandardDO(); standard.setKnowledgeId(2L); standard.setStandardCode("IEC-1");
        when(knowledgeMapper.selectList(any(Wrapper.class))).thenReturn(List.of(first, second));
        when(standardMapper.selectList(any(Wrapper.class))).thenReturn(List.of(standard));
        var service = new KnowledgeSearchService();
        ReflectionTestUtils.setField(service, "knowledgeMapper", knowledgeMapper);
        ReflectionTestUtils.setField(service, "standardMapper", standardMapper);
        var result = service.match(new KnowledgeMatch.Request(null, null, BigDecimal.ONE, null, Set.of("IEC-1")));
        assertEquals(List.of(2L), result.stream().map(KnowledgeDO::getId).toList());
    }

    @Test
    void noKnowledgeDoesNotQueryStandards() {
        var knowledgeMapper = mock(KnowledgeMapper.class);
        var standardMapper = mock(KnowledgeStandardMapper.class);
        when(knowledgeMapper.selectList(any(Wrapper.class))).thenReturn(List.of());
        var service = new KnowledgeSearchService();
        ReflectionTestUtils.setField(service, "knowledgeMapper", knowledgeMapper);
        ReflectionTestUtils.setField(service, "standardMapper", standardMapper);
        assertEquals(List.of(), service.match(new KnowledgeMatch.Request(null, null, null, null, Set.of())));
        verifyNoInteractions(standardMapper);
    }
}