TestCommandLineApp.java

package technology.tabula;

import static org.junit.Assert.*;

import java.io.IOException;
import java.nio.file.Files;
import java.nio.file.Path;
import java.nio.file.FileSystem;
import java.nio.file.FileSystems;

import org.apache.commons.cli.CommandLine;
import org.apache.commons.cli.CommandLineParser;
import org.apache.commons.cli.GnuParser;
import org.apache.commons.cli.ParseException;
import org.junit.Test;

public class TestCommandLineApp {

	private String csvFromCommandLineArgs(String[] args) throws ParseException {
		CommandLineParser parser = new GnuParser();
		CommandLine cmd = parser.parse(CommandLineApp.buildOptions(), args);

		StringBuilder stringBuilder = new StringBuilder();
		new CommandLineApp(stringBuilder, cmd).extractTables(cmd);

		return stringBuilder.toString();
	}

	@Test
	public void testExtractSpreadsheetWithArea() throws ParseException, IOException {

		String expectedCsv = UtilsForTesting.loadCsv("src/test/resources/technology/tabula/csv/spreadsheet_no_bounding_frame.csv");

		assertEquals(expectedCsv, this.csvFromCommandLineArgs(new String[] {
				"src/test/resources/technology/tabula/spreadsheet_no_bounding_frame.pdf",
				"-p", "1", "-a",
				"150.56,58.9,654.7,536.12", "-f",
				"CSV"
		}));
	}

	@Test
	public void testExtractBatchSpreadsheetWithArea() throws ParseException, IOException {
		FileSystem fs = FileSystems.getDefault();
		String expectedCsv = UtilsForTesting.loadCsv("src/test/resources/technology/tabula/csv/spreadsheet_no_bounding_frame.csv");
		Path tmpFolder = Files.createTempDirectory("tabula-java-batch-test");
		tmpFolder.toFile().deleteOnExit();

		Path copiedPDF = tmpFolder.resolve(fs.getPath("spreadsheet.pdf"));
		Path sourcePDF = fs.getPath("src/test/resources/technology/tabula/spreadsheet_no_bounding_frame.pdf");
		Files.copy(sourcePDF, copiedPDF);
		copiedPDF.toFile().deleteOnExit();

		this.csvFromCommandLineArgs(new String[] {
				"-b", tmpFolder.toString(),
				"-p", "1", "-a",
				"150.56,58.9,654.7,536.12", "-f",
				"CSV"
		});

		Path csvPath = tmpFolder.resolve(fs.getPath("spreadsheet.csv"));
		assertTrue(csvPath.toFile().exists());
		assertArrayEquals(expectedCsv.getBytes(), Files.readAllBytes(csvPath));
	}

	@Test
	public void testExtractSpreadsheetWithAreaAndNewFile() throws ParseException, IOException {

		String expectedCsv = UtilsForTesting.loadCsv("src/test/resources/technology/tabula/csv/spreadsheet_no_bounding_frame.csv");

		 this.csvFromCommandLineArgs(new String[] {
					"src/test/resources/technology/tabula/spreadsheet_no_bounding_frame.pdf",
					"-p", "1", "-a",
					"150.56,58.9,654.7,536.12", "-f",
					"CSV", "-o", "outputFile"
			});
		//assertEquals(expectedCsv,);
	}


	@Test
	public void testExtractJSONWithArea() throws ParseException, IOException {

		String expectedJson = UtilsForTesting.loadJson("src/test/resources/technology/tabula/json/spanning_cells_basic.json");

		assertEquals(expectedJson, this.csvFromCommandLineArgs(new String[] {
				"src/test/resources/technology/tabula/spanning_cells.pdf",
				"-p", "1", "-a",
				"150.56,58.9,654.7,536.12", "-f",
				"JSON"
		}));
	}

	@Test
	public void testGuessOption() throws ParseException, IOException {
		String expectedCsvNoGuessing = UtilsForTesting.loadCsv("src/test/resources/technology/tabula/csv/TestCommandLineApp_testGuessOption_no_guessing.csv");
		assertEquals(expectedCsvNoGuessing, this.csvFromCommandLineArgs(new String[] {
				"src/test/resources/technology/tabula/icdar2013-dataset/competition-dataset-eu/eu-001.pdf",
				"-p", "1",
				"-f", "CSV"
		}));

		String expectedCsvWithGuessing = UtilsForTesting.loadCsv("src/test/resources/technology/tabula/csv/TestCommandLineApp_testGuessOption_with_guessing.csv");
		assertEquals(expectedCsvWithGuessing, this.csvFromCommandLineArgs(new String[] {
				"src/test/resources/technology/tabula/icdar2013-dataset/competition-dataset-eu/eu-001.pdf",
				"-p", "1",
				"-f", "CSV",
				"-g"
		}));
	}

}