Examples with NamedOperationCache uk.gov.gchq.gaffer.store.operation.handler.named.cache.NamedOperationCache used on opensource projects

Example 6 with NamedOperationCache

use of uk.gov.gchq.gaffer.store.operation.handler.named.cache.NamedOperationCache in project Gaffer by gchq.

the class NamedOperationScoreResolverTest method shouldGetScore.

@Test
public void shouldGetScore() throws CacheOperationFailedException {
    // Given
    final Integer expectedScore = 5;
    final String opName = "otherOp";
    final NamedOperation<Element, Iterable<? extends Element>> namedOp = mock(NamedOperation.class);
    namedOp.setOperationName(opName);
    final NamedOperationDetail namedOpDetail = mock(NamedOperationDetail.class);
    final NamedOperationCache cache = mock(NamedOperationCache.class);
    final NamedOperationScoreResolver resolver = new NamedOperationScoreResolver(cache);
    given(cache.getFromCache(namedOpDetail.getOperationName())).willReturn(namedOpDetail);
    given(namedOpDetail.getOperationName()).willReturn(opName);
    given(namedOpDetail.getScore()).willReturn(expectedScore);
    // When
    final Integer result = resolver.getScore(namedOp);
    // Then
    assertEquals(expectedScore, result);
}

Example 7 with NamedOperationCache

use of uk.gov.gchq.gaffer.store.operation.handler.named.cache.NamedOperationCache in project Gaffer by gchq.

the class NamedOperationResolverTest method shouldNotExecuteNamedOperationWithParameterOfWrongType.

@Test
public void shouldNotExecuteNamedOperationWithParameterOfWrongType() throws OperationException, CacheOperationFailedException {
    // Given
    final String opName = "opName";
    final NamedOperationCache cache = mock(NamedOperationCache.class);
    final NamedOperationResolver resolver = new NamedOperationResolver(cache);
    final User user = mock(User.class);
    Map<String, Object> paramMap = Maps.newHashMap();
    // A parameter of the wrong type
    paramMap.put("param1", new ArrayList());
    ParameterDetail param = new ParameterDetail.Builder().defaultValue(1L).description("Limit param").valueClass(Long.class).build();
    Map<String, ParameterDetail> paramDetailMap = Maps.newHashMap();
    paramDetailMap.put("param1", param);
    // Make a real NamedOperationDetail with a parameter
    final NamedOperationDetail extendedNamedOperation = new NamedOperationDetail.Builder().operationName(opName).description("standard operation").operationChain("{ \"operations\": [ { \"class\":\"uk.gov.gchq.gaffer.operation.impl.get.GetAllElements\" }, { \"class\":\"uk.gov.gchq.gaffer.operation.impl.Limit\", \"resultLimit\": \"${param1}\" } ] }").parameters(paramDetailMap).build();
    given(cache.getNamedOperation(opName, user)).willReturn(extendedNamedOperation);
    // When
    assertThatIllegalArgumentException().isThrownBy(() -> resolver.preExecute(new OperationChain.Builder().first(new NamedOperation.Builder<>().name(opName).parameters(paramMap).build()).build(), new Context(user)));
}

Example 8 with NamedOperationCache

use of uk.gov.gchq.gaffer.store.operation.handler.named.cache.NamedOperationCache in project Gaffer by gchq.

the class NamedOperationResolverTest method shouldResolveNamedOperation.

@Test
public void shouldResolveNamedOperation() throws CacheOperationFailedException {
    // Given
    final String opName = "opName";
    final NamedOperationCache cache = mock(NamedOperationCache.class);
    final NamedOperationResolver resolver = new NamedOperationResolver(cache);
    final User user = mock(User.class);
    final NamedOperationDetail extendedNamedOperation = mock(NamedOperationDetail.class);
    final GetAdjacentIds op1 = mock(GetAdjacentIds.class);
    final GetElements op2 = mock(GetElements.class);
    final OperationChain namedOperationOpChain = new OperationChain(Arrays.asList(op1, op2));
    final Iterable<?> input = mock(CloseableIterable.class);
    final Map<String, Object> params = null;
    given(op1.getInput()).willReturn(null);
    given(cache.getNamedOperation(opName, user)).willReturn(extendedNamedOperation);
    given(extendedNamedOperation.getOperationChain(params)).willReturn(namedOperationOpChain);
    final OperationChain<Object> opChain = new OperationChain.Builder().first(new NamedOperation.Builder<>().name(opName).input(input).build()).build();
    // When
    resolver.preExecute(opChain, new Context(user));
    // Then
    assertEquals(namedOperationOpChain.getOperations(), opChain.getOperations());
    verify(op1).setInput((Iterable) input);
    verify(op2, never()).setInput((Iterable) input);
}

Example 9 with NamedOperationCache

use of uk.gov.gchq.gaffer.store.operation.handler.named.cache.NamedOperationCache in project Gaffer by gchq.

the class NamedOperationResolverTest method shouldNotExecuteNamedOperationWithWrongParameterName.

@Test
public void shouldNotExecuteNamedOperationWithWrongParameterName() throws OperationException, CacheOperationFailedException {
    // Given
    final String opName = "opName";
    final NamedOperationCache cache = mock(NamedOperationCache.class);
    final NamedOperationResolver resolver = new NamedOperationResolver(cache);
    final User user = mock(User.class);
    Map<String, Object> paramMap = Maps.newHashMap();
    // A parameter with the wrong name
    paramMap.put("param2", 1L);
    ParameterDetail param = new ParameterDetail.Builder().defaultValue(1L).description("Limit param").valueClass(Long.class).build();
    Map<String, ParameterDetail> paramDetailMap = Maps.newHashMap();
    paramDetailMap.put("param1", param);
    // Make a real NamedOperationDetail with a parameter
    final NamedOperationDetail extendedNamedOperation = new NamedOperationDetail.Builder().operationName(opName).description("standard operation").operationChain("{ \"operations\": [ { \"class\":\"uk.gov.gchq.gaffer.operation.impl.get.GetAllElements\" }, { \"class\":\"uk.gov.gchq.gaffer.operation.impl.Limit\", \"resultLimit\": \"${param1}\" } ] }").parameters(paramDetailMap).build();
    given(cache.getNamedOperation(opName, user)).willReturn(extendedNamedOperation);
    // When
    assertThatIllegalArgumentException().isThrownBy(() -> resolver.preExecute(new OperationChain.Builder().first(new NamedOperation.Builder<>().name(opName).parameters(paramMap).build()).build(), new Context(user)));
}

Example 10 with NamedOperationCache

use of uk.gov.gchq.gaffer.store.operation.handler.named.cache.NamedOperationCache in project Gaffer by gchq.

the class NamedOperationResolverTest method shouldResolveNestedNamedOperation.

@Test
public void shouldResolveNestedNamedOperation() throws OperationException, CacheOperationFailedException {
    // Given
    final String opName = "opName";
    final NamedOperationCache cache = mock(NamedOperationCache.class);
    final NamedOperationResolver resolver = new NamedOperationResolver(cache);
    final User user = mock(User.class);
    final NamedOperationDetail extendedNamedOperation = mock(NamedOperationDetail.class);
    final GetAdjacentIds op1 = mock(GetAdjacentIds.class);
    final GetElements op2 = mock(GetElements.class);
    final OperationChain namedOperationOpChain = new OperationChain(Arrays.asList(op1, op2));
    final Iterable<?> input = mock(CloseableIterable.class);
    final Map<String, Object> params = null;
    given(op1.getInput()).willReturn(null);
    given(cache.getNamedOperation(opName, user)).willReturn(extendedNamedOperation);
    given(extendedNamedOperation.getOperationChain(params)).willReturn(namedOperationOpChain);
    final OperationChain<Object> opChain = new OperationChain.Builder().first(new OperationChain.Builder().first(new NamedOperation.Builder<>().name(opName).input(input).build()).build()).build();
    // When
    resolver.preExecute(opChain, new Context(user));
    // Then
    assertEquals(1, opChain.getOperations().size());
    final OperationChain<?> nestedOpChain = (OperationChain<?>) opChain.getOperations().get(0);
    assertEquals(namedOperationOpChain.getOperations(), nestedOpChain.getOperations());
    verify(op1).setInput((Iterable) input);
    verify(op2, never()).setInput((Iterable) input);
}