use of org.apache.rya.shell.util.SparqlPrompt in project incubator-rya by apache.
the class RyaAdminCommandsTest method createPCJ.
@Test
public void createPCJ() throws InstanceDoesNotExistException, RyaClientException, IOException {
// Mock the object that performs the create operation.
final String instanceName = "unitTest";
final String sparql = "SELECT * WHERE { ?person <http://isA> ?noun }";
final String pcjId = "123412342";
final CreatePCJ mockCreatePCJ = mock(CreatePCJ.class);
final Set<ExportStrategy> strategies = Sets.newHashSet(ExportStrategy.RYA);
when(mockCreatePCJ.createPCJ(eq(instanceName), eq(sparql), eq(strategies))).thenReturn(pcjId);
final RyaClient mockCommands = mock(RyaClient.class);
when(mockCommands.getCreatePCJ()).thenReturn(mockCreatePCJ);
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(sparql));
// Execute the command.
final RyaAdminCommands commands = new RyaAdminCommands(state, mock(InstallPrompt.class), mockSparqlPrompt, mock(UninstallPrompt.class));
final String message = commands.createPcj(true, false);
// Verify the values that were provided to the command were passed through to CreatePCJ.
verify(mockCreatePCJ).createPCJ(eq(instanceName), eq(sparql), eq(strategies));
// Verify a message is returned that explains what was created.
final String expected = "The PCJ has been created. Its ID is '123412342'.";
assertEquals(expected, message);
}
use of org.apache.rya.shell.util.SparqlPrompt in project incubator-rya by apache.
the class RyaAdminCommandsTest method createPeriodicPCJ.
@Test
public void createPeriodicPCJ() throws InstanceDoesNotExistException, RyaClientException, IOException {
// Mock the object that performs the create operation.
final String instanceName = "unitTest";
final String sparql = "SELECT * WHERE { ?person <http://isA> ?noun }";
final String topic = "topic";
final String brokers = "brokers";
final String pcjId = "12341234";
final CreatePeriodicPCJ mockCreatePCJ = mock(CreatePeriodicPCJ.class);
when(mockCreatePCJ.createPeriodicPCJ(eq(instanceName), eq(sparql), eq(topic), eq(brokers))).thenReturn(pcjId);
final RyaClient mockCommands = mock(RyaClient.class);
when(mockCommands.getCreatePeriodicPCJ()).thenReturn(java.util.Optional.of(mockCreatePCJ));
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(sparql));
// Execute the command.
final RyaAdminCommands commands = new RyaAdminCommands(state, mock(InstallPrompt.class), mockSparqlPrompt, mock(UninstallPrompt.class));
final String message = commands.createPeriodicPcj(topic, brokers);
// Verify the values that were provided to the command were passed through to CreatePCJ.
verify(mockCreatePCJ).createPeriodicPCJ(eq(instanceName), eq(sparql), eq(topic), eq(brokers));
// Verify a message is returned that explains what was created.
final String expected = "The Periodic PCJ has been created. Its ID is '12341234'.";
assertEquals(expected, message);
}
use of org.apache.rya.shell.util.SparqlPrompt 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.SparqlPrompt 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.SparqlPrompt 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);
}
Aggregations