package com.estate.expert.admin.io.processor.impl;
|
|
import com.estate.expert.admin.TestCommon;
|
import com.estate.expert.admin.io.processor.impl.business.DeviceProcessor;
|
import com.estate.expert.admin.io.processor.impl.business.PlanProcessor;
|
import com.estate.expert.admin.io.processor.impl.business.ProjectProcessor;
|
import com.estate.expert.admin.io.processor.impl.record.DeviceErrorRecordProcessor;
|
import com.estate.expert.admin.io.processor.impl.record.SystemErrorRecordProcessor;
|
import com.estate.expert.admin.io.processor.impl.record.WarningRecordProcessor;
|
import com.estate.expert.admin.service.*;
|
import lombok.extern.slf4j.Slf4j;
|
import org.junit.jupiter.api.Test;
|
import org.junit.jupiter.api.extension.ExtendWith;
|
import org.springframework.boot.test.context.SpringBootTest;
|
import org.springframework.test.context.junit.jupiter.SpringExtension;
|
import org.springframework.transaction.annotation.Transactional;
|
|
import javax.annotation.Resource;
|
import java.io.FileOutputStream;
|
import java.io.IOException;
|
|
@Slf4j
|
@SpringBootTest
|
@ExtendWith(SpringExtension.class)
|
@Transactional
|
class ExportProcessorTest {
|
|
@Resource
|
private ClientService clientService;
|
|
@Resource
|
private ErrorRecordService errorRecordService;
|
|
@Resource
|
private WarningRecordService warningRecordService;
|
|
@Resource
|
private ProjectService projectService;
|
|
@Resource
|
private DeviceService deviceService;
|
|
@Resource
|
private PlanService planService;
|
|
@Resource
|
private CheckTaskService checkTaskService;
|
|
@Test
|
void systemErrors() throws IOException {
|
var records = errorRecordService.listSystemErrors(1);
|
|
var processor = new SystemErrorRecordProcessor();
|
// write to file
|
try(var fos = new FileOutputStream(TestCommon.getTmpPath("system_errors.xlsx"))) {
|
processor.saveWorkbook(fos, records);
|
}
|
}
|
|
@Test
|
void deviceErrors() throws IOException {
|
var records = errorRecordService.listDeviceErrors(1);
|
|
var processor = new DeviceErrorRecordProcessor();
|
// write to file
|
try(var fos = new FileOutputStream(TestCommon.getTmpPath("device_errors.xlsx"))) {
|
processor.saveWorkbook(fos, records);
|
}
|
}
|
|
@Test
|
void warningRecords() throws IOException {
|
var records = warningRecordService.findAll();
|
|
var processor = new WarningRecordProcessor();
|
// write to file
|
try(var fos = new FileOutputStream(TestCommon.getTmpPath("warning_records.xlsx"))) {
|
processor.saveWorkbook(fos, records);
|
}
|
}
|
|
@Test
|
void projects() throws IOException {
|
var projects = projectService.findAll();
|
|
var processor = new ProjectProcessor(clientService, projectService);
|
try(var fos = new FileOutputStream(TestCommon.getTmpPath("projects.xlsx"))) {
|
processor.saveWorkbook(fos, projects);
|
}
|
}
|
|
@Test
|
void devices() throws IOException {
|
var devices = deviceService.findAll();
|
|
var processor = new DeviceProcessor();
|
try(var fos = new FileOutputStream(TestCommon.getTmpPath("devices.xlsx"))) {
|
processor.saveWorkbook(fos, devices);
|
}
|
}
|
|
@Test
|
void plans() throws IOException {
|
var plans = planService.findAll();
|
|
var processor = new PlanProcessor(checkTaskService);
|
try(var fos = new FileOutputStream(TestCommon.getTmpPath("plans.xlsx"))) {
|
processor.saveWorkbook(fos, plans);
|
}
|
}
|
}
|
