iSharkFly-Docs
/
java-tutorials
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

update unit test

This commit is contained in:
hmdrzsharifi 2021-02-17 19:58:50 +03:30
parent 0f19017581
commit 365075cdd3
1 changed files with 15 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

21
apache-poi/src/test/java/com/baeldung/poi/excel/insert/ExcelInsertRowUnitTest.java
View File

@ -1,6 +1,8 @@
package com.baeldung.poi.excel.insert; package com.baeldung.poi.excel.insert;
import org.apache.poi.ss.usermodel.Sheet;
import org.apache.poi.ss.usermodel.Workbook; import org.apache.poi.ss.usermodel.Workbook;
import org.apache.poi.xssf.usermodel.XSSFWorkbook;
import org.junit.Before; import org.junit.Before;
import org.junit.Test; import org.junit.Test;
import org.junit.jupiter.api.Assertions; import org.junit.jupiter.api.Assertions;
@ -15,27 +17,34 @@ public class ExcelInsertRowUnitTest {
private static final String FILE_NAME = "test.xlsx"; private static final String FILE_NAME = "test.xlsx";
private static final String NEW_FILE_NAME = "new_test.xlsx"; private static final String NEW_FILE_NAME = "new_test.xlsx";
private String fileLocation; private String fileLocation;
private InsertRowHelper insertRowHelper;
@Before @Before
public void setup() throws URISyntaxException { public void setup() throws URISyntaxException {
fileLocation = Paths.get(ClassLoader.getSystemResource(FILE_NAME).toURI()).toString(); fileLocation = Paths.get(ClassLoader.getSystemResource(FILE_NAME).toURI()).toString();
insertRowHelper = new InsertRowHelper();
} }
@Test @Test
public void givenWorkbook_whenInsertRowBetween_thenRowCreated() throws IOException { public void givenWorkbook_whenInsertRowBetween_thenRowCreated() throws IOException {
Workbook newWorkbook = insertRowHelper.insertRowBetweenRows(fileLocation, 2, 1); int startRow = 2;
int rowNumber = 1;
Workbook workbook = new XSSFWorkbook(fileLocation);
Sheet sheet = workbook.getSheetAt(0);
int lastRow = sheet.getLastRowNum();
if (lastRow < startRow) {
sheet.createRow(startRow);
}
sheet.shiftRows(startRow, lastRow, rowNumber, true, true);
sheet.createRow(startRow);
FileOutputStream outputStream = new FileOutputStream(NEW_FILE_NAME); FileOutputStream outputStream = new FileOutputStream(NEW_FILE_NAME);
newWorkbook.write(outputStream); workbook.write(outputStream);
File file = new File(NEW_FILE_NAME); File file = new File(NEW_FILE_NAME);
final int expectedRowResult = 5; final int expectedRowResult = 5;
Assertions.assertEquals(expectedRowResult, newWorkbook.getSheetAt(0).getLastRowNum()); Assertions.assertEquals(expectedRowResult, workbook.getSheetAt(0).getLastRowNum());
outputStream.close(); outputStream.close();
file.delete(); file.delete();
newWorkbook.close(); workbook.close();
} }
} }