Gh-3220: Standardise GafferPop Tests (#3243)

* update graph IT tests

* Rename GraphIT -> GraphTest

* create graph IT

* Create Modern Federated example graph

run graph tests on multiple graph types

* Update federated IT tests

move non-federates specific tests to graph tests

* tidying

* remove public + spotless

* sonarcloud

* sonarcloud

* Use allEdges/allEntities

* tidy

---------

Co-authored-by: wb36499 <166839644+wb36499@users.noreply.github.com>

Author: p29876

library/tinkerpop/src/main/java/uk/gov/gchq/gaffer/tinkerpop/GafferPopGraph.java

@@ -249,7 +249,7 @@ public enum DefaultIdManager {
 
     private static final Logger LOGGER = LoggerFactory.getLogger(GafferPopGraph.class);
     private static final String GET_ALL_DEBUG_MSG = "Requested a GetAllElements, results will be truncated to: {}.";
-    private static final Pattern EDGE_ID_REGEX = Pattern.compile("^\\[\\s*(?<src>[a-zA-Z0-9|-]*)\\s*(,\\s*(?<label>[a-zA-Z0-9|-]*))?\\s*,\\s*(?<dest>[a-zA-Z0-9|-]*)\\s*\\]$");
+    private static final Pattern EDGE_ID_REGEX = Pattern.compile("^\\s*\\[\\s*(?<src>[a-zA-Z0-9|-]*)\\s*(,\\s*(?<label>[a-zA-Z0-9|-]*))?\\s*,\\s*(?<dest>[a-zA-Z0-9|-]*)\\s*\\]\\s*$");
 
     public GafferPopGraph(final Configuration configuration) {
         this(configuration, createGraph(configuration));
@@ -405,19 +405,15 @@ public Iterator<Vertex> vertices(final Object... vertexIds) {
             LOGGER.debug(GET_ALL_DEBUG_MSG, variables.getAllElementsLimit());
             getOperation = new Builder()
                     .first(new GetAllElements.Builder()
-                            .view(new View.Builder()
-                                    .entities(graph.getSchema().getEntityGroups())
-                                    .build())
+                            .view(createAllEntitiesView())
                             .build())
                     .then(new Limit<Element>(variables.getAllElementsLimit(), true))
                     .build();
         } else {
             getOperation = new Builder()
                 .first(new GetElements.Builder()
                     .input(getElementSeeds(Arrays.asList(vertexIds)))
-                    .view(new View.Builder()
-                            .entities(graph.getSchema().getEntityGroups())
-                            .build())
+                    .view(createAllEntitiesView())
                     .build())
                 .build();
         }
@@ -558,19 +554,26 @@ public Iterator<Edge> edges(final Object... elementIds) {
             LOGGER.debug(GET_ALL_DEBUG_MSG, variables.getAllElementsLimit());
             getOperation = new Builder()
                 .first(new GetAllElements.Builder()
-                        .view(new View.Builder()
-                            .edges(graph.getSchema().getEdgeGroups())
-                            .build())
+                        .view(createAllEdgesView())
                         .build())
                 .then(new Limit<>(variables.getAllElementsLimit(), true))
                 .build();
         } else {
+
+            View.Builder builder = new View.Builder();
+            Set<String> edgeLabels = getEdgeLabelsFromIds(Arrays.asList(elementIds));
+            if (edgeLabels.isEmpty()) {
+                // Default to all edges
+                builder.allEdges(true);
+            } else {
+                // Get requested edges
+                builder.edges(edgeLabels);
+            }
+
             getOperation = new Builder()
                 .first(new GetElements.Builder()
                     .input(getElementSeeds(Arrays.asList(elementIds)))
-                    .view(new View.Builder()
-                        .edges(getEdgeViewGroup(Arrays.asList(elementIds)))
-                        .build())
+                    .view(builder.build())
                     .build())
                 .build();
         }
@@ -725,9 +728,7 @@ private Iterator<GafferPopVertex> verticesWithSeedsAndView(final List<ElementSee
 
         View entitiesView = view;
         if (null == entitiesView) {
-            entitiesView = new View.Builder()
-                    .entities(graph.getSchema().getEntityGroups())
-                    .build();
+            entitiesView = createAllEntitiesView();
         } else if (entitiesView.hasEdges()) {
             entitiesView = new View.Builder()
                     .merge(entitiesView)
@@ -781,15 +782,16 @@ private Iterator<Vertex> adjVerticesWithSeedsAndView(final List<ElementSeed> see
             throw new UnsupportedOperationException("There could be a lot of vertices, so please add some seeds");
         }
 
-        View processedView = view == null ? createAllEntitiesView() : view;
         final Iterable<? extends Element> result = execute(new OperationChain.Builder()
                 .first(new GetAdjacentIds.Builder()
                         .input(seeds)
-                        .view(processedView)
+                        .view(view)
                         .inOutType(getInOutType(direction))
                         .build())
                 // GetAdjacentIds provides list of entity seeds so run a GetElements to get the actual Entities
-                .then(new GetElements())
+                .then(new GetElements.Builder()
+                        .view(createAllEntitiesView())
+                        .build())
                 .build());
 
         // Translate results to Gafferpop elements
@@ -808,9 +810,7 @@ private Iterator<Edge> edgesWithSeedsAndView(final List<ElementSeed> seeds, fina
 
         View edgesView = view;
         if (null == edgesView) {
-            edgesView = new View.Builder()
-                    .edges(graph.getSchema().getEdgeGroups())
-                    .build();
+            edgesView = createAllEdgesView();
         } else if (edgesView.hasEntities()) {
             edgesView = new View.Builder()
                     .merge(edgesView)
@@ -888,11 +888,11 @@ private View createView(final String... labels) {
     }
 
     private View createAllEntitiesView() {
-        final View.Builder viewBuilder = new View.Builder();
-        for (final String group : graph.getSchema().getEntityGroups()) {
-            viewBuilder.entity(group);
-        }
-        return viewBuilder.build();
+        return new View.Builder().allEntities(true).build();
+    }
+
+    private View createAllEdgesView() {
+        return new View.Builder().allEdges(true).build();
     }
 
     /**
@@ -941,17 +941,17 @@ private List<ElementSeed> getElementSeeds(final Iterable<Object> ids) {
     }
 
     /**
-     * Determines the edge group used in the view based on supplied IDs.
-     * If ID contains a label, this is extracted and used in the view.
-     * If not, all edge groups are used.
+     * Extracts edge labels from edge IDs if found.
+     * All ids must be of the [src, label, dest] format.
+     * Otherwise, returns an empty set.
      *
      * @param ids The iterable of IDs
      * @return Set of edge labels for the view
      */
-    private Set<String> getEdgeViewGroup(final Iterable<Object> ids) {
+    private Set<String> getEdgeLabelsFromIds(final Iterable<Object> ids) {
         Set<String> labels = new HashSet<>();
 
-        ids.forEach(id -> {
+        for (final Object id: ids) {
             if ((id instanceof String) && (EDGE_ID_REGEX.matcher((String) id).matches())) {
                 Matcher edgeIdWithLabelMatcher = EDGE_ID_REGEX.matcher((String) id);
 
@@ -960,10 +960,11 @@ private Set<String> getEdgeViewGroup(final Iterable<Object> ids) {
                     labels.add(edgeIdWithLabelMatcher.group("label"));
                 }
             } else {
-                // Fallback is to use all schema edge groups
-                labels.addAll(graph.getSchema().getEdgeGroups());
+                // If a single ID isn't of the format [src, label, dest]
+                // Then all edge labels must be used
+                return Collections.emptySet();
             }
-        });
+        }
 
         return labels;
     }