diff --git a/src/main/java/com/example/parser/FlowableExcelParser.java b/src/main/java/com/example/parser/FlowableExcelParser.java index 2db2707..d56e090 100644 --- a/src/main/java/com/example/parser/FlowableExcelParser.java +++ b/src/main/java/com/example/parser/FlowableExcelParser.java @@ -32,6 +32,17 @@ public class FlowableExcelParser { } } + public ArrayList> parseExcelSheet(String excelPath) { + try (FileInputStream fileInputStream = new FileInputStream(excelPath)) { + Workbook workBook = new XSSFWorkbook(fileInputStream); + FormulaEvaluator formulaEvaluator = workBook.getCreationHelper().createFormulaEvaluator(); + Sheet workSheet = workBook.getSheetAt(0); + return parseSheet(workSheet, formulaEvaluator); + } catch (IOException e) { + throw new RuntimeException(e); + } + } + protected ArrayList>> parseBook(Workbook workBook ) { ArrayList>> book = new ArrayList<>(); FormulaEvaluator formulaEvaluator = workBook.getCreationHelper().createFormulaEvaluator(); diff --git a/src/test/java/com/example/parser/FlowableExcelParserTest.java b/src/test/java/com/example/parser/FlowableExcelParserTest.java index c641cf2..23c07a4 100644 --- a/src/test/java/com/example/parser/FlowableExcelParserTest.java +++ b/src/test/java/com/example/parser/FlowableExcelParserTest.java @@ -29,13 +29,17 @@ class FlowableExcelParserTest { } protected Stream getArgumentsFromExcel(String path) { - ArrayList>> book = flowableExcelParser.parseExcelBook(getResourcePath(path)); ArrayList argumentList = new ArrayList<>(); + ArrayList>> book = flowableExcelParser.parseExcelBook(getResourcePath(path)); for (ArrayList> sheet : book) { for (ArrayList row : sheet) { argumentList.add(Arguments.of(row)); } } + ArrayList> sheet = flowableExcelParser.parseExcelSheet(getResourcePath(path)); + for (ArrayList row : sheet) { + argumentList.add(Arguments.of(row)); + } return argumentList.stream(); }