Examples with OperationDeclarations uk.gov.gchq.gaffer.store.operationdeclaration.OperationDeclarations used on opensource projects

Example 6 with OperationDeclarations

use of uk.gov.gchq.gaffer.store.operationdeclaration.OperationDeclarations in project Gaffer by gchq.

the class NamedOperationHandlerTest method shouldLoadCacheFromOperationsDeclarationsFile.

@Test
public void shouldLoadCacheFromOperationsDeclarationsFile() throws SerialisationException {
    final InputStream s = StreamUtil.openStream(getClass(), "NamedOperationDeclarations.json");
    final OperationDeclarations deserialised = json.deserialise(s, OperationDeclarations.class);
    INamedOperationCache cache = ((AddNamedOperationHandler) deserialised.getOperations().get(0).getHandler()).getCache();
    assert (cache instanceof MockNamedOperationCache);
}

Example 7 with OperationDeclarations

use of uk.gov.gchq.gaffer.store.operationdeclaration.OperationDeclarations in project Gaffer by gchq.

the class StoreTest method shouldCreateStoreWithValidSchemasAndRegisterOperations.

@Test
public void shouldCreateStoreWithValidSchemasAndRegisterOperations() throws StoreException {
    // Given
    final StoreProperties properties = mock(StoreProperties.class);
    final StoreImpl store = new StoreImpl();
    final OperationHandler<AddElements, Void> 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);
    // When
    store.initialise(schema, properties);
    // Then
    assertNotNull(store.getOperationHandlerExposed(Validate.class));
    assertSame(addElementsHandlerOverridden, store.getOperationHandlerExposed(AddElements.class));
    assertSame(getAllElementsHandler, store.getOperationHandlerExposed(GetAllElements.class));
    assertSame(getAllElementsHandler, store.getOperationHandlerExposed(GetAllEntities.class));
    assertSame(getAllElementsHandler, store.getOperationHandlerExposed(GetAllEdges.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(Deduplicate.class) instanceof DeduplicateHandler);
    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(schema, true);
}