public void setFieldValue(_Fields field, Object value) { switch (field) { case GROUP_NAME: if (value == null) { unsetGroupName(); } else { setGroupName((String)value); } break; } }
private Set<TSentryGroup> getGroups() { Set<TSentryGroup> groups = new LinkedHashSet<TSentryGroup>(); TSentryGroup group = new TSentryGroup(); group.setGroupName(TEST_GROUP); groups.add(group); return groups; } }
@Test public void testAddDeleteGroups() throws Exception { String roleName = "test-groups"; String grantor = "g1"; createRole(roleName); Set<TSentryGroup> groups = Sets.newHashSet(); TSentryGroup group = new TSentryGroup(); group.setGroupName("test-groups-g1"); groups.add(group); group = new TSentryGroup(); group.setGroupName("test-groups-g2"); groups.add(group); sentryStore.alterSentryRoleAddGroups(grantor, roleName, groups); sentryStore.alterSentryRoleDeleteGroups(roleName, groups); MSentryRole role = sentryStore.getMSentryRoleByName(roleName); assertEquals(Collections.emptySet(), role.getGroups()); }
@Test public void testSentryGroupsSize() throws Exception { String role1 = "role1"; String role2 = "role2"; sentryStore.createSentryRole(role1); sentryStore.createSentryRole(role2); Set<TSentryGroup> groups = Sets.newHashSet(); TSentryGroup group = new TSentryGroup(); group.setGroupName("group1"); groups.add(group); String grantor = "g1"; sentryStore.alterSentryRoleAddGroups(grantor, role1, groups); assertEquals(Long.valueOf(1), sentryStore.getGroupCountGauge().getValue()); sentryStore.alterSentryRoleAddGroups(grantor, role2, groups); assertEquals(Long.valueOf(1), sentryStore.getGroupCountGauge().getValue()); groups.add(new TSentryGroup("group2")); sentryStore.alterSentryRoleAddGroups(grantor, role2, groups); assertEquals(Long.valueOf(2), sentryStore.getGroupCountGauge().getValue()); }
group.setGroupName(groupName); groups.add(group); sentryStore.alterSentryRoleAddGroups(grantor, roleName, groups);
@Test public void testCaseInsensitiveRole() throws Exception { String roleName = "newRole"; String grantor = "g1"; Set<TSentryGroup> groups = Sets.newHashSet(); TSentryGroup group = new TSentryGroup(); group.setGroupName("test-groups-g1"); groups.add(group); TSentryPrivilege privilege = new TSentryPrivilege(); privilege.setPrivilegeScope("TABLE"); privilege.setServerName("server1"); privilege.setDbName("default"); privilege.setTableName("table1"); privilege.setAction(AccessConstants.ALL); privilege.setCreateTime(System.currentTimeMillis()); Set<String> users = Sets.newHashSet("user1"); createRole(roleName); sentryStore.alterSentryRoleAddGroups(grantor, roleName, groups); sentryStore.alterSentryRoleDeleteGroups(roleName, groups); sentryStore.alterSentryRoleAddUsers(roleName, users); MSentryUser user = sentryStore.getMSentryUserByName(users.iterator().next()); assertNotNull(user); sentryStore.alterSentryRoleDeleteUsers(roleName, users); user = sentryStore.getMSentryUserByName(users.iterator().next(), false); assertNull(user); sentryStore.alterSentryGrantPrivileges(SentryPrincipalType.ROLE, roleName, Sets.newHashSet(privilege), null); sentryStore.alterSentryRevokePrivileges(SentryPrincipalType.ROLE, roleName, Sets.newHashSet(privilege), null); }
Set<String> users = Sets.newHashSet(); TSentryGroup group = new TSentryGroup(); group.setGroupName(groupName1); groups.add(group); users.add(userName1); users.clear(); group = new TSentryGroup(); group.setGroupName(groupName2); groups.add(group); users.add(userName2);
group.setGroupName(groupName1); groups.add(group); sentryStore.alterSentryRoleAddGroups(grantor, roleName1, groups);
group.setGroupName(groupName1); groups.add(group); sentryStore.alterSentryRoleAddGroups(grantor, roleName1, groups);
group.setGroupName("test-groups-g1"); groups.add(group); group = new TSentryGroup(); group.setGroupName("test-groups-g2"); groups.add(group);