Search in sources :

Example 6 with LoadStatementsFile

use of org.apache.rya.api.client.LoadStatementsFile in project incubator-rya by apache.

the class RyaCommandsTest method loadData_relativePath.

@Test
public void loadData_relativePath() throws Exception {
    // Mock the object that performs the create operation.
    final String instanceName = "unitTest";
    final String statementsFile = "~/statements.nt";
    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);
    final String message = commands.loadData(statementsFile, format);
    // Verify the values that were provided to the command were passed through to LoadStatementsFile
    // using a user rooted filename.
    String rootedFile = System.getProperty("user.home") + "/statements.nt";
    verify(mockLoadStatementsFile).loadStatements(instanceName, Paths.get(rootedFile), 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."));
}
Also used : ConsolePrinter(org.apache.rya.shell.util.ConsolePrinter) LoadStatementsFile(org.apache.rya.api.client.LoadStatementsFile) AccumuloConnectionDetails(org.apache.rya.api.client.accumulo.AccumuloConnectionDetails) SparqlPrompt(org.apache.rya.shell.util.SparqlPrompt) RyaClient(org.apache.rya.api.client.RyaClient) Test(org.junit.Test)

Aggregations

LoadStatementsFile (org.apache.rya.api.client.LoadStatementsFile)6 RyaClient (org.apache.rya.api.client.RyaClient)6 AccumuloConnectionDetails (org.apache.rya.api.client.accumulo.AccumuloConnectionDetails)6 ConsolePrinter (org.apache.rya.shell.util.ConsolePrinter)6 SparqlPrompt (org.apache.rya.shell.util.SparqlPrompt)6 Test (org.junit.Test)6 File (java.io.File)1 ExecuteSparqlQuery (org.apache.rya.api.client.ExecuteSparqlQuery)1