@Test public void createWithIDTest() { GradoopId edgeId = GradoopId.get(); GradoopId sourceId = GradoopId.get(); GradoopId targetId = GradoopId.get(); EPGMEdge e = new EdgeFactory().initEdge(edgeId, sourceId, targetId); assertThat(e.getId(), is(edgeId)); assertThat(e.getSourceId(), is(sourceId)); assertThat(e.getTargetId(), is(targetId)); assertThat(e.getPropertyCount(), is(0)); assertThat(e.getGraphCount(), is(0)); }
@Test public void testEdges() throws Exception { GradoopId edgeId = GradoopId.get(); GradoopId sourceId = GradoopId.get(); GradoopId targetId = GradoopId.get(); String label = "A"; Properties props = Properties.create(); props.set("k1", "v1"); props.set("k2", "v2"); EPGMEdge edge = new EdgeFactory().initEdge(edgeId, label, sourceId, targetId, props); HashMap<String, String> newProps = new HashMap<>(); newProps.put("k1", "new_k1"); TransformationFunction<EPGMEdge> renameFunction = new RenamePropertyKeys<>(newProps); renameFunction.apply(edge, edge); assertThat(edge.getPropertyCount(), is(2)); assertEquals(label, edge.getLabel()); assertThat(edge.getPropertyValue("new_k1").toString(), Is.<Object>is("v1")); assertThat(edge.getPropertyValue("k2").toString(), Is.<Object>is("v2")); assertNull(edge.getPropertyValue("k1")); }
@Test public void createEdgePojoTest() { GradoopId edgeId = GradoopId.get(); GradoopId sourceId = GradoopId.get(); GradoopId targetId = GradoopId.get(); GradoopIdSet graphIds = GradoopIdSet .fromExisting(GradoopId.get(), GradoopId.get()); String label = "A"; Properties props = Properties.create(); props.set("k1", "v1"); props.set("k2", "v2"); EPGMEdge edge = new EdgeFactory() .initEdge(edgeId, label, sourceId, targetId, props, graphIds); assertThat(edge.getId(), is(edgeId)); assertEquals(label, edge.getLabel()); assertThat(edge.getSourceId(), is(sourceId)); assertThat(edge.getTargetId(), is(targetId)); assertThat(edge.getPropertyCount(), is(2)); assertThat(edge.getPropertyValue("k1").getString(), Is.<Object>is("v1")); assertThat(edge.getPropertyValue("k2").getString(), Is.<Object>is("v2")); assertThat(edge.getGraphCount(), is(2)); for (GradoopId graphId : graphIds) { assertTrue(edge.getGraphIds().contains(graphId)); } }