private TAlterSentryRoleGrantPrivilegeRequest getGrantPrivilegeRequest() { TAlterSentryRoleGrantPrivilegeRequest request = new TAlterSentryRoleGrantPrivilegeRequest(); request.setRoleName("testRole"); return request; }
private Set<TSentryPrivilege> grantPrivilegesCore(String requestorUserName, String roleName, Set<TSentryPrivilege> privileges) throws SentryUserException { TAlterSentryRoleGrantPrivilegeRequest request = new TAlterSentryRoleGrantPrivilegeRequest(); request.setProtocol_version(ThriftConstants.TSENTRY_SERVICE_VERSION_CURRENT); request.setRequestorUserName(requestorUserName); request.setRoleName(roleName); request.setPrivileges(privileges); try { TAlterSentryRoleGrantPrivilegeResponse response = client.alter_sentry_role_grant_privilege(request); Status.throwIfNotOk(response.getStatus()); return response.getPrivileges(); } catch (TException e) { throw new SentryUserException(THRIFT_EXCEPTION_MESSAGE, e); } }
unsetRoleName(); } else { setRoleName((String)value);
TAlterSentryRoleGrantPrivilegeRequest request = new TAlterSentryRoleGrantPrivilegeRequest(); request.setRequestorUserName(TEST_USER_NAME); request.setRoleName(TEST_ROLE_NAME);
void verify(Status status) throws Exception { TAlterSentryRoleGrantPrivilegeRequest grantRequest = new TAlterSentryRoleGrantPrivilegeRequest(); grantRequest.setProtocol_version(ThriftConstants.TSENTRY_SERVICE_VERSION_CURRENT); grantRequest.setRequestorUserName(REQUESTOR_USER); grantRequest.setRoleName(roleName); grantRequest.setPrivilege(privilege); TAlterSentryRoleGrantPrivilegeResponse response = processor.alter_sentry_role_grant_privilege(grantRequest); if (response.getStatus().getValue() == Status.OK.getCode()) { Mockito.verify(sentryStore).alterSentryRoleGrantPrivileges(grantRequest.getRoleName(), grantRequest.getPrivileges()); } else { Mockito.verify(sentryStore, Mockito.times(0)) .alterSentryRoleGrantPrivileges(Mockito.anyString(), Mockito.anySet()); } assertEquals("Grant " + privilege.getAction() + " response is not valid", status.getCode(), response.getStatus().getValue()); Mockito.reset(sentryStore); } }