org.jgrapht.graph.ClassBasedEdgeFactory Java Examples

The following examples show how to use org.jgrapht.graph.ClassBasedEdgeFactory. You can vote up the ones you like or vote down the ones you don't like, and go to the original project or source file by following the links above each example. You may check out the related API usage on the sidebar.

Example #1

Source File: RefreshOperationTest.java From jpa-unit with Apache License 2.0

6 votes

@Test
public void testExecute() throws Exception {
    // GIVEN
    final Node n1 = graphElementFactory.createNode("n1", Arrays.asList("A"),
            ImmutableMap.<String, Object>builder().put("id", 1l).put("value", "A").build());
    final Node n2 = graphElementFactory.createNode("n2", Arrays.asList("A"),
            ImmutableMap.<String, Object>builder().put("id", 2l).build());
    final Edge e1 = graphElementFactory.createEdge(n1, n2, "e1", Arrays.asList("E"),
            ImmutableMap.<String, Object>builder().put("id", 3l).put("value", "C").build());

    final Graph<Node, Edge> graph = new DefaultDirectedGraph<>(new ClassBasedEdgeFactory<>(Edge.class));
    graph.addVertex(n1);
    graph.addVertex(n2);
    graph.addEdge(e1.getSourceNode(), e1.getTargetNode(), e1);

    // WHEN
    operation.execute(connection, graph);

    // THEN
    final ArgumentCaptor<String> queryCaptor = ArgumentCaptor.forClass(String.class);
    verify(operation, times(3)).executeQuery(eq(connection), queryCaptor.capture());
    final List<String> queries = queryCaptor.getAllValues();
    assertThat(queries.get(0), containsString("MERGE (n1:A {id:1}) SET n1.value=\"A\""));
    assertThat(queries.get(1), containsString("MERGE (n2:A {id:2})"));
    assertThat(queries.get(2), containsString("MATCH (n1:A {id:1}),(n2:A {id:2}) MERGE (n1)-[e1:E]->(n2) SET e1.id=3,e1.value=\"C\""));
}

Example #2

Source File: UpdateOperationTest.java From jpa-unit with Apache License 2.0

6 votes

@Test
public void testExecute() throws Exception {
    // GIVEN
    final Node n1 = graphElementFactory.createNode("n1", Arrays.asList("A"),
            ImmutableMap.<String, Object>builder().put("id", 1l).put("value", "A").build());
    final Node n2 = graphElementFactory.createNode("n2", Arrays.asList("A"),
            ImmutableMap.<String, Object>builder().put("id", 2l).build());
    final Edge e1 = graphElementFactory.createEdge(n1, n2, "e1", Arrays.asList("E"),
            ImmutableMap.<String, Object>builder().put("id", 3l).put("value", "C").build());

    final Graph<Node, Edge> graph = new DefaultDirectedGraph<>(new ClassBasedEdgeFactory<>(Edge.class));
    graph.addVertex(n1);
    graph.addVertex(n2);
    graph.addEdge(e1.getSourceNode(), e1.getTargetNode(), e1);

    // WHEN
    operation.execute(connection, graph);

    // THEN
    final ArgumentCaptor<String> queryCaptor = ArgumentCaptor.forClass(String.class);
    verify(operation, times(2)).executeQuery(eq(connection), queryCaptor.capture());
    final List<String> queries = queryCaptor.getAllValues();
    assertThat(queries.get(0), containsString("MATCH (n1:A {id:1}) SET n1.value=\"A\""));
    assertThat(queries.get(1), containsString("MATCH (n1:A {id:1}),(n2:A {id:2}) MERGE (n1)-[e1:E]->(n2) SET e1.id=3,e1.value=\"C\""));
}

Example #3

Source File: DeleteAllOperationTest.java From jpa-unit with Apache License 2.0

6 votes

@Test
public void testExecute() throws Exception {
    // GIVEN
    final Node n1 = graphElementFactory.createNode("n1", Arrays.asList("A"),
            ImmutableMap.<String, Object>builder().put("id", 1l).build());
    final Node n2 = graphElementFactory.createNode("n2", Arrays.asList("A"),
            ImmutableMap.<String, Object>builder().put("id", 2l).build());
    final Edge e1 = graphElementFactory.createEdge(n1, n2, "e1", Arrays.asList("E"), Collections.emptyMap());

    final Graph<Node, Edge> graph = new DefaultDirectedGraph<>(new ClassBasedEdgeFactory<>(Edge.class));
    graph.addVertex(n1);
    graph.addVertex(n2);
    graph.addEdge(e1.getSourceNode(), e1.getTargetNode(), e1);

    // WHEN
    operation.execute(connection, graph);

    // THEN
    final ArgumentCaptor<String> queryCaptor = ArgumentCaptor.forClass(String.class);
    verify(operation, times(2)).executeQuery(eq(connection), queryCaptor.capture());
    final String query = queryCaptor.getValue();
    assertThat(query, containsString("MATCH (n:A) DETACH DELETE n"));
}

Example #4

Source File: InsertOperationTest.java From jpa-unit with Apache License 2.0

6 votes

@Test
public void testExecute() throws Exception {
    // GIVEN
    final Node n1 = graphElementFactory.createNode("n1", Arrays.asList("A"),
            ImmutableMap.<String, Object>builder().put("id", 1l).build());
    final Node n2 = graphElementFactory.createNode("n2", Arrays.asList("A"),
            ImmutableMap.<String, Object>builder().put("id", 2l).build());
    final Edge e1 = graphElementFactory.createEdge(n1, n2, "e1", Arrays.asList("E"),
            ImmutableMap.<String, Object>builder().put("id", 3l).build());

    final Graph<Node, Edge> graph = new DefaultDirectedGraph<>(new ClassBasedEdgeFactory<>(Edge.class));
    graph.addVertex(n1);
    graph.addVertex(n2);
    graph.addEdge(e1.getSourceNode(), e1.getTargetNode(), e1);

    // WHEN
    operation.execute(connection, graph);

    // THEN
    final ArgumentCaptor<String> queryCaptor = ArgumentCaptor.forClass(String.class);
    verify(operation).executeQuery(eq(connection), queryCaptor.capture());
    final String query = queryCaptor.getValue();
    assertThat(query, containsString("CREATE (n1:A {id:1}),(n2:A {id:2}),(n1)-[e1:E {id:3}]->(n2)"));
}

Example #5

Source File: CompositeOperationTest.java From jpa-unit with Apache License 2.0

5 votes

@Test
public void testOperationExecution() throws SQLException {
    // GIVEN
    final Graph<Node, Edge> graph = new DefaultDirectedGraph<>(new ClassBasedEdgeFactory<>(Edge.class));
    final CompositeOperation operation = new CompositeOperation(operation1, operation2, operation3);

    // WHEN
    operation.execute(connection, graph);

    // THEN
    final InOrder order = inOrder(operation1, operation2, operation3);
    order.verify(operation1).execute(eq(connection), eq(graph));
    order.verify(operation2).execute(eq(connection), eq(graph));
    order.verify(operation3).execute(eq(connection), eq(graph));
}

Example #6

Source File: DeleteOperationTest.java From jpa-unit with Apache License 2.0

5 votes

@Test
public void testExecute() throws Exception {
    // GIVEN
    final Node n1 = graphElementFactory.createNode("n1", Arrays.asList("A"),
            ImmutableMap.<String, Object>builder().put("id", 1l).build());
    final Node n2 = graphElementFactory.createNode("n2", Arrays.asList("A"),
            ImmutableMap.<String, Object>builder().put("id", 2l).build());
    final Edge e1 = graphElementFactory.createEdge(n1, n2, "e1", Arrays.asList("E"),
            ImmutableMap.<String, Object>builder().put("id", 3l).build());

    final Graph<Node, Edge> graph = new DefaultDirectedGraph<>(new ClassBasedEdgeFactory<>(Edge.class));
    graph.addVertex(n1);
    graph.addVertex(n2);
    graph.addEdge(e1.getSourceNode(), e1.getTargetNode(), e1);

    // WHEN
    operation.execute(connection, graph);

    // THEN
    final ArgumentCaptor<String> queryCaptor = ArgumentCaptor.forClass(String.class);
    verify(operation, times(2)).executeQuery(eq(connection), queryCaptor.capture());
    final List<String> queries = queryCaptor.getAllValues();
    final String query1 = queries.get(0);
    final String query2 = queries.get(1);
    assertThat(query1, containsString("MATCH (n1:A {id:1})-[e1:E {id:3}]->(`n2:A {id:2}`) DELETE e1"));
    assertThat(query2, containsString("MATCH (n1:A {id:1}),(n2:A {id:2}) DELETE n1,n2"));
}

Example #7

Source File: DatabaseReader.java From jpa-unit with Apache License 2.0

5 votes

public Graph<Node, Edge> readGraph(final Connection connection) throws SQLException {
    final List<Node> nodes = new ArrayList<>();
    final List<Edge> edges = new ArrayList<>();

    readGraphElements(connection, edges, nodes);

    final DefaultDirectedGraph<Node, Edge> graph = new DefaultDirectedGraph<>(new ClassBasedEdgeFactory<>(Edge.class));
    nodes.forEach(graph::addVertex);
    edges.forEach(e -> graph.addEdge(e.getSourceNode(), e.getTargetNode(), e));

    return graph;
}

Example #8

Source File: DataSetLoaderProvider.java From jpa-unit with Apache License 2.0

5 votes

@Override
public DataSetLoader<Graph<Node, Edge>> xmlLoader() {
    return (final File path) -> {
        try {
            final DefaultDirectedGraph<Node, Edge> graph = new DefaultDirectedGraph<>(new ClassBasedEdgeFactory<>(Edge.class));
            final GraphMLReader<Node, Edge> importer = new GraphMLReader<>(graphElementFactory, graphElementFactory);
            importer.importGraph(graph, path);
            return graph;
        } catch (final ImportException e) {
            throw new IOException(e);
        }
    };

}

Example #9

Source File: Neo4JDbFeatureExecutor.java From jpa-unit with Apache License 2.0

5 votes

private Graph<Node, Edge> mergeGraphs(final List<Graph<Node, Edge>> graphs) {
    final Graph<Node, Edge> mergedGraph = new DefaultDirectedGraph<>(new ClassBasedEdgeFactory<>(Edge.class));

    for (final Graph<Node, Edge> graph : graphs) {
        Graphs.addGraph(mergedGraph, graph);
    }

    return mergedGraph;
}

Example #10

Source File: CleanupStrategyProvider.java From jpa-unit with Apache License 2.0

5 votes

private Graph<Node, Edge> computeGraphToBeDeleted(final Graph<Node, Edge> graph, final String... nodeTypesToRetain) {
    final DirectedGraph<Node, Edge> toDelete = new DefaultDirectedGraph<>(new ClassBasedEdgeFactory<>(Edge.class));

    // copy graph to a destination, which we are going to modify
    Graphs.addGraph(toDelete, graph);

    // remove the nodes, we have to retain from the graph
    Graphs.removeVerticesAndPreserveConnectivity(toDelete, v -> shouldRetainNode(v, nodeTypesToRetain));

    return toDelete;
}

Example #11

Source File: CleanupStrategyProviderIT.java From jpa-unit with Apache License 2.0

5 votes

private static List<Graph<Node, Edge>> loadDataSet(final String path) {
    try {
        final URL resource = Thread.currentThread().getContextClassLoader().getResource(path);
        final File file = new File(resource.toURI());

        final DefaultDirectedGraph<Node, Edge> graph = new DefaultDirectedGraph<>(new ClassBasedEdgeFactory<>(Edge.class));
        final GraphMLReader<Node, Edge> importer = new GraphMLReader<>(factory, factory);
        importer.importGraph(graph, file);

        return Arrays.asList(graph);
    } catch (final Exception e) {
        throw new RuntimeException(e);
    }
}

Example #12

Source File: Neo4JDbFeatureExecutorTest.java From jpa-unit with Apache License 2.0

5 votes

@SuppressWarnings("unchecked")
@Test
public void testCleanupFeatureExecution() throws DbFeatureException {
    // GIVEN
    when(cleanupStrategy.provide(any(CleanupStrategy.StrategyProvider.class))).thenReturn(cleanupStrategyExecutor);
    final List<Graph<Node, Edge>> initialDataSets = Arrays.asList(new DefaultDirectedGraph<>(new ClassBasedEdgeFactory<>(Edge.class)));

    // WHEN
    final DbFeature<Connection> feature = featureExecutor.createCleanupFeature(cleanupStrategy, initialDataSets);
    assertThat(feature, notNullValue());
    feature.execute(connection);

    // THEN
    verify(cleanupStrategyExecutor).execute(eq(connection), eq(initialDataSets));
}

Example #13

Source File: AutomationEngine.java From JTAF-XCore with Apache License 2.0

4 votes

private AutomationEngine() {
	try {
		InputStream fi = getFrameworkFile();
		GenericApplicationContext ctx = new GenericApplicationContext();

		XmlBeanDefinitionReader xmlReader = new XmlBeanDefinitionReader(ctx);
		xmlReader
				.setValidationMode(XmlBeanDefinitionReader.VALIDATION_NONE);

		xmlReader.loadBeanDefinitions(new InputSource(fi));

		ctx.refresh();

		this.pluginManager = (PluginManager) ctx.getBean("PluginManager");

		digraph = new TestDigraph(
				new ClassBasedEdgeFactory<DiNode, DiEdge>(DiEdge.class));
		commandlibParser = new CommandLibraryParser();
		scriptParser = new ScriptParser();
		scriptParser.setDigraph(digraph);
		commandlibParser
				.setAutomationClassLoader(new DefaultAutomationClassLoader());
		testStrategyParser = new TestStrategyParser();
		testStrategyParser.setDigraph(digraph);
		initPostParseStrategyElementPlugins();
		testRoot = null;

		this.interpreter = (Interpreter) ctx.getBean("Interpreter");

		this.interpreter.setCommandRunnerPlugins(pluginManager
				.getCommandRunnerPlugins());
		this.interpreter.setTestRunnerPlugins(pluginManager
				.getTestRunnerPlugins());
		this.interpreter.setTearDownPlugins(pluginManager.getTearDownPlugins());

		initPostParseAllPlugins();
		initPostParseSuitePlugins();
		initPostParseTestPlugins();

	} catch (Exception e) {
		// If something goes wrong here, we have a serious issue
		logger.fatal(e);
		throw new RuntimeException(e);
	}
}

Example #14

Source File: AssociationAnalyzer.java From Knowage-Server with GNU Affero General Public License v3.0

4 votes

public AssociationAnalyzer(Collection<Association> associations) {
	this.associations = associations;
	this.datasetToAssociationToColumnMap = new HashMap<String, Map<String, String>>();
	this.graph = new Pseudograph<String, LabeledEdge<String>>(
			new ClassBasedEdgeFactory<String, LabeledEdge<String>>((Class<LabeledEdge<String>>) (Object) LabeledEdge.class));
}

Example #15

Source File: GraphComparatorTest.java From jpa-unit with Apache License 2.0

4 votes

private Graph<Node, Edge> createGraph(final List<Node> nodes, final List<Edge> edges) {
    final Graph<Node, Edge> graph = new DefaultDirectedGraph<>(new ClassBasedEdgeFactory<>(Edge.class));
    nodes.forEach(graph::addVertex);
    edges.forEach(e -> graph.addEdge(e.getSourceNode(), e.getTargetNode(), e));
    return graph;
}

Example #16