SchemaUtils { public Set<String> getEdgeGroups() { return gafferSchema.getEdgeGroups(); } SchemaUtils(final Schema gafferSchema); MessageType getParquetSchema(final String group); List<String> getCoreProperties(final String group); List<String> getCorePropertiesForReversedEdges(); List<String> columnsToSortBy(final String group, final boolean reversed); Map<String, String[]> getColumnToPaths(final String group); String[] getPaths(final String group, final String column); Map<String, GafferGroupObjectConverter> getGroupToObjectConverter(); StructType buildSparkSchema(final String group); StructType getSparkSchema(final String group); StructType getMergedSparkSchema(final Set<String> groups); Serialiser getSerialiser(final String serialiserClassName); Map<String, Serialiser> getSerialisers(); Map<String, String> getColumnToSerialiser(final String group); Set<String> getEntityGroups(); Set<String> getEdgeGroups(); Set<String> getGroups(); Schema getGafferSchema(); GafferGroupObjectConverter getConverter(final String group);  }
@Test public void getEdgeGroupsTest() { final Set<String> edgeGroups = utils.getEdgeGroups(); final LinkedHashSet<String> expected = new LinkedHashSet<>(2); expected.add(TestGroups.EDGE); expected.add(TestGroups.EDGE_2); assertEquals(expected, edgeGroups); }