Search in sources :

Example 16 with Filter

use of uk.gov.gchq.gaffer.operation.impl.function.Filter in project Gaffer by gchq.

the class FederatedFilterHandlerTest method shouldDelegateToHandler.

@Test
public void shouldDelegateToHandler() throws OperationException {
    // Given
    final FederatedStore store = mock(FederatedStore.class);
    final FilterHandler handler = mock(FilterHandler.class);
    final Filter op = mock(Filter.class);
    final Context context = mock(Context.class);
    final Iterable expectedResult = mock(Iterable.class);
    final Schema schema = mock(Schema.class);
    given(store.getSchema(op, context)).willReturn(schema);
    given(handler.doOperation(op, schema)).willReturn(expectedResult);
    final FederatedFilterHandler federatedHandler = new FederatedFilterHandler(handler);
    // When
    final Object result = federatedHandler.doOperation(op, context, store);
    // Then
    assertSame(expectedResult, result);
    verify(handler).doOperation(op, schema);
}
Also used : Context(uk.gov.gchq.gaffer.store.Context) Filter(uk.gov.gchq.gaffer.operation.impl.function.Filter) Schema(uk.gov.gchq.gaffer.store.schema.Schema) FederatedFilterHandler(uk.gov.gchq.gaffer.federatedstore.operation.handler.FederatedFilterHandler) FederatedFilterHandler(uk.gov.gchq.gaffer.federatedstore.operation.handler.FederatedFilterHandler) FilterHandler(uk.gov.gchq.gaffer.store.operation.handler.function.FilterHandler) FederatedStore(uk.gov.gchq.gaffer.federatedstore.FederatedStore) Test(org.junit.jupiter.api.Test)

Aggregations

Filter (uk.gov.gchq.gaffer.operation.impl.function.Filter)16 Test (org.junit.jupiter.api.Test)14 ElementFilter (uk.gov.gchq.gaffer.data.element.function.ElementFilter)14 Edge (uk.gov.gchq.gaffer.data.element.Edge)12 Element (uk.gov.gchq.gaffer.data.element.Element)8 Schema (uk.gov.gchq.gaffer.store.schema.Schema)6 IsMoreThan (uk.gov.gchq.koryphe.impl.predicate.IsMoreThan)6 Entity (uk.gov.gchq.gaffer.data.element.Entity)4 ArrayList (java.util.ArrayList)2 FederatedFilterHandler (uk.gov.gchq.gaffer.federatedstore.operation.handler.FederatedFilterHandler)2 Context (uk.gov.gchq.gaffer.store.Context)2 TypeDefinition (uk.gov.gchq.gaffer.store.schema.TypeDefinition)2 Strings (com.google.common.base.Strings)1 Sets (com.google.common.collect.Sets)1 Collection (java.util.Collection)1 Collections (java.util.Collections)1 HashMap (java.util.HashMap)1 List (java.util.List)1 Map (java.util.Map)1 Objects.isNull (java.util.Objects.isNull)1