Examples with AdaptedPredicate uk.gov.gchq.koryphe.predicate.AdaptedPredicate used on opensource projects

Example 1 with AdaptedPredicate

use of uk.gov.gchq.koryphe.predicate.AdaptedPredicate in project Gaffer by gchq.

the class FederatedAddGraphHandlerTest method shouldAddGraphWithCustomReadAccessPredicate.

@Test
public void shouldAddGraphWithCustomReadAccessPredicate() throws Exception {
    store.initialise(FEDERATEDSTORE_GRAPH_ID, null, federatedStoreProperties);
    final Schema expectedSchema = new Schema.Builder().build();
    assertEquals(0, store.getGraphs(testUser, null, ignore).size());
    final AccessPredicate allowBlankUserAndTestUserReadAccess = new AccessPredicate(new AdaptedPredicate(new CallMethod("getUserId"), new Or<>(new IsEqual(testUser.getUserId()), new IsEqual(blankUser.getUserId()))));
    new FederatedAddGraphHandler().doOperation(new AddGraph.Builder().graphId(EXPECTED_GRAPH_ID).schema(expectedSchema).storeProperties(PROPERTIES).readAccessPredicate(allowBlankUserAndTestUserReadAccess).build(), new Context(testUser), store);
    assertEquals(1, store.getGraphs(blankUser, null, ignore).size());
    assertEquals(1, store.getGraphs(testUser, null, ignore).size());
}

Example 2 with AdaptedPredicate

use of uk.gov.gchq.koryphe.predicate.AdaptedPredicate in project Gaffer by gchq.

the class NamedViewDetailTest method shouldTestAccessUsingCustomAccessPredicatesWhenConfigured.

@Test
public void shouldTestAccessUsingCustomAccessPredicatesWhenConfigured() {
    // Given
    final User testUser = new User.Builder().userId("testUserId").build();
    final User differentUser = new User.Builder().userId("differentUserId").opAuth("different").build();
    final String adminAuth = "adminAuth";
    final AccessPredicate readAccessPredicate = new AccessPredicate(new AdaptedPredicate(new CallMethod("getUserId"), new IsEqual("testUserId")));
    final AccessPredicate writeAccessPredicate = new AccessPredicate(new AdaptedPredicate(new CallMethod("getOpAuths"), new CollectionContains("different")));
    // When
    final NamedViewDetail namedViewDetail = createNamedViewDetailBuilder().readAccessPredicate(readAccessPredicate).writeAccessPredicate(writeAccessPredicate).build();
    // Then
    assertTrue(namedViewDetail.hasReadAccess(testUser, adminAuth));
    assertFalse(namedViewDetail.hasReadAccess(differentUser, adminAuth));
    assertFalse(namedViewDetail.hasWriteAccess(testUser, adminAuth));
    assertTrue(namedViewDetail.hasWriteAccess(differentUser, adminAuth));
}

Example 3 with AdaptedPredicate

use of uk.gov.gchq.koryphe.predicate.AdaptedPredicate in project Gaffer by gchq.

the class FederatedAccessAuthTest method shouldSerialiseAndDeserialiseAccessPredicatesToJson.

@Test
public void shouldSerialiseAndDeserialiseAccessPredicatesToJson() throws SerialisationException {
    // Given
    final FederatedAccess federatedAccess = new FederatedAccess.Builder().addingUserId(AUTH_USER_ID).isPublic(false).readAccessPredicate(new AccessPredicate(new AdaptedPredicate(new CallMethod("getDataAuths"), new CollectionContains(ALL_USERS)))).writeAccessPredicate(new AccessPredicate(new AdaptedPredicate(new CallMethod("getDataAuths"), new CollectionContains(AUTH_1)))).build();
    // When
    final String serialised = new String(JSONSerialiser.serialise(federatedAccess));
    // Then
    final String expected = "{" + "   \"addingUserId\": \"authUser\"," + "   \"disabledByDefault\": false," + "   \"public\": false," + "   \"readAccessPredicate\": {" + "       \"class\": \"uk.gov.gchq.gaffer.access.predicate.AccessPredicate\"," + "       \"userPredicate\": {" + "           \"class\": \"uk.gov.gchq.koryphe.predicate.AdaptedPredicate\"," + "           \"inputAdapter\": {" + "               \"class\": \"uk.gov.gchq.koryphe.impl.function.CallMethod\"," + "               \"method\": \"getDataAuths\"" + "           }," + "           \"predicate\": {" + "               \"class\": \"uk.gov.gchq.koryphe.impl.predicate.CollectionContains\"," + "               \"value\": \"allUsers\"" + "           }" + "       }" + "   }," + "   \"writeAccessPredicate\": {" + "       \"class\": \"uk.gov.gchq.gaffer.access.predicate.AccessPredicate\"," + "       \"userPredicate\": {" + "           \"class\": \"uk.gov.gchq.koryphe.predicate.AdaptedPredicate\"," + "           \"inputAdapter\": {" + "               \"class\": \"uk.gov.gchq.koryphe.impl.function.CallMethod\"," + "               \"method\": \"getDataAuths\"" + "           }," + "           \"predicate\": {" + "               \"class\": \"uk.gov.gchq.koryphe.impl.predicate.CollectionContains\"," + "               \"value\": \"auth1\"" + "           }" + "       }" + "   }" + "}";
    JsonAssert.assertEquals(expected, serialised);
    final FederatedAccess deserialised = JSONSerialiser.deserialise(serialised, FederatedAccess.class);
    assertEquals(federatedAccess, deserialised);
}

Example 4 with AdaptedPredicate

use of uk.gov.gchq.koryphe.predicate.AdaptedPredicate in project Gaffer by gchq.

the class FederatedAddGraphWithHooksHandlerTest method shouldAddGraphWithCustomReadAccessPredicate.

@Test
public void shouldAddGraphWithCustomReadAccessPredicate() throws Exception {
    store.initialise(FEDERATEDSTORE_GRAPH_ID, null, federatedStoreProperties);
    final Schema expectedSchema = new Schema.Builder().build();
    assertEquals(0, store.getGraphs(testUser, null, ignore).size());
    final AccessPredicate allowBlankUserAndTestUserReadAccess = new AccessPredicate(new AdaptedPredicate(new CallMethod("getUserId"), new Or<>(new IsEqual(testUser.getUserId()), new IsEqual(blankUser.getUserId()))));
    new FederatedAddGraphWithHooksHandler().doOperation(new AddGraphWithHooks.Builder().graphId(EXPECTED_GRAPH_ID).schema(expectedSchema).storeProperties(PROPERTIES).readAccessPredicate(allowBlankUserAndTestUserReadAccess).build(), new Context(testUser), store);
    assertEquals(1, store.getGraphs(blankUser, null, ignore).size());
    assertEquals(1, store.getGraphs(testUser, null, ignore).size());
}