use of uk.gov.gchq.gaffer.store.operation.handler.CountGroupsHandler in project Gaffer by gchq.
the class StoreTest method shouldCreateStoreWithValidSchemasAndRegisterOperations.
@Test
public void shouldCreateStoreWithValidSchemasAndRegisterOperations() throws StoreException {
// Given
final StoreProperties properties = mock(StoreProperties.class);
final OperationHandler<AddElements> addElementsHandlerOverridden = mock(OperationHandler.class);
final OperationDeclarations opDeclarations = new OperationDeclarations.Builder().declaration(new OperationDeclaration.Builder().operation(AddElements.class).handler(addElementsHandlerOverridden).build()).build();
given(properties.getOperationDeclarations()).willReturn(opDeclarations);
given(properties.getJobExecutorThreadCount()).willReturn(1);
// When
store.initialise("graphId", schema, properties);
// Then
assertNotNull(store.getOperationHandlerExposed(Validate.class));
assertSame(addElementsHandlerOverridden, store.getOperationHandlerExposed(AddElements.class));
assertSame(getAllElementsHandler, store.getOperationHandlerExposed(GetAllElements.class));
assertTrue(store.getOperationHandlerExposed(GenerateElements.class) instanceof GenerateElementsHandler);
assertTrue(store.getOperationHandlerExposed(GenerateObjects.class) instanceof GenerateObjectsHandler);
assertTrue(store.getOperationHandlerExposed(CountGroups.class) instanceof CountGroupsHandler);
assertTrue(store.getOperationHandlerExposed(ToSet.class) instanceof ToSetHandler);
assertTrue(store.getOperationHandlerExposed(ExportToSet.class) instanceof ExportToSetHandler);
assertTrue(store.getOperationHandlerExposed(GetSetExport.class) instanceof GetSetExportHandler);
assertEquals(1, store.getCreateOperationHandlersCallCount());
assertSame(schema, store.getSchema());
assertSame(properties, store.getProperties());
verify(schemaOptimiser).optimise(store.getSchema(), true);
}
Aggregations