package com.sinata.appointment.service;

import static org.junit.jupiter.api.Assertions.assertEquals;
import static org.mockito.ArgumentMatchers.any;
import static org.mockito.Mockito.times;
import static org.mockito.Mockito.verify;
import static org.mockito.Mockito.when;

import com.sinata.appointment.constants.ApprovalNodeCode;
import com.sinata.appointment.constants.ApprovalInstanceStatus;
import com.sinata.appointment.constants.ApprovalTaskStatus;
import com.sinata.appointment.constants.LocalUnitApproverType;
import com.sinata.appointment.entity.AppointmentOrder;
import com.sinata.appointment.entity.MpPassLocalUnit;
import com.sinata.appointment.entity.MpUser;
import com.sinata.appointment.entity.ApprovalInstance;
import com.sinata.appointment.entity.ApprovalTask;
import com.sinata.appointment.mapper.ApprovalInstanceMapper;
import com.sinata.appointment.mapper.ApprovalTaskMapper;
import com.sinata.appointment.mapper.BizApprovalNodeConfigMapper;
import com.sinata.appointment.mapper.MpPassLocalUnitMapper;
import com.sinata.appointment.mapper.MpPassLocalUnitMemberMapper;
import com.sinata.appointment.mapper.MpStaffGroupMemberMapper;
import com.sinata.appointment.mapper.MpUserMapper;
import com.fasterxml.jackson.databind.ObjectMapper;
import java.util.List;
import org.junit.jupiter.api.Test;
import org.junit.jupiter.api.extension.ExtendWith;
import org.mockito.ArgumentCaptor;
import org.mockito.Mock;
import org.mockito.junit.jupiter.MockitoExtension;
import org.springframework.test.util.ReflectionTestUtils;

@ExtendWith(MockitoExtension.class)
class ApprovalEngineServiceTest {

    @Mock
    private BizApprovalNodeConfigMapper bizApprovalNodeConfigMapper;
    @Mock
    private ApprovalInstanceMapper approvalInstanceMapper;
    @Mock
    private ApprovalTaskMapper approvalTaskMapper;
    @Mock
    private MpUserMapper mpUserMapper;
    @Mock
    private MpPassLocalUnitMapper mpPassLocalUnitMapper;
    @Mock
    private MpPassLocalUnitMemberMapper mpPassLocalUnitMemberMapper;
    @Mock
    private MpStaffGroupMemberMapper mpStaffGroupMemberMapper;
    @Mock
    private SysBizSettingCacheService sysBizSettingCacheService;

    @Test
    void cancelInstanceMarksPendingAndWaitingTasksAsCancelled() {
        ApprovalEngineService approvalEngineService =
                new ApprovalEngineService(
                        bizApprovalNodeConfigMapper,
                        approvalInstanceMapper,
                        approvalTaskMapper,
                        mpUserMapper,
                        mpPassLocalUnitMapper,
                        mpPassLocalUnitMemberMapper,
                        mpStaffGroupMemberMapper,
                        sysBizSettingCacheService,
                        new ObjectMapper());

        ApprovalInstance instance = new ApprovalInstance();
        instance.setId(1L);
        instance.setStatus(ApprovalInstanceStatus.IN_PROGRESS);

        ApprovalTask pendingTask = new ApprovalTask();
        pendingTask.setId(10L);
        pendingTask.setTaskStatus(ApprovalTaskStatus.PENDING);

        ApprovalTask waitingTask = new ApprovalTask();
        waitingTask.setId(11L);
        waitingTask.setTaskStatus(ApprovalTaskStatus.WAITING);

        when(approvalTaskMapper.selectList(any())).thenReturn(List.of(pendingTask, waitingTask));

        approvalEngineService.cancelInstance(instance, "预约人取消");

        assertEquals(ApprovalInstanceStatus.CANCELLED, instance.getStatus());
        verify(approvalInstanceMapper).updateById(instance);

        ArgumentCaptor<ApprovalTask> taskCaptor = ArgumentCaptor.forClass(ApprovalTask.class);
        verify(approvalTaskMapper, org.mockito.Mockito.times(2)).updateById(taskCaptor.capture());
        List<ApprovalTask> updatedTasks = taskCaptor.getAllValues();
        assertEquals(ApprovalTaskStatus.CANCELLED, updatedTasks.get(0).getTaskStatus());
        assertEquals("预约人取消", updatedTasks.get(0).getComment());
        assertEquals(ApprovalTaskStatus.CANCELLED, updatedTasks.get(1).getTaskStatus());
        assertEquals("预约人取消", updatedTasks.get(1).getComment());
    }

    @Test
    void createTasksForLocalUnitNodeStrictShouldMergeDeptApproversIntoOneUnitGroup() {
        ApprovalEngineService approvalEngineService =
                new ApprovalEngineService(
                        bizApprovalNodeConfigMapper,
                        approvalInstanceMapper,
                        approvalTaskMapper,
                        mpUserMapper,
                        mpPassLocalUnitMapper,
                        mpPassLocalUnitMemberMapper,
                        mpStaffGroupMemberMapper,
                        sysBizSettingCacheService,
                        new ObjectMapper());

        AppointmentOrder order = new AppointmentOrder();
        order.setInvolveOtherLocalUnit(true);
        order.setPassLocalUnitIds("[11]");

        MpPassLocalUnit unit = new MpPassLocalUnit();
        unit.setId(11L);
        unit.setApproverType(LocalUnitApproverType.DEPT);
        unit.setApproverDeptIds("[100,101]");

        MpUser deptUser1 = new MpUser();
        deptUser1.setId(1L);
        MpUser deptUser2 = new MpUser();
        deptUser2.setId(2L);
        MpUser deptUser3 = new MpUser();
        deptUser3.setId(3L);

        when(mpPassLocalUnitMapper.selectList(any())).thenReturn(List.of(unit));
        when(mpUserMapper.selectList(any()))
                .thenReturn(List.of(deptUser1, deptUser2), List.of(deptUser2, deptUser3));

        int created =
                (int)
                        ReflectionTestUtils.invokeMethod(
                                approvalEngineService,
                                "createTasksForLocalUnitNodeStrict",
                                99L,
                                order,
                                ApprovalTaskStatus.PENDING);

        assertEquals(3, created);

        ArgumentCaptor<ApprovalTask> taskCaptor = ArgumentCaptor.forClass(ApprovalTask.class);
        verify(approvalTaskMapper, times(3)).insert(taskCaptor.capture());
        List<ApprovalTask> insertedTasks = taskCaptor.getAllValues();
        assertEquals(
                List.of(1L, 2L, 3L),
                insertedTasks.stream().map(ApprovalTask::getAssigneeUserId).toList());
        assertEquals(
                List.of(
                        ApprovalNodeCode.LEVEL3_LOCAL_DEPT_LEADER + ":UNIT:11",
                        ApprovalNodeCode.LEVEL3_LOCAL_DEPT_LEADER + ":UNIT:11",
                        ApprovalNodeCode.LEVEL3_LOCAL_DEPT_LEADER + ":UNIT:11"),
                insertedTasks.stream().map(ApprovalTask::getGroupKey).toList());
    }
}