@Test
public void testImport() throws Exception {
PGImport importCommand = new PGImport();
importCommand.all = true;
importCommand.support.dataStoreFactory = TestHelper.createTestFactory();
importCommand.run(cli);
}
@Test
public void testInvalidDatabaseParams() throws Exception {
PGImport importCommand = new PGImport();
importCommand.commonArgs.host = "nonexistent";
importCommand.all = true;
exception.expect(CommandFailedException.class);
importCommand.run(cli);
}
@Test
public void testImportGetNamesException() throws Exception {
PGImport importCommand = new PGImport();
importCommand.all = true;
importCommand.support.dataStoreFactory = TestHelper.createFactoryWithGetNamesException();
exception.expect(CommandFailedException.class);
importCommand.run(cli);
}
@Test
public void testNullDataStore() throws Exception {
PGImport importCommand = new PGImport();
importCommand.all = true;
importCommand.support.dataStoreFactory = TestHelper.createNullTestFactory();
exception.expect(CommandFailedException.class);
importCommand.run(cli);
}
@Test
public void testNoTableNotAll() throws Exception {
PGImport importCommand = new PGImport();
importCommand.all = false;
importCommand.table = "";
importCommand.support.dataStoreFactory = TestHelper.createTestFactory();
exception.expect(CommandFailedException.class);
importCommand.run(cli);
}
@Test
public void testImportException() throws Exception {
when(cli.getConsole()).thenThrow(new MockitoException("Exception"));
PGImport importCommand = new PGImport();
importCommand.all = true;
importCommand.support.dataStoreFactory = TestHelper.createTestFactory();
exception.expect(MockitoException.class);
importCommand.run(cli);
}