public static String createCmdForRevokePrivilege( TAlterSentryRoleRevokePrivilegeRequest request) { return createCmdForGrantOrRevokePrivileges(request.getRoleName(), request.getPrivileges(), false); }
public Object getFieldValue(_Fields field) { switch (field) { case PROTOCOL_VERSION: return getProtocol_version(); case REQUESTOR_USER_NAME: return getRequestorUserName(); case ROLE_NAME: return getRoleName(); case PRIVILEGE: return getPrivilege(); case PRIVILEGES: return getPrivileges(); } throw new IllegalStateException(); }
Map<TSentryPrivilege, Update> privilegesUpdateMap = new HashMap<>(); for (SentryPolicyStorePlugin plugin : sentryPlugins) { plugin.onAlterSentryRoleRevokePrivilege(request.getRoleName(), request.getPrivileges(), privilegesUpdateMap); sentryStore.alterSentryRoleRevokePrivileges(request.getRoleName(), request.getPrivileges(), privilegesUpdateMap); } else { sentryStore.alterSentryRoleRevokePrivileges(request.getRoleName(), request.getPrivileges());
void verify(Status status) throws Exception { TAlterSentryRoleRevokePrivilegeRequest revokeRequest = new TAlterSentryRoleRevokePrivilegeRequest(); revokeRequest.setProtocol_version(ThriftConstants.TSENTRY_SERVICE_VERSION_CURRENT); revokeRequest.setRequestorUserName(REQUESTOR_USER); revokeRequest.setRoleName(roleName); revokeRequest.setPrivilege(privilege); TAlterSentryRoleRevokePrivilegeResponse response = processor.alter_sentry_role_revoke_privilege(revokeRequest); if (response.getStatus().getValue() == Status.OK.getCode()) { Mockito.verify(sentryStore).alterSentryRoleRevokePrivileges(revokeRequest.getRoleName(), revokeRequest.getPrivileges()); } else { Mockito.verify(sentryStore, Mockito.times(0)) .alterSentryRoleRevokePrivileges(Mockito.anyString(), Mockito.anySet()); } assertEquals("Revoke " + privilege.getAction() + " response is not valid", status.getCode(), response.getStatus().getValue()); Mockito.reset(sentryStore); } }