use of org.apache.rya.shell.util.ConsolePrinter in project incubator-rya by apache.
the class RyaCommandsTest method testLoadData_specifyInvalidFilenameFormat.
@Test(expected = RuntimeException.class)
public void testLoadData_specifyInvalidFilenameFormat() throws InstanceDoesNotExistException, RyaClientException, IOException {
// Mock the object that performs the create operation.
final String instanceName = "unitTest";
final String statementsFile = "/path/to/statements.invalidFormat";
final String format = null;
final LoadStatementsFile mockLoadStatementsFile = mock(LoadStatementsFile.class);
final RyaClient mockCommands = mock(RyaClient.class);
when(mockCommands.getLoadStatementsFile()).thenReturn(mockLoadStatementsFile);
final SharedShellState state = new SharedShellState();
state.connectedToAccumulo(mock(AccumuloConnectionDetails.class), mockCommands);
state.connectedToInstance(instanceName);
final SparqlPrompt mockSparqlPrompt = mock(SparqlPrompt.class);
final ConsolePrinter mockConsolePrinter = mock(ConsolePrinter.class);
// Execute the command.
final RyaCommands commands = new RyaCommands(state, mockSparqlPrompt, mockConsolePrinter);
commands.loadData(statementsFile, format);
}
use of org.apache.rya.shell.util.ConsolePrinter in project incubator-rya by apache.
the class RyaCommandsTest method testSparqlQuery_fromPrompt.
@Test
public void testSparqlQuery_fromPrompt() throws InstanceDoesNotExistException, RyaClientException, IOException {
// Mock the object that performs the create operation.
final String instanceName = "unitTest";
final String queryContent = "SELECT * WHERE { ?person <http://isA> ?noun }";
final String queryFile = null;
final String expectedMessage = "MockAnswer";
final ExecuteSparqlQuery mockExecuteSparqlQuery = mock(ExecuteSparqlQuery.class);
when(mockExecuteSparqlQuery.executeSparqlQuery(instanceName, queryContent)).thenReturn(expectedMessage);
final RyaClient mockCommands = mock(RyaClient.class);
when(mockCommands.getExecuteSparqlQuery()).thenReturn(mockExecuteSparqlQuery);
final SharedShellState state = new SharedShellState();
state.connectedToAccumulo(mock(AccumuloConnectionDetails.class), mockCommands);
state.connectedToInstance(instanceName);
final SparqlPrompt mockSparqlPrompt = mock(SparqlPrompt.class);
when(mockSparqlPrompt.getSparql()).thenReturn(Optional.of(queryContent));
final ConsolePrinter mockConsolePrinter = mock(ConsolePrinter.class);
// Execute the command.
final RyaCommands commands = new RyaCommands(state, mockSparqlPrompt, mockConsolePrinter);
final String message = commands.sparqlQuery(queryFile);
// Verify the values that were provided to the command were passed through to LoadStatementsFile.
verify(mockExecuteSparqlQuery).executeSparqlQuery(instanceName, queryContent);
assertEquals(expectedMessage, message);
// Verify a message is returned that explains what was created.
}
use of org.apache.rya.shell.util.ConsolePrinter in project incubator-rya by apache.
the class RyaCommandsTest method testSparqlQuery_nonexistentFile.
@Test(expected = RuntimeException.class)
public void testSparqlQuery_nonexistentFile() throws InstanceDoesNotExistException, RyaClientException, IOException {
// Mock the object that performs the create operation.
final String instanceName = "unitTest";
final String queryFile = "src/test/resources/Nonexistent.sparql";
final RyaClient mockCommands = mock(RyaClient.class);
final SharedShellState state = new SharedShellState();
state.connectedToAccumulo(mock(AccumuloConnectionDetails.class), mockCommands);
state.connectedToInstance(instanceName);
final SparqlPrompt mockSparqlPrompt = mock(SparqlPrompt.class);
final ConsolePrinter mockConsolePrinter = mock(ConsolePrinter.class);
// Execute the command.
final RyaCommands commands = new RyaCommands(state, mockSparqlPrompt, mockConsolePrinter);
commands.sparqlQuery(queryFile);
}
use of org.apache.rya.shell.util.ConsolePrinter in project incubator-rya by apache.
the class RyaCommandsTest method testSparqlQuery.
@Test
public void testSparqlQuery() throws InstanceDoesNotExistException, RyaClientException, IOException {
// Mock the object that performs the create operation.
final String instanceName = "unitTest";
final String queryFile = "src/test/resources/Query1.sparql";
final String queryContent = FileUtils.readFileToString(new File(queryFile), StandardCharsets.UTF_8);
final String expectedMessage = "MockAnswer";
final ExecuteSparqlQuery mockExecuteSparqlQuery = mock(ExecuteSparqlQuery.class);
when(mockExecuteSparqlQuery.executeSparqlQuery(instanceName, queryContent)).thenReturn(expectedMessage);
final RyaClient mockCommands = mock(RyaClient.class);
when(mockCommands.getExecuteSparqlQuery()).thenReturn(mockExecuteSparqlQuery);
final SharedShellState state = new SharedShellState();
state.connectedToAccumulo(mock(AccumuloConnectionDetails.class), mockCommands);
state.connectedToInstance(instanceName);
final SparqlPrompt mockSparqlPrompt = mock(SparqlPrompt.class);
final ConsolePrinter mockConsolePrinter = mock(ConsolePrinter.class);
// Execute the command.
final RyaCommands commands = new RyaCommands(state, mockSparqlPrompt, mockConsolePrinter);
final String message = commands.sparqlQuery(queryFile);
// Verify the values that were provided to the command were passed through to LoadStatementsFile.
verify(mockExecuteSparqlQuery).executeSparqlQuery(instanceName, queryContent);
assertEquals(expectedMessage, message);
// Verify a message is returned that explains what was created.
}
use of org.apache.rya.shell.util.ConsolePrinter in project incubator-rya by apache.
the class RyaCommandsTest method testLoadData_specifyFormat.
@Test
public void testLoadData_specifyFormat() throws InstanceDoesNotExistException, RyaClientException, IOException {
// Mock the object that performs the create operation.
final String instanceName = "unitTest";
final String statementsFile = "/path/to/statements.nt";
final String format = "N-TRIPLES";
final LoadStatementsFile mockLoadStatementsFile = mock(LoadStatementsFile.class);
final RyaClient mockCommands = mock(RyaClient.class);
when(mockCommands.getLoadStatementsFile()).thenReturn(mockLoadStatementsFile);
final SharedShellState state = new SharedShellState();
state.connectedToAccumulo(mock(AccumuloConnectionDetails.class), mockCommands);
state.connectedToInstance(instanceName);
final SparqlPrompt mockSparqlPrompt = mock(SparqlPrompt.class);
final ConsolePrinter mockConsolePrinter = mock(ConsolePrinter.class);
// Execute the command.
final RyaCommands commands = new RyaCommands(state, mockSparqlPrompt, mockConsolePrinter);
final String message = commands.loadData(statementsFile, format);
// Verify the values that were provided to the command were passed through to LoadStatementsFile.
verify(mockLoadStatementsFile).loadStatements(instanceName, Paths.get(statementsFile), RDFFormat.NTRIPLES);
// Verify a message is returned that explains what was created.
assertTrue(message.startsWith("Loaded the file: '" + statementsFile + "' successfully in "));
assertTrue(message.endsWith(" seconds."));
}
Aggregations