SchemaUtils { public Map<String, String> getColumnToSerialiser(final String group) { return groupColumnToSerialiserName.get(group); } 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 getColumnToSerialiserTest() { final Map<String, String> columnToSerialiser = utils.getColumnToSerialiser(TestGroups.EDGE); assertEquals("uk.gov.gchq.gaffer.parquetstore.serialisation.impl.StringParquetSerialiser", columnToSerialiser.get(ParquetStore.SOURCE)); assertEquals("uk.gov.gchq.gaffer.parquetstore.serialisation.impl.StringParquetSerialiser", columnToSerialiser.get(ParquetStore.DESTINATION)); assertEquals("uk.gov.gchq.gaffer.parquetstore.serialisation.impl.BooleanParquetSerialiser", columnToSerialiser.get(ParquetStore.DIRECTED)); assertEquals("uk.gov.gchq.gaffer.parquetstore.serialisation.impl.ByteParquetSerialiser", columnToSerialiser.get("byte")); assertEquals("uk.gov.gchq.gaffer.parquetstore.serialisation.impl.DoubleParquetSerialiser", columnToSerialiser.get("double")); assertEquals("uk.gov.gchq.gaffer.parquetstore.serialisation.impl.FloatParquetSerialiser", columnToSerialiser.get("float")); assertEquals("uk.gov.gchq.gaffer.parquetstore.serialisation.impl.TreeSetStringParquetSerialiser", columnToSerialiser.get("treeSet")); assertEquals("uk.gov.gchq.gaffer.parquetstore.serialisation.impl.LongParquetSerialiser", columnToSerialiser.get("long")); assertEquals("uk.gov.gchq.gaffer.parquetstore.serialisation.impl.ShortParquetSerialiser", columnToSerialiser.get("short")); assertEquals("uk.gov.gchq.gaffer.parquetstore.serialisation.impl.DateParquetSerialiser", columnToSerialiser.get("date")); assertEquals("uk.gov.gchq.gaffer.parquetstore.serialisation.impl.FreqMapParquetSerialiser", columnToSerialiser.get("freqMap")); assertEquals("uk.gov.gchq.gaffer.parquetstore.serialisation.impl.IntegerParquetSerialiser", columnToSerialiser.get("count")); }