Examples with GraphMLWriter org.pentaho.metaverse.graph.GraphMLWriter used on opensource projects

Example 1 with GraphMLWriter

use of org.pentaho.metaverse.graph.GraphMLWriter in project pentaho-metaverse by pentaho.

the class FileSystemLineageWriterTest method testGetSetGraphOutputStream.

@Test
public void testGetSetGraphOutputStream() {
    assertNull(writer.getGraphOutputStream(null));
    IGraphWriter graphWriter = new GraphMLWriter();
    writer.setGraphWriter(graphWriter);
    assertNotNull(writer.getGraphOutputStream(holder));
    graphWriter = new GraphSONWriter();
    writer.setGraphWriter(graphWriter);
    assertNotNull(writer.getGraphOutputStream(holder));
    graphWriter = new GraphCsvWriter();
    writer.setGraphWriter(graphWriter);
    assertNotNull(writer.getGraphOutputStream(holder));
}

Example 2 with GraphMLWriter

use of org.pentaho.metaverse.graph.GraphMLWriter in project pentaho-metaverse by pentaho.

the class FileSystemLocatorTest method testStartLocator.

/**
 * Runs the spyLocator and checks the results
 *
 * @throws Exception When bad things happen
 */
@Test
public void testStartLocator() throws Exception {
    TinkerGraph graph = new TinkerGraph();
    IMetaverseBuilder metaverseBuilder = new MetaverseBuilder(graph);
    spyLocator.setMetaverseBuilder(metaverseBuilder);
    spyLocator.setRepositoryId("testrepo");
    spyLocator.addDocumentListener(this);
    spyLocator.setRootFolder("src/test/resources/solution");
    assertEquals("Root folder is wrong", "src/test/resources/solution", spyLocator.getRootFolder());
    TestFileSystemLocator.delay = 0;
    spyLocator.setRootFolder("bogus");
    events = new ArrayList<IDocumentEvent>();
    try {
        spyLocator.startScan();
        MetaverseCompletionService.getInstance().waitTillEmpty();
        fail();
    } catch (MetaverseLocatorException e) {
        assertEquals("Event count is wrong", 0, events.size());
    }
    spyLocator.setRootFolder("src/test/resources/solution/folder 2/parse.ktr");
    events = new ArrayList<IDocumentEvent>();
    try {
        spyLocator.startScan();
        MetaverseCompletionService.getInstance().waitTillEmpty();
        fail();
    } catch (MetaverseLocatorException e) {
        assertEquals("Event count is wrong", 0, events.size());
    }
    spyLocator.setRootFolder("src/test/resources/solution");
    assertEquals("Repo id is wrong", "testrepo", spyLocator.getRepositoryId());
    assertNotNull("Indexer type is null", spyLocator.getLocatorType());
    events = new ArrayList<IDocumentEvent>();
    spyLocator.startScan();
    MetaverseCompletionService.getInstance().waitTillEmpty();
    assertEquals("Event count is wrong", 7, events.size());
    for (IDocumentEvent event : events) {
        System.out.println(event.getDocument().getStringID());
        assertNotNull(event.getDocument());
        MetaverseDocument document = (MetaverseDocument) event.getDocument();
        if (document.getExtension().equals("ktr")) {
            assertTrue(document.getContent() instanceof String);
        } else if (document.getExtension().equals("kjb")) {
            assertTrue(document.getContent() instanceof String);
        }
    }
    spyLocator.removeDocumentListener(this);
    events = new ArrayList<IDocumentEvent>();
    spyLocator.startScan();
    MetaverseCompletionService.getInstance().waitTillEmpty();
    assertEquals("Event count is wrong", 0, events.size());
    assertNotNull("Graph is null", graph);
    GraphMLWriter writer = new GraphMLWriter();
    writer.outputGraph(graph, new FileOutputStream(OUTPUT_FOLDER + "FileSystemLocatorTest.graphml"));
}

Example 3 with GraphMLWriter

use of org.pentaho.metaverse.graph.GraphMLWriter in project pentaho-metaverse by pentaho.

the class VfsLineageWriterTest method testSetGraphOutputStream.

@Test
public void testSetGraphOutputStream() throws IOException {
    try (OutputStream graphOutputStream = writer.getGraphOutputStream(null)) {
        assertNull(graphOutputStream);
    }
    IGraphWriter graphWriter = new GraphMLWriter();
    writer.setGraphWriter(graphWriter);
    try (OutputStream graphOutputStream = writer.getGraphOutputStream(holder)) {
        assertNotNull(graphOutputStream);
    }
    graphWriter = new GraphSONWriter();
    writer.setGraphWriter(graphWriter);
    try (OutputStream graphOutputStream = writer.getGraphOutputStream(holder)) {
        assertNotNull(graphOutputStream);
    }
    graphWriter = new GraphCsvWriter();
    writer.setGraphWriter(graphWriter);
    try (OutputStream graphOutputStream = writer.getGraphOutputStream(holder)) {
        assertNotNull(graphOutputStream);
    }
}