use of uk.gov.gchq.koryphe.impl.function.DivideBy in project Gaffer by gchq.
the class FunctionAuthoriserTest method shouldNotAllowGetElementsOperationWithUnauthorisedFunctionsInTheView.
@Test
public void shouldNotAllowGetElementsOperationWithUnauthorisedFunctionsInTheView() {
final OperationChain<CloseableIterable<? extends Element>> viewOperation = new OperationChain.Builder().first(new GetElements.Builder().view(new View.Builder().globalElements(new GlobalViewElementDefinition.Builder().transformFunctions(Lists.newArrayList(new TupleAdaptedFunction(new String[] { "input" }, new DivideBy(6), new String[] { "output" }))).build()).build()).build()).build();
FunctionAuthoriser functionAuthoriser = new FunctionAuthoriser();
// When
functionAuthoriser.setUnauthorisedFunctions(Lists.newArrayList(DivideBy.class));
// Then
assertThatExceptionOfType(UnauthorisedException.class).isThrownBy(() -> functionAuthoriser.preExecute(viewOperation, new Context())).withMessage("Operation chain contained an unauthorised function: uk.gov.gchq.koryphe.impl.function.DivideBy");
}
use of uk.gov.gchq.koryphe.impl.function.DivideBy 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.impl.function.DivideBy in project gaffer-doc by gchq.
the class DivideByExample method divideBy2.
public void divideBy2() {
// ---------------------------------------------------------
final DivideBy function = new DivideBy(2);
// ---------------------------------------------------------
runExample(function, null, 6, 5, null, 6.1);
}
use of uk.gov.gchq.koryphe.impl.function.DivideBy 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.impl.function.DivideBy 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));
}
Aggregations