package com.ruoyi.system.service.impl;

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

import java.util.Arrays;
import java.util.Collections;
import java.util.List;
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 com.ruoyi.common.exception.ServiceException;
import com.ruoyi.system.domain.TBatteryGroupField;
import com.ruoyi.system.domain.TBatteryGroupTemplate;
import com.ruoyi.system.domain.TBatteryGroupTemplateField;
import com.ruoyi.system.mapper.TBatteryGroupTemplateMapper;

/** 电池组模板 Service 单元测试。 */
@ExtendWith(MockitoExtension.class)
class TBatteryGroupTemplateServiceImplTest
{
    @Mock
    private TBatteryGroupTemplateMapper batteryGroupTemplateMapper;

    @InjectMocks
    private TBatteryGroupTemplateServiceImpl batteryGroupTemplateService;

    @Test
    void insertShouldValidateFieldsAndReplaceRelations()
    {
        TBatteryGroupTemplate template = template(null, "  默认模板  ", Arrays.asList(1L, 2L));
        when(batteryGroupTemplateMapper.selectEnabledFieldsByIds(template.getFieldIds()))
                .thenReturn(Arrays.asList(field(1L), field(2L)));
        doAnswer(invocation -> {
            invocation.<TBatteryGroupTemplate>getArgument(0).setTemplateId(10L);
            return 1;
        }).when(batteryGroupTemplateMapper).insertBatteryGroupTemplate(any(TBatteryGroupTemplate.class));

        assertEquals(1, batteryGroupTemplateService.insertBatteryGroupTemplate(template));

        ArgumentCaptor<List<TBatteryGroupTemplateField>> captor = ArgumentCaptor.forClass(List.class);
        verify(batteryGroupTemplateMapper).batchInsertTemplateFields(captor.capture());
        assertEquals(2, captor.getValue().size());
        assertEquals(0, captor.getValue().get(0).getSortOrder());
        assertEquals(1, captor.getValue().get(1).getSortOrder());
        assertEquals("默认模板", template.getTemplateName());
    }

    @Test
    void insertShouldRejectDuplicateFieldIds()
    {
        TBatteryGroupTemplate template = template(null, "模板", Arrays.asList(1L, 1L));

        ServiceException exception = assertThrows(ServiceException.class,
                () -> batteryGroupTemplateService.insertBatteryGroupTemplate(template));

        assertEquals("模板字段不能重复或为空", exception.getMessage());
        verify(batteryGroupTemplateMapper, never()).insertBatteryGroupTemplate(any(TBatteryGroupTemplate.class));
    }

    @Test
    void insertShouldRejectDisabledOrMissingField()
    {
        TBatteryGroupTemplate template = template(null, "模板", Collections.singletonList(1L));
        when(batteryGroupTemplateMapper.selectEnabledFieldsByIds(template.getFieldIds()))
                .thenReturn(Collections.emptyList());

        ServiceException exception = assertThrows(ServiceException.class,
                () -> batteryGroupTemplateService.insertBatteryGroupTemplate(template));

        assertEquals("模板包含不存在或已禁用的电池组字段", exception.getMessage());
    }

    @Test
    void updateShouldDeleteOldRelationsBeforeInsert()
    {
        TBatteryGroupTemplate template = template(10L, "模板", Collections.singletonList(1L));
        when(batteryGroupTemplateMapper.selectBatteryGroupTemplateById(10L)).thenReturn(template);
        when(batteryGroupTemplateMapper.selectEnabledFieldsByIds(template.getFieldIds()))
                .thenReturn(Collections.singletonList(field(1L)));
        when(batteryGroupTemplateMapper.updateBatteryGroupTemplate(template)).thenReturn(1);

        assertEquals(1, batteryGroupTemplateService.updateBatteryGroupTemplate(template));
        verify(batteryGroupTemplateMapper).deleteTemplateFields(10L);
        verify(batteryGroupTemplateMapper).batchInsertTemplateFields(any(List.class));
    }

    private TBatteryGroupTemplate template(Long id, String name, java.util.List<Long> fieldIds)
    {
        TBatteryGroupTemplate template = new TBatteryGroupTemplate();
        template.setTemplateId(id);
        template.setTemplateName(name);
        template.setFieldIds(fieldIds);
        template.setCreateBy("admin");
        template.setUpdateBy("admin");
        return template;
    }

    private TBatteryGroupField field(Long id)
    {
        TBatteryGroupField field = new TBatteryGroupField();
        field.setFieldId(id);
        field.setStatus("0");
        return field;
    }
}