package cn.iocoder.yudao.module.mes.framework.qrcode;
|
|
import cn.hutool.extra.qrcode.QrCodeUtil;
|
import org.junit.jupiter.api.Test;
|
|
import javax.imageio.ImageIO;
|
import java.io.ByteArrayInputStream;
|
import java.io.ByteArrayOutputStream;
|
import java.util.zip.ZipInputStream;
|
import java.util.zip.ZipOutputStream;
|
|
import static org.junit.jupiter.api.Assertions.assertEquals;
|
import static org.junit.jupiter.api.Assertions.assertNotNull;
|
import static org.junit.jupiter.api.Assertions.assertNull;
|
|
class QrCodeImageUtilsZipTest {
|
|
@Test
|
void writesMultipleReadablePngEntries() throws Exception {
|
ByteArrayOutputStream bytes = new ByteArrayOutputStream();
|
try (ZipOutputStream zip = new ZipOutputStream(bytes)) {
|
QrCodeImageUtils.writeZipEntry(zip, QrLabelInfo.builder()
|
.fileName("1234567890123456").code("1234567890123456").build());
|
QrCodeImageUtils.writeZipEntry(zip, QrLabelInfo.builder()
|
.fileName("1234567890123457").code("1234567890123457").build());
|
}
|
|
try (ZipInputStream zip = new ZipInputStream(new ByteArrayInputStream(bytes.toByteArray()))) {
|
assertEquals("1234567890123456.png", zip.getNextEntry().getName());
|
var first = ImageIO.read(zip);
|
assertNotNull(first);
|
assertEquals("1234567890123456", QrCodeUtil.decode(first));
|
assertEquals("1234567890123457.png", zip.getNextEntry().getName());
|
var second = ImageIO.read(zip);
|
assertNotNull(second);
|
assertEquals("1234567890123457", QrCodeUtil.decode(second));
|
assertNull(zip.getNextEntry());
|
}
|
}
|
|
}
|
