package cn.iocoder.yudao.module.mes.service.pro.qrexport;
|
|
import cn.iocoder.yudao.module.mes.dal.dataobject.pro.qrexport.MesProQrExportTaskDO;
|
import cn.iocoder.yudao.module.mes.dal.mysql.pro.qrexport.MesProQrExportTaskMapper;
|
import cn.iocoder.yudao.module.mes.enums.pro.MesProQrExportTaskStatusEnum;
|
import org.junit.jupiter.api.Test;
|
import org.junit.jupiter.api.io.TempDir;
|
import org.springframework.mock.web.MockHttpServletResponse;
|
import org.springframework.test.util.ReflectionTestUtils;
|
|
import java.nio.file.Files;
|
import java.nio.file.Path;
|
import java.lang.reflect.Proxy;
|
|
import static org.junit.jupiter.api.Assertions.assertArrayEquals;
|
import static org.junit.jupiter.api.Assertions.assertEquals;
|
import static org.junit.jupiter.api.Assertions.assertTrue;
|
|
class MesProQrExportTaskDownloadTest {
|
|
@TempDir
|
Path tempDir;
|
|
@Test
|
void rangeDownloadReturnsExactBytesAndKeepsFileForRetry() throws Exception {
|
byte[] content = "0123456789".getBytes();
|
Path file = tempDir.resolve("export.zip");
|
Files.write(file, content);
|
MesProQrExportTaskDO task = MesProQrExportTaskDO.builder()
|
.id(13L).taskName("test").status(MesProQrExportTaskStatusEnum.FINISHED.getStatus())
|
.filePath(file.toString()).build();
|
MesProQrExportTaskMapper mapper = (MesProQrExportTaskMapper) Proxy.newProxyInstance(
|
getClass().getClassLoader(), new Class[]{MesProQrExportTaskMapper.class},
|
(proxy, method, args) -> {
|
if (method.getName().equals("selectById")) {
|
return task;
|
}
|
throw new UnsupportedOperationException(method.getName());
|
});
|
MesProQrExportTaskServiceImpl service = new MesProQrExportTaskServiceImpl();
|
ReflectionTestUtils.setField(service, "taskMapper", mapper);
|
|
MockHttpServletResponse partial = new MockHttpServletResponse();
|
service.download(13L, "bytes=3-6", partial);
|
assertEquals(206, partial.getStatus());
|
assertEquals("bytes 3-6/10", partial.getHeader("Content-Range"));
|
assertEquals("4", partial.getHeader("Content-Length"));
|
assertArrayEquals("3456".getBytes(), partial.getContentAsByteArray());
|
|
MockHttpServletResponse invalid = new MockHttpServletResponse();
|
service.download(13L, "bytes=10-11", invalid);
|
assertEquals(416, invalid.getStatus());
|
assertEquals("bytes */10", invalid.getHeader("Content-Range"));
|
|
MockHttpServletResponse full = new MockHttpServletResponse();
|
service.download(13L, null, full);
|
assertEquals(200, full.getStatus());
|
assertArrayEquals(content, full.getContentAsByteArray());
|
assertTrue(Files.exists(file));
|
}
|
|
}
|
