use of uk.gov.gchq.gaffer.serialisation.implementation.SerialisationFactory in project Gaffer by gchq.
the class SchemaOptimiserTest method shouldAddDefaultSerialisers.
@Test
public void shouldAddDefaultSerialisers() {
//Given
final SerialisationFactory serialisationFactory = mock(SerialisationFactory.class);
final SchemaOptimiser optimiser = new SchemaOptimiser(serialisationFactory);
final boolean isOrdered = true;
final StringSerialiser stringSerialiser = mock(StringSerialiser.class);
final StringSerialiser intSerialiser = mock(StringSerialiser.class);
final JavaSerialiser javaSerialiser = mock(JavaSerialiser.class);
given(serialisationFactory.getSerialiser(String.class, true)).willReturn(stringSerialiser);
given(serialisationFactory.getSerialiser(Integer.class, false)).willReturn(intSerialiser);
given(serialisationFactory.getSerialiser(Serializable.class, true)).willReturn(javaSerialiser);
given(javaSerialiser.canHandle(Mockito.any(Class.class))).willReturn(true);
schema = new Schema.Builder().merge(schema).type("obj", Serializable.class).entity(TestGroups.ENTITY_2, new SchemaEntityDefinition.Builder().vertex("obj").build()).build();
// When
final Schema optimisedSchema = optimiser.optimise(schema, isOrdered);
// Then
assertSame(stringSerialiser, optimisedSchema.getType("string").getSerialiser());
assertSame(intSerialiser, optimisedSchema.getType("int").getSerialiser());
assertSame(javaSerialiser, optimisedSchema.getVertexSerialiser());
verify(serialisationFactory, never()).getSerialiser(String.class, false);
verify(serialisationFactory, never()).getSerialiser(Serializable.class, false);
}
Aggregations