/* Copyright (c) 2013-2014 Boundless and others.
* All rights reserved. This program and the accompanying materials
* are made available under the terms of the Eclipse Distribution License v1.0
* which accompanies this distribution, and is available at
* https://www.eclipse.org/org/documents/edl-v10.html
*
* Contributors:
* Gabriel Roldan (Boundless) - initial implementation
*/
package org.locationtech.geogig.geotools.cli.porcelain;
import java.util.Arrays;
import jline.UnsupportedTerminal;
import jline.console.ConsoleReader;
import org.junit.Rule;
import org.junit.Test;
import org.junit.rules.ExpectedException;
import org.locationtech.geogig.api.porcelain.CommitOp;
import org.locationtech.geogig.cli.CommandFailedException;
import org.locationtech.geogig.cli.GeogigCLI;
import org.locationtech.geogig.cli.InvalidParameterException;
import org.locationtech.geogig.test.integration.RepositoryTestCase;
public class PGExportTest extends RepositoryTestCase {
@Rule
public ExpectedException exception = ExpectedException.none();
private GeogigCLI cli;
@Override
public void setUpInternal() throws Exception {
ConsoleReader consoleReader = new ConsoleReader(System.in, System.out,
new UnsupportedTerminal());
cli = new GeogigCLI(consoleReader);
cli.setGeogig(geogig);
// Add points
insertAndAdd(points1);
insertAndAdd(points2);
insertAndAdd(points3);
geogig.command(CommitOp.class).call();
// Add lines
insertAndAdd(lines1);
insertAndAdd(lines2);
insertAndAdd(lines3);
geogig.command(CommitOp.class).call();
}
@Override
public void tearDownInternal() throws Exception {
cli.close();
}
@Test
public void testExport() throws Exception {
PGExport exportCommand = new PGExport();
exportCommand.args = Arrays.asList("Points", "Points");
exportCommand.dataStoreFactory = TestHelper.createTestFactory();
exportCommand.run(cli);
}
@Test
public void testNullDataStore() throws Exception {
PGExport exportCommand = new PGExport();
exportCommand.args = Arrays.asList("Points", "Points");
exportCommand.dataStoreFactory = TestHelper.createNullTestFactory();
exception.expect(CommandFailedException.class);
exportCommand.run(cli);
}
@Test
public void testNoArgs() throws Exception {
PGExport exportCommand = new PGExport();
exportCommand.args = Arrays.asList();
exception.expect(CommandFailedException.class);
exportCommand.dataStoreFactory = TestHelper.createNullTestFactory();
exportCommand.run(cli);
}
@Test
public void testExportToTableThatExists() throws Exception {
PGExport exportCommand = new PGExport();
exportCommand.args = Arrays.asList("Points", "table1");
exportCommand.dataStoreFactory = TestHelper.createTestFactory();
exception.expect(InvalidParameterException.class);
exportCommand.run(cli);
}
@Test
public void testExportToTableThatExistsWithOverwrite() throws Exception {
PGExport exportCommand = new PGExport();
exportCommand.args = Arrays.asList("WORK_HEAD:Points", "testTable");
exportCommand.dataStoreFactory = TestHelper.createTestFactory();
exportCommand.run(cli);
exportCommand.args = Arrays.asList("Lines", "testTable");
exportCommand.overwrite = true;
exportCommand.run(cli);
}
@Test
public void testExportWithNonexistentFeatureTypeTree() throws Exception {
PGExport exportCommand = new PGExport();
exportCommand.args = Arrays.asList("invalidType", "invalidTable");
exportCommand.dataStoreFactory = TestHelper.createTestFactory();
exception.expect(InvalidParameterException.class);
exportCommand.run(cli);
}
@Test
public void testExportWithNullTable() throws Exception {
PGExport exportCommand = new PGExport();
exportCommand.args = Arrays.asList("Points", null);
exportCommand.dataStoreFactory = TestHelper.createTestFactory();
exception.expect(InvalidParameterException.class);
exportCommand.run(cli);
}
@Test
public void testExportWithNullFeatureType() throws Exception {
PGExport exportCommand = new PGExport();
exportCommand.args = Arrays.asList(null, "invalidTable");
exportCommand.dataStoreFactory = TestHelper.createTestFactory();
exception.expect(InvalidParameterException.class);
exportCommand.run(cli);
}
@Test
public void testExportWithEmptyStringForFeatureType() throws Exception {
PGExport exportCommand = new PGExport();
exportCommand.args = Arrays.asList("", "invalidTable");
exportCommand.dataStoreFactory = TestHelper.createTestFactory();
exception.expect(InvalidParameterException.class);
exportCommand.run(cli);
}
@Test
public void testExportWithEmptyStringForTable() throws Exception {
PGExport exportCommand = new PGExport();
exportCommand.args = Arrays.asList("Points", "");
exportCommand.dataStoreFactory = TestHelper.createTestFactory();
exception.expect(InvalidParameterException.class);
exportCommand.run(cli);
}
@Test
public void testExportWithFeatureNameInsteadOfType() throws Exception {
PGExport exportCommand = new PGExport();
exportCommand.args = Arrays.asList("Points/Points.1", "invalidTable");
exportCommand.dataStoreFactory = TestHelper.createTestFactory();
exception.expect(InvalidParameterException.class);
exportCommand.run(cli);
}
}