package com.ruoyi.other.service.impl;

import com.baomidou.mybatisplus.core.MybatisConfiguration;
import com.baomidou.mybatisplus.core.metadata.TableInfoHelper;
import com.baomidou.mybatisplus.extension.plugins.pagination.Page;
import com.ruoyi.common.core.domain.R;
import com.ruoyi.other.api.domain.TGroupCompany;
import com.ruoyi.other.api.dto.CompanyRechargeRecordExcelDTO;
import com.ruoyi.other.api.dto.CompanyRechargeRecordQueryDTO;
import com.ruoyi.other.api.model.TCompanyRechargeRecord;
import com.ruoyi.other.mapper.TCompanyRechargeRecordMapper;
import com.ruoyi.other.mapper.TGroupCompanyMapper;
import com.ruoyi.other.service.CompanyAccountAccessService;
import com.ruoyi.system.api.domain.SysUser;
import com.ruoyi.system.api.feignClient.SysUserClient;
import org.apache.ibatis.builder.MapperBuilderAssistant;
import org.junit.jupiter.api.BeforeAll;
import org.junit.jupiter.api.Test;
import org.mockito.Mockito;
import org.springframework.test.util.ReflectionTestUtils;

import java.math.BigDecimal;
import java.time.LocalDateTime;
import java.util.Arrays;
import java.util.Collections;
import java.util.List;

import static org.junit.jupiter.api.Assertions.assertEquals;
import static org.junit.jupiter.api.Assertions.assertNotEquals;
import static org.junit.jupiter.api.Assertions.assertNotNull;
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.ArgumentMatchers.eq;

/**
 * 公司充值记录 服务实现类 单测：覆盖充值流水号生成、状态默认值、公司隔离、名称回填、Excel DTO 映射。
 */
class TCompanyRechargeRecordServiceImplTest {

    @BeforeAll
    static void initTableInfo() {
        TableInfoHelper.initTableInfo(new MapperBuilderAssistant(new MybatisConfiguration(), ""),
                TCompanyRechargeRecord.class);
    }

    @Test
    void rechargeGeneratesSerialNoAndSetsStatusSuccess() {
        TGroupCompany company = new TGroupCompany();
        company.setId(100);
        company.setName("M公司");

        TGroupCompanyMapper groupCompanyMapper = Mockito.mock(TGroupCompanyMapper.class);
        Mockito.when(groupCompanyMapper.selectById(100)).thenReturn(company);

        TCompanyRechargeRecordMapper mapper = Mockito.mock(TCompanyRechargeRecordMapper.class);
        Mockito.when(mapper.insert(any())).thenAnswer(invocation -> {
            TCompanyRechargeRecord r = invocation.getArgument(0);
            r.setId(555);
            return 1;
        });

        CompanyAccountAccessService access = Mockito.mock(CompanyAccountAccessService.class);
        SysUserClient userClient = Mockito.mock(SysUserClient.class);

        TCompanyRechargeRecordServiceImpl service = new TCompanyRechargeRecordServiceImpl();
        ReflectionTestUtils.setField(service, "groupCompanyMapper", groupCompanyMapper);
        ReflectionTestUtils.setField(service, "companyAccountAccessService", access);
        ReflectionTestUtils.setField(service, "sysUserClient", userClient);
        ReflectionTestUtils.setField(service, "baseMapper", mapper);

        TCompanyRechargeRecord in = new TCompanyRechargeRecord();
        in.setCompanyId(100);
        in.setPayAmount(new BigDecimal("1000.00"));
        in.setGiftAmount(new BigDecimal("200.00"));
        Integer newId = service.recharge(in);
        assertEquals(555, newId);
        assertEquals(1, in.getStatus());
        assertNotNull(in.getSerialNo());
        assertTrue(in.getSerialNo().startsWith("CR"));
        assertEquals(LocalDateTime.now().toLocalDate(), in.getRechargeTime().toLocalDate());
        assertEquals(new BigDecimal("1200.00"), in.getArrivalAmount());
        // 验证公司余额原子加 setSql 触发
        Mockito.verify(groupCompanyMapper, Mockito.times(1))
                .update(eq(null), any());
    }

    @Test
    void rechargeFailsWhenCompanyNotExists() {
        TGroupCompanyMapper groupCompanyMapper = Mockito.mock(TGroupCompanyMapper.class);
        Mockito.when(groupCompanyMapper.selectById(999)).thenReturn(null);

        TCompanyRechargeRecordMapper mapper = Mockito.mock(TCompanyRechargeRecordMapper.class);
        CompanyAccountAccessService access = Mockito.mock(CompanyAccountAccessService.class);
        SysUserClient userClient = Mockito.mock(SysUserClient.class);

        TCompanyRechargeRecordServiceImpl service = new TCompanyRechargeRecordServiceImpl();
        ReflectionTestUtils.setField(service, "groupCompanyMapper", groupCompanyMapper);
        ReflectionTestUtils.setField(service, "companyAccountAccessService", access);
        ReflectionTestUtils.setField(service, "sysUserClient", userClient);
        ReflectionTestUtils.setField(service, "baseMapper", mapper);

        TCompanyRechargeRecord in = new TCompanyRechargeRecord();
        in.setCompanyId(999);
        in.setPayAmount(new BigDecimal("100"));

        IllegalArgumentException ex = assertThrows(IllegalArgumentException.class, () -> service.recharge(in));
        assertTrue(ex.getMessage().contains("公司不存在"));
        Mockito.verifyNoInteractions(mapper);
    }

    @Test
    @SuppressWarnings({"rawtypes", "unchecked"})
    void pageListAppliesCompanyFilterAndFillsNames() {
        TCompanyRechargeRecord r1 = new TCompanyRechargeRecord();
        r1.setId(1);
        r1.setCompanyId(100);
        r1.setOperatorId(9001L);
        r1.setPayAmount(new BigDecimal("100"));
        TCompanyRechargeRecord r2 = new TCompanyRechargeRecord();
        r2.setId(2);
        r2.setCompanyId(200);
        r2.setOperatorId(9002L);
        r2.setPayAmount(new BigDecimal("50"));

        Page pageResult = new Page(1, 10);
        pageResult.setRecords(Arrays.asList(r1, r2));
        pageResult.setTotal(2L);

        TCompanyRechargeRecordMapper mapper = Mockito.mock(TCompanyRechargeRecordMapper.class);
        Mockito.when(mapper.selectPage(any(Page.class), any())).thenReturn(pageResult);

        TGroupCompanyMapper groupCompanyMapper = Mockito.mock(TGroupCompanyMapper.class);
        TGroupCompany c1 = new TGroupCompany();
        c1.setId(100);
        c1.setName("A公司");
        TGroupCompany c2 = new TGroupCompany();
        c2.setId(200);
        c2.setName("B公司");
        Mockito.when(groupCompanyMapper.selectBatchIds(any())).thenReturn(Arrays.asList(c1, c2));

        SysUserClient userClient = Mockito.mock(SysUserClient.class);
        SysUser u1 = new SysUser();
        u1.setUserId(9001L);
        u1.setUserName("张三");
        SysUser u2 = new SysUser();
        u2.setUserId(9002L);
        u2.setUserName("李四");
        Mockito.when(userClient.getSysUser(9001L)).thenReturn(R.ok(u1));
        Mockito.when(userClient.getSysUser(9002L)).thenReturn(R.ok(u2));

        CompanyAccountAccessService access = Mockito.mock(CompanyAccountAccessService.class);
        // 当前请求携带 companyId=null，且当前账号非公司账号 → resolveCompanyFilter 返回 null（不限）
        Mockito.when(access.resolveCompanyFilter(null)).thenReturn(null);

        TCompanyRechargeRecordServiceImpl service = new TCompanyRechargeRecordServiceImpl();
        ReflectionTestUtils.setField(service, "baseMapper", mapper);
        ReflectionTestUtils.setField(service, "groupCompanyMapper", groupCompanyMapper);
        ReflectionTestUtils.setField(service, "sysUserClient", userClient);
        ReflectionTestUtils.setField(service, "companyAccountAccessService", access);

        CompanyRechargeRecordQueryDTO query = new CompanyRechargeRecordQueryDTO();
        query.setCurrent(1);
        query.setSize(10);
        com.ruoyi.common.core.web.page.PageInfo<TCompanyRechargeRecord> page = service.pageList(query);

        assertEquals(2L, page.getTotal());
        assertEquals(2, page.getRecords().size());
        assertEquals("A公司", page.getRecords().get(0).getCompanyName());
        assertEquals("张三", page.getRecords().get(0).getOperatorName());
        assertEquals("B公司", page.getRecords().get(1).getCompanyName());
        assertEquals("李四", page.getRecords().get(1).getOperatorName());
    }

    @Test
    @SuppressWarnings("unchecked")
    void selectExportListMapsToExcelDto() {
        TCompanyRechargeRecord r1 = new TCompanyRechargeRecord();
        r1.setId(1);
        r1.setSerialNo("CR20260930101500001");
        r1.setCompanyId(100);
        r1.setPayAmount(new BigDecimal("100"));
        r1.setGiftAmount(new BigDecimal("20"));
        r1.setArrivalAmount(new BigDecimal("120"));
        r1.setStatus(1);
        r1.setOperatorId(9001L);
        r1.setCompanyName("A公司");
        r1.setOperatorName("张三");
        r1.setRechargeTime(LocalDateTime.of(2026, 9, 30, 10, 15, 0));

        TCompanyRechargeRecordMapper mapper = Mockito.mock(TCompanyRechargeRecordMapper.class);
        Mockito.when(mapper.selectList(any())).thenReturn(Collections.singletonList(r1));

        // fillNames 会重新覆盖 companyName/operatorName，必须 stub 远程查询
        TGroupCompanyMapper groupCompanyMapper = Mockito.mock(TGroupCompanyMapper.class);
        TGroupCompany c1 = new TGroupCompany();
        c1.setId(100);
        c1.setName("A公司");
        Mockito.when(groupCompanyMapper.selectBatchIds(any())).thenReturn(Collections.singletonList(c1));
        SysUserClient userClient = Mockito.mock(SysUserClient.class);
        SysUser u1 = new SysUser();
        u1.setUserId(9001L);
        u1.setUserName("张三");
        Mockito.when(userClient.getSysUser(9001L)).thenReturn(R.ok(u1));

        TCompanyRechargeRecordServiceImpl service = new TCompanyRechargeRecordServiceImpl();
        ReflectionTestUtils.setField(service, "baseMapper", mapper);
        ReflectionTestUtils.setField(service, "groupCompanyMapper", groupCompanyMapper);
        ReflectionTestUtils.setField(service, "sysUserClient", userClient);
        ReflectionTestUtils.setField(service, "companyAccountAccessService",
                Mockito.mock(CompanyAccountAccessService.class));

        CompanyRechargeRecordQueryDTO query = new CompanyRechargeRecordQueryDTO();
        query.setStatus(1);
        List<CompanyRechargeRecordExcelDTO> list = service.selectExportList(query);

        assertEquals(1, list.size());
        CompanyRechargeRecordExcelDTO dto = list.get(0);
        assertEquals("CR20260930101500001", dto.getSerialNo());
        assertEquals("A公司", dto.getCompanyName());
        assertEquals(new BigDecimal("120"), dto.getArrivalAmount());
        assertEquals(1, dto.getStatus());
        assertEquals("张三", dto.getOperatorName());
        assertNotEquals(null, dto.getRechargeTime());
    }
}