forked from fanruan/easyexcel
Jiaju Zhuang
4 years ago
committed by
GitHub
5 changed files with 109 additions and 5 deletions
@ -0,0 +1,15 @@
|
||||
package com.alibaba.easyexcel.test.core.skip; |
||||
|
||||
import com.alibaba.excel.annotation.ExcelProperty; |
||||
|
||||
import lombok.Data; |
||||
|
||||
/** |
||||
* @author Jiaju Zhuang |
||||
*/ |
||||
@Data |
||||
public class SkipData { |
||||
|
||||
@ExcelProperty("姓名") |
||||
private String name; |
||||
} |
@ -0,0 +1,86 @@
|
||||
package com.alibaba.easyexcel.test.core.skip; |
||||
|
||||
import java.io.File; |
||||
import java.util.ArrayList; |
||||
import java.util.List; |
||||
|
||||
import org.junit.Assert; |
||||
import org.junit.BeforeClass; |
||||
import org.junit.FixMethodOrder; |
||||
import org.junit.Test; |
||||
import org.junit.runners.MethodSorters; |
||||
|
||||
import com.alibaba.easyexcel.test.core.simple.SimpleData; |
||||
import com.alibaba.easyexcel.test.core.simple.SimpleDataListener; |
||||
import com.alibaba.easyexcel.test.core.simple.SimpleDataSheetNameListener; |
||||
import com.alibaba.easyexcel.test.util.TestFileUtil; |
||||
import com.alibaba.excel.EasyExcel; |
||||
import com.alibaba.excel.ExcelReader; |
||||
import com.alibaba.excel.ExcelWriter; |
||||
import com.alibaba.excel.event.SyncReadListener; |
||||
import com.alibaba.excel.read.metadata.ReadSheet; |
||||
import com.alibaba.excel.write.metadata.WriteSheet; |
||||
|
||||
/** |
||||
* @author Jiaju Zhuang |
||||
*/ |
||||
@FixMethodOrder(MethodSorters.NAME_ASCENDING) |
||||
public class SkipDataTest { |
||||
|
||||
private static File file07; |
||||
private static File file03; |
||||
|
||||
@BeforeClass |
||||
public static void init() { |
||||
file07 = TestFileUtil.createNewFile("skip.xlsx"); |
||||
file03 = TestFileUtil.createNewFile("skip.xls"); |
||||
} |
||||
|
||||
@Test |
||||
public void t01ReadAndWrite07() { |
||||
readAndWrite(file07); |
||||
} |
||||
|
||||
@Test |
||||
public void t02ReadAndWrite03() { |
||||
readAndWrite(file03); |
||||
} |
||||
|
||||
private void readAndWrite(File file) { |
||||
ExcelWriter excelWriter = EasyExcel.write(file, SimpleData.class).build(); |
||||
WriteSheet writeSheet0 = EasyExcel.writerSheet(0, "第一个").build(); |
||||
WriteSheet writeSheet1 = EasyExcel.writerSheet(1, "第二个").build(); |
||||
WriteSheet writeSheet2 = EasyExcel.writerSheet(2, "第三个").build(); |
||||
WriteSheet writeSheet3 = EasyExcel.writerSheet(3, "第四个").build(); |
||||
excelWriter.write(data("name1"), writeSheet0); |
||||
excelWriter.write(data("name2"), writeSheet1); |
||||
excelWriter.write(data("name3"), writeSheet2); |
||||
excelWriter.write(data("name4"), writeSheet3); |
||||
excelWriter.finish(); |
||||
|
||||
List<SkipData> list = EasyExcel.read(file, SkipData.class, null).sheet("第二个").doReadSync(); |
||||
Assert.assertEquals(1, list.size()); |
||||
Assert.assertEquals("name2", list.get(0).getName()); |
||||
|
||||
SyncReadListener syncReadListener = new SyncReadListener(); |
||||
ExcelReader excelReader = EasyExcel.read(file, SkipData.class, null).registerReadListener(syncReadListener) |
||||
.build(); |
||||
ReadSheet readSheet1 = EasyExcel.readSheet("第二个").build(); |
||||
ReadSheet readSheet3 = EasyExcel.readSheet("第四个").build(); |
||||
excelReader.read(readSheet1, readSheet3); |
||||
List<Object> syncList = syncReadListener.getList(); |
||||
Assert.assertEquals(2, syncList.size()); |
||||
Assert.assertEquals("name2", ((SkipData) syncList.get(0)).getName()); |
||||
Assert.assertEquals("name4", ((SkipData) syncList.get(1)).getName()); |
||||
excelReader.finish(); |
||||
} |
||||
|
||||
|
||||
private List<SkipData> data(String name) { |
||||
List<SkipData> list = new ArrayList<SkipData>(); |
||||
SkipData data = new SkipData(); |
||||
data.setName(name); |
||||
list.add(data); |
||||
return list; |
||||
} |
||||
} |
Loading…
Reference in new issue