use of uk.gov.gchq.koryphe.function.FunctionComposite in project Gaffer by gchq.
the class KeyFunctionMatchTest method shouldMatchElementsOfDifferentGroupsBasedOnKeyFunctions.
@Test
public void shouldMatchElementsOfDifferentGroupsBasedOnKeyFunctions() {
// given
Entity testItem = new Entity.Builder().group(TEST_ENTITY_GROUP).vertex("test").property(PROP_1, 2L).build();
List<Entity> testList = Lists.newArrayList(new Entity.Builder().group(TEST_ENTITY_GROUP_2).vertex("test2").property(PROP_2, 2).build(), new Entity.Builder().group(TEST_ENTITY_GROUP).vertex("test3").property(PROP_1, 3L).build(), new Entity.Builder().group(TEST_ENTITY_GROUP).vertex("test1").property(PROP_1, 4L).build(), new Entity.Builder().group(TEST_ENTITY_GROUP_2).vertex("test3").property(PROP_2, 2).build());
// when
KeyFunctionMatch match = new KeyFunctionMatch.Builder().firstKeyFunction(new ExtractProperty(PROP_1)).secondKeyFunction(new FunctionComposite(Lists.newArrayList(new ExtractProperty(PROP_2), new ToLong()))).build();
match.init(testList);
// then
ArrayList<Entity> expected = Lists.newArrayList(new Entity.Builder().group(TEST_ENTITY_GROUP_2).vertex("test2").property(PROP_2, 2).build(), new Entity.Builder().group(TEST_ENTITY_GROUP_2).vertex("test3").property(PROP_2, 2).build());
assertEquals(expected, match.matching(testItem));
}
use of uk.gov.gchq.koryphe.function.FunctionComposite in project Gaffer by gchq.
the class KeyFunctionMatchTest method shouldThrowExceptionFromFunctionIfInputIsInvalid.
@Test
public void shouldThrowExceptionFromFunctionIfInputIsInvalid() {
// given
// Performing a FirstItem on null should throw IllegalArgumentException
List<Long> testList = Lists.newArrayList(100L, 200L, 300L, null);
// when
KeyFunctionMatch match = new KeyFunctionMatch.Builder().firstKeyFunction(new FunctionComposite(Lists.newArrayList(new CallMethod("getValue"), new ToInteger()))).secondKeyFunction(new FunctionComposite(Lists.newArrayList(new ToInteger(), new DivideBy(10), new FirstItem<>()))).build();
// then
try {
match.init(testList);
} catch (final IllegalArgumentException e) {
// copied from docs of FirstItem
assertEquals("Input cannot be null", e.getMessage());
}
}
use of uk.gov.gchq.koryphe.function.FunctionComposite in project Gaffer by gchq.
the class KeyFunctionMatchTest method shouldJsonSerialiseAndDeserialiseWithKeyFunctions.
@Test
public void shouldJsonSerialiseAndDeserialiseWithKeyFunctions() throws SerialisationException {
// given
KeyFunctionMatch match = new KeyFunctionMatch.Builder().firstKeyFunction(new FunctionComposite(Lists.newArrayList(new DivideBy(20), new FirstItem()))).secondKeyFunction(new ExtractProperty("count")).build();
// when / then
String expected = "{\n" + " \"class\" : \"uk.gov.gchq.gaffer.store.operation.handler.join.match.KeyFunctionMatch\",\n" + " \"firstKeyFunction\" : {\n" + " \"class\" : \"uk.gov.gchq.koryphe.function.FunctionComposite\",\n" + " \"functions\" : [ {\n" + " \"class\" : \"uk.gov.gchq.koryphe.impl.function.DivideBy\",\n" + " \"by\" : 20\n" + " }, {\n" + " \"class\" : \"uk.gov.gchq.koryphe.impl.function.FirstItem\"\n" + " } ]\n" + " },\n" + " \"secondKeyFunction\" : {\n" + " \"class\" : \"uk.gov.gchq.gaffer.data.element.function.ExtractProperty\",\n" + " \"name\" : \"count\"\n" + " }\n" + "}";
assertEquals(expected, new String(JSONSerialiser.serialise(match, true)));
assertEquals(match, JSONSerialiser.deserialise(expected, KeyFunctionMatch.class));
}
use of uk.gov.gchq.koryphe.function.FunctionComposite in project Gaffer by gchq.
the class KeyFunctionMatchTest method shouldMatchObjectsBasedOnKeyFunctions.
@Test
public void shouldMatchObjectsBasedOnKeyFunctions() {
// given
TypeSubTypeValue testValue = new TypeSubTypeValue("myType", "mySubType", "30");
List<Long> testList = Lists.newArrayList(100L, 200L, 300L, 400L);
// when
KeyFunctionMatch match = new KeyFunctionMatch.Builder().firstKeyFunction(new FunctionComposite(Lists.newArrayList(new CallMethod("getValue"), new ToInteger()))).secondKeyFunction(new FunctionComposite(Lists.newArrayList(new ToInteger(), new DivideBy(10), new FirstItem<>()))).build();
match.init(testList);
// then
List<Long> expected = Lists.newArrayList(300L);
assertEquals(expected, match.matching(testValue));
}
use of uk.gov.gchq.koryphe.function.FunctionComposite in project Gaffer by gchq.
the class TypeValueToTupleTest method shouldGetAndSetUsingCompositeFunction.
@Test
public void shouldGetAndSetUsingCompositeFunction() {
// Given
final TypeValue typeValue = new TypeValue("type", "value");
final Function<Object, Object> compositeFunction = new FunctionComposite(Lists.newArrayList(new TypeValueToTuple(), new TupleAdaptedFunctionComposite.Builder().select(new String[] { "value" }).execute(new FunctionComposite(Arrays.asList(new Length(), new ToString()))).project(new String[] { "type" }).build()));
// When
compositeFunction.apply(typeValue);
// Then
assertEquals(new TypeValue("5", "value"), typeValue);
}
Aggregations