package jnpf.dmsService.impl;

import com.baomidou.mybatisplus.core.conditions.query.LambdaQueryWrapper;
import com.baomidou.mybatisplus.core.conditions.update.LambdaUpdateWrapper;
import com.baomidou.mybatisplus.core.metadata.TableInfoHelper;
import jnpf.dmsEntity.DmsWenjianBianhaoGuizeEntity;
import jnpf.dmsMapper.DmsWenjianBianhaoGuizeMapper;
import jnpf.exception.DataException;
import org.apache.ibatis.builder.MapperBuilderAssistant;
import org.apache.ibatis.session.Configuration;
import org.junit.jupiter.api.BeforeEach;
import org.junit.jupiter.api.Test;
import org.junit.jupiter.api.extension.ExtendWith;
import org.mockito.ArgumentCaptor;
import org.mockito.InjectMocks;
import org.mockito.Mock;
import org.mockito.junit.jupiter.MockitoExtension;

import java.util.Collections;

import static org.junit.jupiter.api.Assertions.assertTrue;
import static org.mockito.ArgumentMatchers.any;
import static org.mockito.ArgumentMatchers.isNull;
import static org.mockito.Mockito.never;
import static org.mockito.Mockito.verify;
import static org.mockito.Mockito.when;

@ExtendWith(MockitoExtension.class)
class DmsWenjianBianhaoGuizeTongguoServiceImplTest {

    @Mock private DmsWenjianBianhaoGuizeMapper guizeMapper;

    @InjectMocks
    private DmsWenjianBianhaoGuizeTongguoServiceImpl service;

    @BeforeEach
    void initTableInfo() {
        if (TableInfoHelper.getTableInfo(DmsWenjianBianhaoGuizeEntity.class) == null) {
            TableInfoHelper.initTableInfo(new MapperBuilderAssistant(new Configuration(), ""),
                    DmsWenjianBianhaoGuizeEntity.class);
        }
    }

    @Test
    void disablesOnlyOtherRulesWithTheSameSerialNumber() throws DataException {
        DmsWenjianBianhaoGuizeEntity currentRule = new DmsWenjianBianhaoGuizeEntity();
        currentRule.setId("rule-current");
        currentRule.setFlowId("flow-1");
        currentRule.setFlowTaskId("task-1");
        currentRule.setGuizeLiushuihao("serial-001");
        when(guizeMapper.selectList(any())).thenReturn(Collections.singletonList(currentRule));

        service.tongguo("flow-1", "task-1");

        ArgumentCaptor<LambdaUpdateWrapper> captor = ArgumentCaptor.forClass(LambdaUpdateWrapper.class);
        verify(guizeMapper).update(isNull(), captor.capture());
        ArgumentCaptor<LambdaQueryWrapper> queryCaptor = ArgumentCaptor.forClass(LambdaQueryWrapper.class);
        verify(guizeMapper).selectList(queryCaptor.capture());
        assertTrue(queryCaptor.getValue().getCustomSqlSegment().contains("f_flow_id"));
        assertTrue(queryCaptor.getValue().getCustomSqlSegment().contains("f_flow_task_id"));
        String condition = captor.getValue().getCustomSqlSegment();
        assertTrue(condition.contains("guize_liushuihao"));
        assertTrue(condition.contains("f_id NOT IN"));
        assertTrue(captor.getValue().getSqlSet().contains("biz_enabled"));
    }

    @Test
    void doesNotDisableRulesWhenTheCurrentRuleHasNoSerialNumber() throws DataException {
        DmsWenjianBianhaoGuizeEntity currentRule = new DmsWenjianBianhaoGuizeEntity();
        currentRule.setId("rule-current");
        currentRule.setFlowId("flow-1");
        currentRule.setFlowTaskId("task-1");
        currentRule.setGuizeLiushuihao(" ");
        when(guizeMapper.selectList(any())).thenReturn(Collections.singletonList(currentRule));

        service.tongguo("flow-1", "task-1");

        verify(guizeMapper, never()).update(isNull(), any());
    }
}