package com.sinata.appointment.service;
|
|
import static org.assertj.core.api.Assertions.assertThat;
|
import static org.mockito.ArgumentMatchers.any;
|
import static org.mockito.Mockito.mock;
|
import static org.mockito.Mockito.when;
|
|
import com.baomidou.mybatisplus.core.conditions.query.LambdaQueryWrapper;
|
import com.baomidou.mybatisplus.extension.plugins.pagination.Page;
|
import com.fasterxml.jackson.databind.ObjectMapper;
|
import com.sinata.appointment.entity.AppointmentOrder;
|
import com.sinata.appointment.entity.MpUser;
|
import com.sinata.appointment.mapper.AppointmentOrderMapper;
|
import com.sinata.appointment.mapper.AppointmentVehicleMapper;
|
import com.sinata.appointment.mapper.AppointmentVisitorMapper;
|
import com.sinata.appointment.mapper.ApprovalInstanceMapper;
|
import com.sinata.appointment.mapper.ApprovalTaskMapper;
|
import com.sinata.appointment.mapper.BizVisitReasonMapper;
|
import com.sinata.appointment.mapper.MpPassAccessAreaMapper;
|
import com.sinata.appointment.mapper.MpUserMapper;
|
import com.sinata.appointment.mapper.MpVisitorMapper;
|
import com.sinata.appointment.mapper.SysDeptMapper;
|
import jakarta.servlet.http.HttpServletResponse;
|
import java.util.List;
|
import org.junit.jupiter.api.Test;
|
import org.springframework.mock.web.MockHttpServletResponse;
|
|
class ExportStreamingTest {
|
|
@Test
|
void appointmentExportWritesWorkbookToResponseStream() throws Exception {
|
AppointmentOrderMapper appointmentOrderMapper = mock(AppointmentOrderMapper.class);
|
when(appointmentOrderMapper.pageAdminOrders(any(Page.class), any(), any(), any(), any(), any(), any(), any(), any(), any(), any(), any()))
|
.thenReturn(new Page<AppointmentOrder>(1, 200));
|
AppointmentApplicationService service =
|
new AppointmentApplicationService(
|
appointmentOrderMapper,
|
mock(AppointmentVehicleMapper.class),
|
mock(AppointmentVisitorMapper.class),
|
mock(ApprovalInstanceMapper.class),
|
mock(ApprovalTaskMapper.class),
|
mock(BizVisitReasonMapper.class),
|
mock(MpUserMapper.class),
|
mock(MpVisitorMapper.class),
|
mock(MpPassAccessAreaMapper.class),
|
mock(SysDeptMapper.class),
|
new ObjectMapper());
|
MockHttpServletResponse response = new MockHttpServletResponse();
|
|
service.exportExcel(response, null, null, null, null, null, null, null, null, null, null, null);
|
|
assertWorkbookResponse(response, "appointments.xlsx");
|
}
|
|
@Test
|
void visitorExportWritesWorkbookToResponseStream() throws Exception {
|
MpUserMapper mpUserMapper = mock(MpUserMapper.class);
|
when(mpUserMapper.selectPage(any(Page.class), any(LambdaQueryWrapper.class)))
|
.thenReturn(new Page<MpUser>(1, 500));
|
VisitorUserManageApplicationService service =
|
new VisitorUserManageApplicationService(
|
mpUserMapper,
|
mock(com.sinata.appointment.mapper.AppointmentOrderMapper.class));
|
MockHttpServletResponse response = new MockHttpServletResponse();
|
|
service.exportExcel(response, null, null, null, null, null, null);
|
|
assertWorkbookResponse(response, "visitor-users.xlsx");
|
}
|
|
private void assertWorkbookResponse(MockHttpServletResponse response, String filename) {
|
assertThat(response.getContentType())
|
.contains("application/vnd.openxmlformats-officedocument.spreadsheetml.sheet");
|
assertThat(response.getHeader("Content-Disposition")).contains(filename);
|
assertThat(response.getContentAsByteArray()).startsWith(new byte[] {'P', 'K'});
|
}
|
}
|
