package com.tcm.training.admin.member.service;

import cn.idev.excel.FastExcel;
import com.tcm.training.admin.agent.service.ProvincialAgentAppService;
import com.tcm.training.admin.agent.vo.ProvincialAgentListItemVO;
import com.tcm.training.admin.common.PageResponse;
import org.junit.jupiter.api.Test;
import org.springframework.mock.web.MockHttpServletResponse;

import java.io.ByteArrayInputStream;
import java.util.Arrays;
import java.util.Collections;
import java.util.List;
import java.util.Map;

import static org.junit.jupiter.api.Assertions.assertEquals;
import static org.mockito.Mockito.doReturn;
import static org.mockito.Mockito.spy;
import static org.mockito.Mockito.verify;

class ProvincialAgentExportTest {
    private final ProvincialAgentAppService service =
            spy(new ProvincialAgentAppService(null, null, null, null, null));

    @Test
    void shouldExportAllBatchesWithFiltersAndPlainPhone() throws Exception {
        ProvincialAgentListItemVO item = new ProvincialAgentListItemVO();
        item.setAgentName("省代测试");
        item.setContactInfo("13800138000");
        item.setRegionNames(Arrays.asList("广东省", "湖南省"));
        item.setAddress("测试路88号");
        item.setStatus(1);
        item.setClinicCount(3L);
        doReturn(PageResponse.of(Collections.nCopies(200, item), 201, 1, 200))
                .when(service).page(1, 200, "44", "诊所", "138", 1);
        doReturn(PageResponse.of(Collections.singletonList(item), 201, 2, 200))
                .when(service).page(2, 200, "44", "诊所", "138", 1);

        MockHttpServletResponse response = new MockHttpServletResponse();
        service.export("44", "诊所", "138", 1, response);
        List<Map<Integer, String>> rows = readSheet(response);

        assertEquals(202, rows.size());
        assertEquals("负责人", rows.get(0).get(1));
        assertEquals("详细地址", rows.get(0).get(4));
        assertEquals("13800138000", rows.get(1).get(2));
        assertEquals("广东省、湖南省", rows.get(1).get(3));
        assertEquals("测试路88号", rows.get(1).get(4));
        assertEquals("3", rows.get(1).get(5));
        assertEquals("启用", rows.get(1).get(6));
        assertEquals("201", rows.get(201).get(0));
        verify(service).page(2, 200, "44", "诊所", "138", 1);
    }

    @Test
    void shouldExportHeadersWhenNoMatches() throws Exception {
        doReturn(PageResponse.of(Collections.emptyList(), 0, 1, 200))
                .when(service).page(1, 200, null, null, null, null);
        MockHttpServletResponse response = new MockHttpServletResponse();

        service.export(null, null, null, null, response);

        List<Map<Integer, String>> rows = readSheet(response);
        assertEquals(1, rows.size());
        assertEquals(8, rows.get(0).size());
    }

    private List<Map<Integer, String>> readSheet(MockHttpServletResponse response) {
        return FastExcel.read(new ByteArrayInputStream(response.getContentAsByteArray()))
                .headRowNumber(0).sheet().doReadSync();
    }
}