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

Example 1 with OperationDeclaration

use of uk.gov.gchq.gaffer.store.operationdeclaration.OperationDeclaration 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 OperationDeclaration

use of uk.gov.gchq.gaffer.store.operationdeclaration.OperationDeclaration 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 = json.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 OperationDeclaration

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

the class SparkOperationDeclarationsTest method shouldContainAllSparkOperationsAndHandlers.

@Test
public void shouldContainAllSparkOperationsAndHandlers() throws SerialisationException {
    // Given
    final JSONSerialiser jsonSerialiser = new JSONSerialiser();
    // When
    final OperationDeclarations deserialised = jsonSerialiser.deserialise(StreamUtil.openStream(getClass(), ACCUMULO_OP_DECLARATIONS_JSON_PATH), OperationDeclarations.class);
    // Then
    assertEquals(9, deserialised.getOperations().size());
    final OperationDeclaration od0 = deserialised.getOperations().get(0);
    assertEquals(GetJavaRDDOfElements.class, od0.getOperation());
    assertTrue(od0.getHandler() instanceof GetJavaRDDOfElementsHandler);
    final OperationDeclaration od1 = deserialised.getOperations().get(1);
    assertEquals(GetRDDOfElements.class, od1.getOperation());
    assertTrue(od1.getHandler() instanceof GetRDDOfElementsHandler);
    final OperationDeclaration od2 = deserialised.getOperations().get(2);
    assertEquals(GetRDDOfAllElements.class, od2.getOperation());
    assertTrue(od2.getHandler() instanceof GetRDDOfAllElementsHandler);
    final OperationDeclaration od3 = deserialised.getOperations().get(3);
    assertEquals(GetJavaRDDOfAllElements.class, od3.getOperation());
    assertTrue(od3.getHandler() instanceof GetJavaRDDOfAllElementsHandler);
    final OperationDeclaration od4 = deserialised.getOperations().get(4);
    assertEquals(GetDataFrameOfElements.class, od4.getOperation());
    assertTrue(od4.getHandler() instanceof GetDataFrameOfElementsHandler);
    final OperationDeclaration od5 = deserialised.getOperations().get(5);
    assertEquals(ImportKeyValueJavaPairRDDToAccumulo.class, od5.getOperation());
    assertTrue(od5.getHandler() instanceof ImportKeyValueJavaPairRDDToAccumuloHandler);
    final OperationDeclaration od6 = deserialised.getOperations().get(6);
    assertEquals(ImportJavaRDDOfElements.class, od6.getOperation());
    assertTrue(od6.getHandler() instanceof ImportJavaRDDOfElementsHandler);
    final OperationDeclaration od7 = deserialised.getOperations().get(7);
    assertEquals(ImportKeyValuePairRDDToAccumulo.class, od7.getOperation());
    assertTrue(od7.getHandler() instanceof ImportKeyValuePairRDDToAccumuloHandler);
    final OperationDeclaration od8 = deserialised.getOperations().get(8);
    assertEquals(ImportRDDOfElements.class, od8.getOperation());
    assertTrue(od8.getHandler() instanceof ImportRDDOfElementsHandler);
}