package technology.tabula; import java.io.BufferedReader; import java.io.File; import java.io.FileNotFoundException; import java.io.FileReader; import java.io.IOException; import java.util.List; import org.apache.pdfbox.pdmodel.PDDocument; import com.google.gson.Gson; import static org.junit.Assert.*; public class UtilsForTesting { public static Page getAreaFromFirstPage(String path, float top, float left, float bottom, float right) throws IOException { return getAreaFromPage(path, 1, top, left, bottom, right); } public static Page getAreaFromPage(String path, int page, float top, float left, float bottom, float right) throws IOException { return getPage(path, page).getArea(top, left, bottom, right); } public static Page getPage(String path, int pageNumber) throws IOException { ObjectExtractor oe = null; try { PDDocument document = PDDocument .load(path); oe = new ObjectExtractor(document); Page page = oe.extract(pageNumber); return page; } finally { if (oe != null) oe.close(); } } public static void assertTableEquals(Table table, String[][] arrayOfRows) { List> tableRows = table.getRows(); assertEquals(arrayOfRows.length, tableRows.size()); for (int i = 0; i < arrayOfRows.length; i++) { String[] row = arrayOfRows[i]; assertEquals(row.length, tableRows.get(i).size()); for (int j = 0; j < row.length; j++) { assertEquals(row[j].trim(), table.getCell(i, j).getText().trim()); } } } public static String loadJson(String path) throws IOException { BufferedReader reader = new BufferedReader( new FileReader (path)); StringBuilder stringBuilder = new StringBuilder(); String line = null; while( ( line = reader.readLine() ) != null ) { stringBuilder.append( line ); } return stringBuilder.toString(); } }