@Override protected ExtractGroup getInstance() { return new ExtractGroup(); }
@Test public void shouldReturnNullForNullElement() { // Given final ExtractGroup function = new ExtractGroup(); // When final String result = function.apply(null); // Then assertNull(result); }
@Test public void shouldReturnGroupFromEntity() { // Given final ExtractGroup function = new ExtractGroup(); final String group = "testGroup_2"; final Entity entity = new Entity.Builder() .vertex("1") .group(group) .build(); // When final String result = function.apply(entity); // Then assertEquals(group, result); }
@Test public void shouldReturnGroupFromEdge() { // Given final ExtractGroup function = new ExtractGroup(); final String group = "testGroup"; final Edge edge = new Edge.Builder() .source("src") .dest("dest") .directed(true) .group(group) .build(); // When final String result = function.apply(edge); // Then assertEquals(group, result); }