@Override public Iterable<Relationship> getRelationships() { return gdb.getAllRelationships(); }
@Test public void getAllRelationships() { Set<Relationship> existingRelationships = Iterables.addToCollection( getGraphDb().getAllRelationships(), new HashSet<>() ); Set<Relationship> createdRelationships = new HashSet<>(); Node node = getGraphDb().createNode(); for ( int i = 0; i < 100; i++ ) { createdRelationships.add( node.createRelationshipTo( getGraphDb().createNode(), TEST ) ); } newTransaction(); Set<Relationship> allRelationships = new HashSet<>(); allRelationships.addAll( existingRelationships ); allRelationships.addAll( createdRelationships ); int count = 0; for ( Relationship rel : getGraphDb().getAllRelationships() ) { assertTrue( "Unexpected rel " + rel + ", expected one of " + allRelationships, allRelationships.contains( rel ) ); count++; } assertEquals( allRelationships.size(), count ); }
private void cleanDatabaseContent( GraphDatabaseService db ) { try ( Transaction tx = db.beginTx() ) { db.getAllRelationships().forEach( Relationship::delete ); db.getAllNodes().forEach( Node::delete ); tx.success(); } }
public static void cleanupAllRelationshipsAndNodes( GraphDatabaseService db ) { try ( Transaction tx = db.beginTx() ) { for ( Relationship relationship : db.getAllRelationships() ) { relationship.delete(); } for ( Node node : db.getAllNodes() ) { node.delete(); } tx.success(); } } }
for ( Relationship relationship : db.getAllRelationships() )
for ( Relationship relationship : db.getAllRelationships() )
private void verifyData( int expectedNodeCount, int expectedRelationshipCount, Validator<Node> nodeAdditionalValidation, Validator<Relationship> relationshipAdditionalValidation ) { GraphDatabaseService db = dbRule.getGraphDatabaseAPI(); try ( Transaction tx = db.beginTx() ) { int nodeCount = 0; int relationshipCount = 0; for ( Node node : db.getAllNodes() ) { assertTrue( node.hasProperty( "name" ) ); nodeAdditionalValidation.validate( node ); nodeCount++; } assertEquals( expectedNodeCount, nodeCount ); for ( Relationship relationship : db.getAllRelationships() ) { assertTrue( relationship.hasProperty( "created" ) ); relationshipAdditionalValidation.validate( relationship ); relationshipCount++; } assertEquals( expectedRelationshipCount, relationshipCount ); tx.success(); } }
@Test public void shouldBeEmptyWhenFirstStarted() { // When File storeDir = testDir.absolutePath(); GraphDatabaseService db = new TestGraphDatabaseFactory().newEmbeddedDatabase( storeDir ); // Then try ( Transaction ignore = db.beginTx() ) { assertEquals( 0, count( db.getAllNodes() ) ); assertEquals( 0, count( db.getAllRelationships() ) ); assertEquals( 0, count( db.getAllRelationshipTypes() ) ); assertEquals( 0, count( db.getAllLabels() ) ); assertEquals( 0, count( db.getAllPropertyKeys() ) ); } db.shutdown(); } }
@Description("apoc.graph.fromDB('name',{properties}) - creates a virtual graph object for later processing") @Procedure public Stream<VirtualGraph> fromDB(@Name("name") String name, @Name("properties") Map<String,Object> properties) { return Stream.of(new VirtualGraph(name,db.getAllNodes(),db.getAllRelationships(),properties)); }
for ( Relationship relationship : graphDb.getAllRelationships() )
private void assertCorrectNumberOfNodesAndRelationships(GraphDatabaseService database, int numberOfNodes, int numberOfEdges) { try (Transaction tx = database.beginTx()) { assertEquals(numberOfNodes, count(database.getAllNodes())); assertEquals(numberOfEdges, count(database.getAllRelationships())); tx.success(); } }
private void assertCorrectNumberOfNodesAndRelationships(GraphDatabaseService database, int numberOfNodes, int edgesPerNewNode) { try (Transaction tx = database.beginTx()) { assertEquals(numberOfNodes, count(database.getAllNodes())); assertEquals(numberOfNodes * edgesPerNewNode - (edgesPerNewNode * (edgesPerNewNode + 1) / 2), count(database.getAllRelationships())); tx.success(); } }
@Test public void shouldGenerateErdosRenyi2() { db.execute("CALL apoc.generate.er(null,null,null,null)"); try (Transaction tx = db.beginTx()) { assertEquals(1000, count(db.getAllNodes())); assertEquals(10000, count(db.getAllRelationships())); assertEquals("Person", firstOrNull(firstOrNull(db.getAllNodes()).getLabels()).name()); assertEquals("FRIEND_OF", firstOrNull(db.getAllRelationships()).getType().name()); assertTrue(firstOrNull(db.getAllNodes()).hasProperty("name")); tx.success(); } }
@Test public void shouldGenerateComplete1() { db.execute("CALL apoc.generate.complete(10,'TestLabel','TEST_REL')"); try (Transaction tx = db.beginTx()) { assertEquals(10, count(db.getAllNodes())); assertEquals(45, count(db.getAllRelationships())); assertEquals("TestLabel", firstOrNull(firstOrNull(db.getAllNodes()).getLabels()).name()); assertEquals("TEST_REL", firstOrNull(db.getAllRelationships()).getType().name()); assertTrue(firstOrNull(db.getAllNodes()).hasProperty("uuid")); tx.success(); } }
@Test public void shouldGenerateBarabasiAlbert3() { db.execute("CALL apoc.generate.ba(null,null,'TestLabel','TEST_REL')"); try (Transaction tx = db.beginTx()) { assertEquals(1000, count(db.getAllNodes())); assertEquals(1997, count(db.getAllRelationships())); assertEquals("TestLabel", firstOrNull(firstOrNull(db.getAllNodes()).getLabels()).name()); assertEquals("TEST_REL", firstOrNull(db.getAllRelationships()).getType().name()); assertTrue(firstOrNull(db.getAllNodes()).hasProperty("uuid")); tx.success(); } }
@Test public void shouldGenerateWattsStrogatz1() { db.execute("CALL apoc.generate.ws(100,10,0.5,null,null)"); try (Transaction tx = db.beginTx()) { assertEquals(100, count(db.getAllNodes())); assertEquals(500, count(db.getAllRelationships())); assertEquals("Person", firstOrNull(firstOrNull(db.getAllNodes()).getLabels()).name()); assertEquals("FRIEND_OF", firstOrNull(db.getAllRelationships()).getType().name()); assertTrue(firstOrNull(db.getAllNodes()).hasProperty("name")); tx.success(); } }
@Test public void shouldGenerateWattsStrogatz3() { db.execute("CALL apoc.generate.ws(null,null,null,'TestLabel','TEST_REL')"); try (Transaction tx = db.beginTx()) { assertEquals(1000, count(db.getAllNodes())); assertEquals(2000, count(db.getAllRelationships())); assertEquals("TestLabel", firstOrNull(firstOrNull(db.getAllNodes()).getLabels()).name()); assertEquals("TEST_REL", firstOrNull(db.getAllRelationships()).getType().name()); assertTrue(firstOrNull(db.getAllNodes()).hasProperty("uuid")); tx.success(); } }
@Test public void shouldGenerateSimple1() { db.execute("CALL apoc.generate.simple([2,2,2,2],'TestLabel','TEST_REL')"); try (Transaction tx = db.beginTx()) { assertEquals(4, count(db.getAllNodes())); assertEquals(4, count(db.getAllRelationships())); assertEquals("TestLabel", firstOrNull(firstOrNull(db.getAllNodes()).getLabels()).name()); assertEquals("TEST_REL", firstOrNull(db.getAllRelationships()).getType().name()); assertTrue(firstOrNull(db.getAllNodes()).hasProperty("uuid")); tx.success(); } }
@Test public void shouldGenerateBarabasiAlbert1() { db.execute("CALL apoc.generate.ba(100,3,null,null)"); try (Transaction tx = db.beginTx()) { assertEquals(100, count(db.getAllNodes())); assertEquals(294, count(db.getAllRelationships())); assertEquals("Person", firstOrNull(firstOrNull(db.getAllNodes()).getLabels()).name()); assertEquals("FRIEND_OF", firstOrNull(db.getAllRelationships()).getType().name()); assertTrue(firstOrNull(db.getAllNodes()).hasProperty("name")); tx.success(); } }
@Test public void shouldGenerateComplete2() { db.execute("CALL apoc.generate.complete(10,null,null)"); try (Transaction tx = db.beginTx()) { assertEquals(10, count(db.getAllNodes())); assertEquals(45, count(db.getAllRelationships())); assertEquals("Person", firstOrNull(firstOrNull(db.getAllNodes()).getLabels()).name()); assertEquals("FRIEND_OF", firstOrNull(db.getAllRelationships()).getType().name()); assertTrue(firstOrNull(db.getAllNodes()).hasProperty("name")); tx.success(); } }