Examples with ElementGeneratorImpl uk.gov.gchq.gaffer.data.generator.ElementGeneratorImpl used on opensource projects

Example 1 with ElementGeneratorImpl

use of uk.gov.gchq.gaffer.data.generator.ElementGeneratorImpl in project Gaffer by gchq.

the class GenerateElementsTest method builderShouldCreatePopulatedOperation.

@Test
@Override
public void builderShouldCreatePopulatedOperation() {
    GenerateElements generateElements = new GenerateElements.Builder<String>().generator(new ElementGeneratorImpl()).objects(Arrays.asList("Test1", "Test2")).view(new View.Builder().edge("TestEdgeGroup").build()).option("testOption", "true").build();
    assertNotNull(generateElements.getView());
    assertEquals("true", generateElements.getOption("testOption"));
    Iterator iter = generateElements.getInput().iterator();
    assertEquals("Test1", iter.next());
    assertEquals("Test2", iter.next());
}

Example 2 with ElementGeneratorImpl

use of uk.gov.gchq.gaffer.data.generator.ElementGeneratorImpl in project Gaffer by gchq.

the class GenerateObjectsTest method builderShouldCreatePopulatedOperation.

@Test
@Override
public void builderShouldCreatePopulatedOperation() {
    Element entity = new Entity("testEntityGroup", "A");
    GenerateObjects generateObjects = new GenerateObjects.Builder<Element, String>().elements(Arrays.asList(entity)).generator(new ElementGeneratorImpl()).option("testOption", "true").view(new View.Builder().edge("testEntityGroup").build()).build();
    assertEquals(entity, generateObjects.getElements().iterator().next());
    assertEquals(ElementGeneratorImpl.class, generateObjects.getElementGenerator().getClass());
    assertEquals("true", generateObjects.getOption("testOption"));
    assertNotNull(generateObjects.getView());
}

Example 3 with ElementGeneratorImpl

use of uk.gov.gchq.gaffer.data.generator.ElementGeneratorImpl in project Gaffer by gchq.

the class GenerateElementsTest method shouldSerialiseAndDeserialiseOperation.

@Test
@Override
public void shouldSerialiseAndDeserialiseOperation() throws SerialisationException {
    // Given
    final GenerateElements<String> op = new GenerateElements<>(Arrays.asList("obj 1", "obj 2"), new ElementGeneratorImpl());
    // When
    byte[] json = serialiser.serialise(op, true);
    final GenerateElements deserialisedOp = serialiser.deserialise(json, GenerateElements.class);
    // Then
    final Iterator itr = deserialisedOp.getInput().iterator();
    assertEquals("obj 1", itr.next());
    assertEquals("obj 2", itr.next());
    assertFalse(itr.hasNext());
    assertTrue(deserialisedOp.getElementGenerator() instanceof ElementGeneratorImpl);
}

Example 4 with ElementGeneratorImpl

use of uk.gov.gchq.gaffer.data.generator.ElementGeneratorImpl in project Gaffer by gchq.

the class GenerateObjectsTest method shouldSerialiseAndDeserialiseOperation.

@Test
@Override
public void shouldSerialiseAndDeserialiseOperation() throws SerialisationException {
    // Given
    final List<Element> elements = new ArrayList<>();
    {
        final Entity elm1 = new Entity("entity type 1", "vertex 1");
        elm1.putProperty("property 1", "property 1 value");
        elements.add(elm1);
    }
    {
        final Edge elm2 = new Edge("edge type 2", "source vertex 1", "dest vertex 1", true);
        elm2.putProperty("property 2", "property 2 value");
        elements.add(elm2);
    }
    final GenerateObjects<Element, String> op = new GenerateObjects<>(elements, new ElementGeneratorImpl());
    // When
    byte[] json = serialiser.serialise(op, true);
    final GenerateObjects deserialisedOp = serialiser.deserialise(json, GenerateObjects.class);
    // Then
    final Iterator<Element> itr = deserialisedOp.getElements().iterator();
    final Entity elm1 = (Entity) itr.next();
    assertEquals("vertex 1", elm1.getVertex());
    assertEquals(1, elm1.getProperties().size());
    assertEquals("property 1 value", elm1.getProperty("property 1"));
    final Edge elm2 = (Edge) itr.next();
    assertEquals("source vertex 1", elm2.getSource());
    assertEquals("dest vertex 1", elm2.getDestination());
    assertTrue(elm2.isDirected());
    assertEquals(1, elm2.getProperties().size());
    assertEquals("property 2 value", elm2.getProperty("property 2"));
    assertFalse(itr.hasNext());
    assertTrue(deserialisedOp.getElementGenerator() instanceof ElementGeneratorImpl);
}