package com.tcm.training.admin.member.service;
|
|
import cn.idev.excel.FastExcel;
|
import com.tcm.training.admin.member.mapper.ClinicQueryMapper;
|
import com.tcm.training.admin.member.row.ClinicListRow;
|
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.mock;
|
import static org.mockito.Mockito.when;
|
|
class ClinicExportTest {
|
|
@Test
|
void shouldExportAddressAgentAndPlainPhone() throws Exception {
|
ClinicListRow row = new ClinicListRow();
|
row.setPhone("13800138000");
|
row.setAddress("人民路88号");
|
row.setAgentName("测试省代");
|
|
List<Map<Integer, String>> sheet = exportSheet(row);
|
|
assertEquals(Arrays.asList("序号", "诊所", "电话", "诊所区域", "详细地址",
|
"省代", "绑定用户数", "状态", "注册时间"),
|
Arrays.asList(sheet.get(0).values().toArray(new String[0])));
|
assertEquals("13800138000", sheet.get(1).get(2));
|
assertEquals("人民路88号", sheet.get(1).get(4));
|
assertEquals("测试省代", sheet.get(1).get(5));
|
}
|
|
@Test
|
void shouldUsePlaceholderForMissingAddressAndAgent() throws Exception {
|
List<Map<Integer, String>> sheet = exportSheet(new ClinicListRow());
|
|
assertEquals("-", sheet.get(1).get(4));
|
assertEquals("-", sheet.get(1).get(5));
|
}
|
|
/** 读取实际生成的 Excel，同时验证表头顺序和字段映射。 */
|
private List<Map<Integer, String>> exportSheet(ClinicListRow row) throws Exception {
|
ClinicQueryMapper mapper = mock(ClinicQueryMapper.class);
|
when(mapper.selectClinicList(null, null, null, null, null, null, null))
|
.thenReturn(Collections.singletonList(row));
|
ClinicAppService service = new ClinicAppService(mapper, null, null, null, null, null);
|
MockHttpServletResponse response = new MockHttpServletResponse();
|
service.export(null, null, null, null, null, null, null, response);
|
return FastExcel.read(new ByteArrayInputStream(response.getContentAsByteArray()))
|
.headRowNumber(0).sheet().doReadSync();
|
}
|
}
|
