package com.flightfeather.uav.common.utils;
|
|
import com.flightfeather.uav.common.exception.BizException;
|
import com.flightfeather.uav.domain.entity.GridDataDetail;
|
import com.flightfeather.uav.lightshare.service.SatelliteTelemetryService;
|
import org.apache.commons.fileupload.disk.DiskFileItem;
|
import org.junit.Test;
|
import org.junit.runner.RunWith;
|
import org.springframework.beans.factory.annotation.Autowired;
|
import org.springframework.boot.test.context.SpringBootTest;
|
import org.springframework.mock.web.MockMultipartFile;
|
import org.springframework.test.context.junit4.SpringRunner;
|
import org.springframework.web.multipart.commons.CommonsMultipartFile;
|
|
import java.io.File;
|
import java.io.FileInputStream;
|
import java.io.IOException;
|
import java.net.URL;
|
import java.time.LocalDateTime;
|
import java.util.List;
|
import java.util.Objects;
|
import java.util.function.Function;
|
|
import static org.junit.Assert.assertEquals;
|
|
@RunWith(SpringRunner.class)
|
@SpringBootTest
|
public class FileExchangeTest {
|
|
@Autowired
|
private SatelliteTelemetryService satelliteTelemetryService;
|
|
public void start() throws BizException, IOException {
|
test1();
|
test2();
|
test3();
|
test4();
|
test5();
|
test6();
|
test7();
|
}
|
|
// 第一列数字类型错误
|
@Test
|
public void test1() throws IOException, BizException {
|
URL resource = getClass().getClassLoader().getResource("templates/GridData-1_type_error.xlsx");
|
if (resource != null) {
|
System.out.println(resource.getPath());
|
// 创建文件输入流
|
FileInputStream inputStream = new FileInputStream(new File(resource.getPath()));
|
// 创建MockMultipartFile对象
|
MockMultipartFile multipartFile = new MockMultipartFile("file", "111.xlsx", "application/vnd.openxmlformats-officedocument.spreadsheetml.sheet", inputStream);
|
try {
|
satelliteTelemetryService.importGridData(1, 1, LocalDateTime.of(2029, 7, 29, 0, 0), 0, multipartFile);
|
}catch (BizException e) {
|
// 验证异常消息是否与预期相符
|
if (!e.getMessage().contains("不是数字类型")) {
|
System.out.println(e.getMessage());
|
throw e;
|
}
|
}
|
} else {
|
System.out.println("资源文件未找到");
|
}
|
}
|
// 第二列数字类型错误
|
@Test
|
public void test2() throws IOException, BizException {
|
URL resource = getClass().getClassLoader().getResource("templates/GridData-2_type_error.xlsx");
|
if (resource != null) {
|
System.out.println(resource.getPath());
|
// 创建文件输入流
|
FileInputStream inputStream = new FileInputStream(new File(resource.getPath()));
|
// 创建MockMultipartFile对象
|
MockMultipartFile multipartFile = new MockMultipartFile("file", "111.xlsx", "application/vnd.openxmlformats-officedocument.spreadsheetml.sheet", inputStream);
|
try {
|
satelliteTelemetryService.importGridData(1, 1, LocalDateTime.of(2029, 7, 29, 0, 0), 0, multipartFile);
|
}catch (BizException e) {
|
// 验证异常消息是否与预期相符
|
if (!e.getMessage().contains("不是数字类型")) {
|
System.out.println(e.getMessage());
|
throw e;
|
}
|
}
|
} else {
|
System.out.println("资源文件未找到");
|
}
|
}
|
// 网格编号越界错误
|
@Test
|
public void test3() throws IOException, BizException {
|
URL resource = getClass().getClassLoader().getResource("templates/GridData-index_out.xlsx");
|
if (resource != null) {
|
System.out.println(resource.getPath());
|
// 创建文件输入流
|
FileInputStream inputStream = new FileInputStream(new File(resource.getPath()));
|
// 创建MockMultipartFile对象
|
MockMultipartFile multipartFile = new MockMultipartFile("file", "111.xlsx", "application/vnd.openxmlformats-officedocument.spreadsheetml.sheet", inputStream);
|
try {
|
satelliteTelemetryService.importGridData(1, 1, LocalDateTime.of(2029, 7, 29, 0, 0), 0, multipartFile);
|
}catch (BizException e) {
|
// 验证异常消息是否与预期相符
|
if (!e.getMessage().contains("导入数据中有多余网格单元格")) {
|
System.out.println(e.getMessage());
|
throw e;
|
}
|
}
|
} else {
|
System.out.println("资源文件未找到");
|
}
|
}
|
// 网格编号缺失错误
|
@Test
|
public void test4() throws IOException, BizException {
|
URL resource = getClass().getClassLoader().getResource("templates/GridData-index_short.xlsx");
|
if (resource != null) {
|
System.out.println(resource.getPath());
|
// 创建文件输入流
|
FileInputStream inputStream = new FileInputStream(new File(resource.getPath()));
|
// 创建MockMultipartFile对象
|
MockMultipartFile multipartFile = new MockMultipartFile("file", "111.xlsx", "application/vnd.openxmlformats-officedocument.spreadsheetml.sheet", inputStream);
|
try {
|
satelliteTelemetryService.importGridData(1, 1, LocalDateTime.of(2029, 7, 29, 0, 0), 0, multipartFile);
|
}catch (BizException e) {
|
// 验证异常消息是否与预期相符
|
if (!e.getMessage().contains("导入数据中缺少以下网格单元格")) {
|
System.out.println(e.getMessage());
|
throw e;
|
}
|
}
|
} else {
|
System.out.println("资源文件未找到");
|
}
|
}
|
// 文件类型错误
|
@Test
|
public void test5() throws IOException, BizException {
|
URL resource = getClass().getClassLoader().getResource("templates/GridData-fileType-error.txt");
|
if (resource != null) {
|
System.out.println(resource.getPath());
|
// 创建文件输入流
|
FileInputStream inputStream = new FileInputStream(new File(resource.getPath()));
|
// 创建MockMultipartFile对象
|
MockMultipartFile multipartFile = new MockMultipartFile("file", "111.xlsx", "application/vnd.openxmlformats-officedocument.spreadsheetml.sheet", inputStream);
|
try {
|
satelliteTelemetryService.importGridData(1, 1, LocalDateTime.of(2029, 7, 29, 0, 0), 0, multipartFile);
|
}catch (BizException e) {
|
// 验证异常消息是否与预期相符
|
if (!e.getMessage().contains("导入数据中缺少以下网格单元格")) {
|
System.out.println(e.getMessage());
|
throw e;
|
}
|
}
|
} else {
|
System.out.println("资源文件未找到");
|
}
|
}
|
// 新增数据导入成功
|
@Test
|
public void test6() throws IOException, BizException {
|
URL resource = getClass().getClassLoader().getResource("templates/GridData-success-insert.xlsx");
|
if (resource != null) {
|
System.out.println(resource.getPath());
|
// 创建文件输入流
|
FileInputStream inputStream = new FileInputStream(new File(resource.getPath()));
|
// 创建MockMultipartFile对象
|
MockMultipartFile multipartFile = new MockMultipartFile("file", "111.xlsx", "application/vnd.openxmlformats-officedocument.spreadsheetml.sheet", inputStream);
|
try {
|
satelliteTelemetryService.importGridData(1, 1,
|
LocalDateTime.of(2029, 7, 29, 0, 0), 0, multipartFile);
|
}catch (BizException e) {
|
throw e;
|
}
|
} else {
|
System.out.println("资源文件未找到");
|
}
|
}
|
// 更新数据导入成功
|
@Test
|
public void test7() throws IOException, BizException {
|
URL resource = getClass().getClassLoader().getResource("templates/GridData-success-update.xlsx");
|
if (resource != null) {
|
System.out.println(resource.getPath());
|
// 创建文件输入流
|
FileInputStream inputStream = new FileInputStream(new File(resource.getPath()));
|
// 创建MockMultipartFile对象
|
MockMultipartFile multipartFile = new MockMultipartFile("file", "111.xlsx", "application/vnd.openxmlformats-officedocument.spreadsheetml.sheet", inputStream);
|
try {
|
satelliteTelemetryService.importGridData(1, 1,
|
LocalDateTime.of(2029, 7, 29, 0, 0), 1, multipartFile);
|
}catch (BizException e) {
|
throw e;
|
}
|
} else {
|
System.out.println("资源文件未找到");
|
}
|
}
|
}
|
