Examples with GenerateElementsHandler uk.gov.gchq.gaffer.store.operation.handler.generate.GenerateElementsHandler used on opensource projects

Example 1 with GenerateElementsHandler

use of uk.gov.gchq.gaffer.store.operation.handler.generate.GenerateElementsHandler in project Gaffer by gchq.

the class OperationDeclarationsTest method testDeserialiseFile.

@Test
public void testDeserialiseFile() throws SerialisationException {
    // Given
    final String paths = "operationDeclarations1.json,operationDeclarations2.json";
    // When
    final OperationDeclarations deserialised = OperationDeclarations.fromPaths(paths);
    // Then
    assertEquals(2, deserialised.getOperations().size());
    final OperationDeclaration od0 = deserialised.getOperations().get(0);
    final OperationDeclaration od1 = deserialised.getOperations().get(1);
    assertEquals(GenerateElements.class, od0.getOperation());
    assertTrue(od0.getHandler() instanceof GenerateElementsHandler);
    assertEquals(GenerateObjects.class, od1.getOperation());
    assertTrue(od1.getHandler() instanceof GenerateObjectsHandler);
}

Example 2 with GenerateElementsHandler

use of uk.gov.gchq.gaffer.store.operation.handler.generate.GenerateElementsHandler in project Gaffer by gchq.

the class OperationDeclarationsTest method testSerialiseDeserialise.

@Test
public void testSerialiseDeserialise() throws SerialisationException {
    // Given
    final OperationDeclarations declarations = new OperationDeclarations.Builder().declaration(new OperationDeclaration.Builder().handler(new GenerateElementsHandler()).operation(GenerateElements.class).build()).build();
    // When
    final byte[] definitionJson = JSONSerialiser.serialise(declarations);
    final OperationDeclarations deserialised = OperationDeclarations.fromJson(definitionJson);
    assertEquals(1, deserialised.getOperations().size());
    final OperationDeclaration deserialisedDeclaration = deserialised.getOperations().get(0);
    assertEquals(GenerateElements.class, deserialisedDeclaration.getOperation());
    assertTrue(deserialisedDeclaration.getHandler() instanceof GenerateElementsHandler);
}

Example 3 with GenerateElementsHandler

use of uk.gov.gchq.gaffer.store.operation.handler.generate.GenerateElementsHandler in project Gaffer by gchq.

the class AccumuloStoreTest method testStoreReturnsHandlersForRegisteredOperations.

public void testStoreReturnsHandlersForRegisteredOperations(final AccumuloStore store) {
    OperationHandler op;
    // Then
    assertNotNull(store.getOperationHandler(Validate.class));
    op = store.getOperationHandler(AddElementsFromHdfs.class);
    assertTrue(op instanceof AddElementsFromHdfsHandler);
    op = store.getOperationHandler(GetElementsBetweenSets.class);
    assertTrue(op instanceof GetElementsBetweenSetsHandler);
    op = store.getOperationHandler(GetElementsInRanges.class);
    assertTrue(op instanceof GetElementsInRangesHandler);
    op = store.getOperationHandler(GetElementsWithinSet.class);
    assertTrue(op instanceof GetElementsWithinSetHandler);
    op = store.getOperationHandler(SplitStore.class);
    assertTrue(op instanceof SplitStoreHandler);
    op = store.getOperationHandler(SampleDataForSplitPoints.class);
    assertTrue(op instanceof SampleDataForSplitPointsHandler);
    op = store.getOperationHandler(ImportAccumuloKeyValueFiles.class);
    assertTrue(op instanceof ImportAccumuloKeyValueFilesHandler);
    op = store.getOperationHandler(GenerateElements.class);
    assertTrue(op instanceof GenerateElementsHandler);
    op = store.getOperationHandler(GenerateObjects.class);
    assertTrue(op instanceof GenerateObjectsHandler);
}

Example 4 with GenerateElementsHandler

use of uk.gov.gchq.gaffer.store.operation.handler.generate.GenerateElementsHandler 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);
}