mirror of https://github.com/alibaba/easyexcel
Jiaju Zhuang
3 years ago
32 changed files with 278 additions and 106 deletions
@ -0,0 +1,17 @@
|
||||
package com.alibaba.easyexcel.test.core.charset; |
||||
|
||||
import com.alibaba.excel.annotation.ExcelProperty; |
||||
|
||||
import lombok.EqualsAndHashCode; |
||||
import lombok.Getter; |
||||
import lombok.Setter; |
||||
|
||||
@Getter |
||||
@Setter |
||||
@EqualsAndHashCode |
||||
public class CharsetData { |
||||
@ExcelProperty("姓名") |
||||
private String name; |
||||
@ExcelProperty("年纪") |
||||
private Integer age; |
||||
} |
||||
@ -0,0 +1,119 @@
|
||||
package com.alibaba.easyexcel.test.core.charset; |
||||
|
||||
import java.io.File; |
||||
import java.nio.charset.Charset; |
||||
import java.nio.charset.StandardCharsets; |
||||
import java.util.ArrayList; |
||||
import java.util.List; |
||||
import java.util.Map; |
||||
|
||||
import com.alibaba.easyexcel.test.core.simple.SimpleData; |
||||
import com.alibaba.easyexcel.test.core.simple.SimpleDataListener; |
||||
import com.alibaba.easyexcel.test.demo.read.DemoData; |
||||
import com.alibaba.easyexcel.test.util.TestFileUtil; |
||||
import com.alibaba.excel.EasyExcel; |
||||
import com.alibaba.excel.context.AnalysisContext; |
||||
import com.alibaba.excel.metadata.data.ReadCellData; |
||||
import com.alibaba.excel.read.listener.PageReadListener; |
||||
import com.alibaba.excel.read.listener.ReadListener; |
||||
import com.alibaba.excel.util.ConverterUtils; |
||||
import com.alibaba.excel.util.ListUtils; |
||||
import com.alibaba.fastjson.JSON; |
||||
|
||||
import lombok.extern.slf4j.Slf4j; |
||||
import org.apache.commons.compress.utils.Charsets; |
||||
import org.apache.commons.compress.utils.Lists; |
||||
import org.junit.Assert; |
||||
import org.junit.BeforeClass; |
||||
import org.junit.FixMethodOrder; |
||||
import org.junit.Test; |
||||
import org.junit.runners.MethodSorters; |
||||
|
||||
/** |
||||
* charset |
||||
* |
||||
* @author Jiaju Zhuang |
||||
*/ |
||||
@Slf4j |
||||
@FixMethodOrder(MethodSorters.NAME_ASCENDING) |
||||
public class CharsetDataTest { |
||||
private static final Charset GBK = Charset.forName("GBK"); |
||||
private static File fileCsvGbk; |
||||
private static File fileCsvUtf8; |
||||
private static File fileCsvError; |
||||
|
||||
@BeforeClass |
||||
public static void init() { |
||||
fileCsvGbk = TestFileUtil.createNewFile("charset" + File.separator + "fileCsvGbk.csv"); |
||||
fileCsvUtf8 = TestFileUtil.createNewFile("charset" + File.separator + "fileCsvUtf8.csv"); |
||||
fileCsvError = TestFileUtil.createNewFile("charset" + File.separator + "fileCsvError.csv"); |
||||
} |
||||
|
||||
@Test |
||||
public void t01ReadAndWriteCsv() { |
||||
readAndWrite(fileCsvGbk, GBK); |
||||
readAndWrite(fileCsvUtf8, StandardCharsets.UTF_8); |
||||
} |
||||
|
||||
@Test |
||||
public void t02ReadAndWriteCsvError() { |
||||
EasyExcel.write(fileCsvError, CharsetData.class).charset(GBK).sheet().doWrite(data()); |
||||
EasyExcel.read(fileCsvError, CharsetData.class, new ReadListener<CharsetData>() { |
||||
|
||||
private final List<CharsetData> dataList = Lists.newArrayList(); |
||||
|
||||
@Override |
||||
public void invokeHead(Map<Integer, ReadCellData<?>> headMap, AnalysisContext context) { |
||||
String head = headMap.get(0).getStringValue(); |
||||
Assert.assertNotEquals("姓名", head); |
||||
} |
||||
|
||||
@Override |
||||
public void invoke(CharsetData data, AnalysisContext context) { |
||||
dataList.add(data); |
||||
} |
||||
|
||||
@Override |
||||
public void doAfterAllAnalysed(AnalysisContext context) { |
||||
} |
||||
}).charset(StandardCharsets.UTF_8).sheet().doRead(); |
||||
} |
||||
|
||||
private void readAndWrite(File file, Charset charset) { |
||||
EasyExcel.write(file, CharsetData.class).charset(charset).sheet().doWrite(data()); |
||||
EasyExcel.read(file, CharsetData.class, new ReadListener<CharsetData>() { |
||||
|
||||
private final List<CharsetData> dataList = Lists.newArrayList(); |
||||
|
||||
@Override |
||||
public void invokeHead(Map<Integer, ReadCellData<?>> headMap, AnalysisContext context) { |
||||
String head = headMap.get(0).getStringValue(); |
||||
Assert.assertEquals("姓名", head); |
||||
} |
||||
|
||||
@Override |
||||
public void invoke(CharsetData data, AnalysisContext context) { |
||||
dataList.add(data); |
||||
} |
||||
|
||||
@Override |
||||
public void doAfterAllAnalysed(AnalysisContext context) { |
||||
Assert.assertEquals(dataList.size(), 10); |
||||
CharsetData charsetData = dataList.get(0); |
||||
Assert.assertEquals("姓名0", charsetData.getName()); |
||||
Assert.assertEquals(0, (long)charsetData.getAge()); |
||||
} |
||||
}).charset(charset).sheet().doRead(); |
||||
} |
||||
|
||||
private List<CharsetData> data() { |
||||
List<CharsetData> list = Lists.newArrayList(); |
||||
for (int i = 0; i < 10; i++) { |
||||
CharsetData data = new CharsetData(); |
||||
data.setName("姓名" + i); |
||||
data.setAge(i); |
||||
list.add(data); |
||||
} |
||||
return list; |
||||
} |
||||
} |
||||
Loading…
Reference in new issue