Examples with NiFiUser org.apache.nifi.authorization.user.NiFiUser used on opensource projects

Example 56 with NiFiUser

use of org.apache.nifi.authorization.user.NiFiUser in project nifi by apache.

the class TestRemoteProcessGroupAuditor method setup.

@Before
public void setup() {
    final SecurityContext securityContext = SecurityContextHolder.getContext();
    final Authentication authentication = mock(Authentication.class);
    securityContext.setAuthentication(authentication);
    final NiFiUser user = new Builder().identity("user-id").build();
    final NiFiUserDetails userDetail = new NiFiUserDetails(user);
    when(authentication.getPrincipal()).thenReturn(userDetail);
}

Example 57 with NiFiUser

use of org.apache.nifi.authorization.user.NiFiUser in project nifi by apache.

the class DataAuthorizableTest method testCheckAuthorizationUserChain.

@Test
public void testCheckAuthorizationUserChain() {
    final NiFiUser proxy2 = new Builder().identity(PROXY_2).build();
    final NiFiUser proxy1 = new Builder().identity(PROXY_1).chain(proxy2).build();
    final NiFiUser user = new Builder().identity(IDENTITY_1).chain(proxy1).build();
    final AuthorizationResult result = testDataAuthorizable.checkAuthorization(testAuthorizer, RequestAction.READ, user, null);
    assertEquals(Result.Approved, result.getResult());
    verify(testAuthorizer, times(3)).authorize(any(AuthorizationRequest.class));
    verifyAuthorizeForUser(IDENTITY_1);
    verifyAuthorizeForUser(PROXY_1);
    verifyAuthorizeForUser(PROXY_2);
}

Example 58 with NiFiUser

use of org.apache.nifi.authorization.user.NiFiUser in project nifi by apache.

the class DataAuthorizableTest method testAuthorizeUnauthorizedUser.

@Test(expected = AccessDeniedException.class)
public void testAuthorizeUnauthorizedUser() {
    final NiFiUser user = new Builder().identity("unknown").build();
    testDataAuthorizable.authorize(testAuthorizer, RequestAction.READ, user, null);
}

Example 59 with NiFiUser

use of org.apache.nifi.authorization.user.NiFiUser in project nifi by apache.

the class DataAuthorizableTest method testCheckAuthorizationUnauthorizedUser.

@Test
public void testCheckAuthorizationUnauthorizedUser() {
    final NiFiUser user = new Builder().identity("unknown").build();
    final AuthorizationResult result = testDataAuthorizable.checkAuthorization(testAuthorizer, RequestAction.READ, user, null);
    assertEquals(Result.Denied, result.getResult());
}

Example 60 with NiFiUser

use of org.apache.nifi.authorization.user.NiFiUser in project nifi by apache.

the class DataAuthorizableTest method testAuthorizedUser.

@Test
public void testAuthorizedUser() {
    final NiFiUser user = new Builder().identity(IDENTITY_1).build();
    testDataAuthorizable.authorize(testAuthorizer, RequestAction.READ, user, null);
    verify(testAuthorizer, times(1)).authorize(argThat(new ArgumentMatcher<AuthorizationRequest>() {

        @Override
        public boolean matches(Object o) {
            return IDENTITY_1.equals(((AuthorizationRequest) o).getIdentity());
        }
    }));
}