use of com.yahoo.athenz.zts.cache.DataCache in project athenz by yahoo.
the class DataCacheTest method testRoleNoMembers.
@Test
public void testRoleNoMembers() {
Role role = new Role();
role.setName("dom.role1");
DataCache cache = new DataCache();
cache.processRole(role);
Set<MemberRole> set1 = cache.getMemberRoleSet("user_domain.user1");
assertNull(set1);
}
use of com.yahoo.athenz.zts.cache.DataCache in project athenz by yahoo.
the class DataCacheTest method testPolicyWithNoAssertions.
@Test
public void testPolicyWithNoAssertions() {
Domain domain = new Domain();
domain.setName("testDomain");
Role role1 = new Role();
role1.setName("testDomain.role.role1");
List<RoleMember> members1 = new ArrayList<>();
members1.add(new RoleMember().setMemberName("user_domain.user1"));
members1.add(new RoleMember().setMemberName("user_domain.user2"));
role1.setRoleMembers(members1);
Role role2 = new Role();
role2.setName("testDomain.role.role2");
List<RoleMember> members2 = new ArrayList<>();
members2.add(new RoleMember().setMemberName("user_domain.user2"));
role2.setRoleMembers(members2);
Role role3 = new Role();
role3.setName("testDomain.role.role3");
List<RoleMember> members3 = new ArrayList<>();
members3.add(new RoleMember().setMemberName("user_domain.user3"));
role3.setRoleMembers(members3);
Policy policy = new Policy();
policy.setName("testDomain.policy.policy1");
HashMap<String, Role> roleList = new HashMap<>();
roleList.put(role1.getName(), role1);
roleList.put(role2.getName(), role2);
roleList.put(role3.getName(), role3);
DataCache cache = new DataCache();
cache.processRole(role1);
cache.processRole(role2);
cache.processRole(role3);
cache.processPolicy(domain.getName(), policy, roleList);
Set<MemberRole> set1 = cache.getMemberRoleSet("user_domain.user1");
assertNotNull(set1);
assertTrue(set1.contains(new MemberRole("testDomain.role.role1", 0)));
assertEquals(set1.size(), 1);
Set<MemberRole> set2 = cache.getMemberRoleSet("user_domain.user2");
assertNotNull(set2);
assertTrue(set2.contains(new MemberRole("testDomain.role.role1", 0)));
assertTrue(set2.contains(new MemberRole("testDomain.role.role2", 0)));
assertEquals(set2.size(), 2);
Set<MemberRole> set3 = cache.getMemberRoleSet("user_domain.user3");
assertNotNull(set3);
assertTrue(set3.contains(new MemberRole("testDomain.role.role3", 0)));
assertEquals(set3.size(), 1);
}
use of com.yahoo.athenz.zts.cache.DataCache in project athenz by yahoo.
the class DataCacheTest method testRolesWithTrust.
@Test
public void testRolesWithTrust() {
Role role1 = new Role();
role1.setName("dom.role1");
role1.setTrust("dom2");
Role role2 = new Role();
role2.setName("dom.role2");
role2.setTrust("dom3");
Role role3 = new Role();
role3.setName("dom.role3");
role3.setTrust("dom3");
DataCache cache = new DataCache();
cache.processRole(role1);
cache.processRole(role2);
cache.processRole(role3);
Map<String, Set<String>> map = cache.getTrustMap();
assertNotNull(map);
assertEquals(map.size(), 2);
assertTrue(map.containsKey("dom2"));
assertEquals(map.get("dom2").size(), 1);
assertTrue(map.get("dom2").contains("dom.role1"));
assertTrue(map.containsKey("dom3"));
assertEquals(map.get("dom3").size(), 2);
assertTrue(map.get("dom3").contains("dom.role2"));
assertTrue(map.get("dom3").contains("dom.role3"));
}
use of com.yahoo.athenz.zts.cache.DataCache in project athenz by yahoo.
the class DataCacheTest method testRoleWithTrust.
@Test
public void testRoleWithTrust() {
Role role1 = new Role();
role1.setName("dom.role1");
role1.setTrust("dom2");
DataCache cache = new DataCache();
cache.processRole(role1);
Map<String, Set<String>> map = cache.getTrustMap();
assertNotNull(map);
assertEquals(map.size(), 1);
assertTrue(map.containsKey("dom2"));
assertEquals(map.get("dom2").size(), 1);
assertTrue(map.get("dom2").contains("dom.role1"));
}
use of com.yahoo.athenz.zts.cache.DataCache in project athenz by yahoo.
the class DataCacheTest method testMultipleHostsSingleService.
@Test
public void testMultipleHostsSingleService() {
Domain domain = new Domain();
domain.setName("testDomain");
ServiceIdentity service = new ServiceIdentity();
service.setName("testDomain.storage");
List<String> hosts = new ArrayList<>();
hosts.add("host1");
hosts.add("host2");
service.setHosts(hosts);
DataCache cache = new DataCache();
cache.processServiceIdentity(service);
Map<String, Set<String>> hostMap = cache.getHostMap();
assertEquals(hostMap.size(), 2);
assertTrue(hostMap.containsKey("host1"));
Set<String> set = hostMap.get("host1");
assertEquals(set.size(), 1);
assertTrue(set.contains("testDomain.storage"));
assertTrue(hostMap.containsKey("host2"));
set = hostMap.get("host2");
assertEquals(set.size(), 1);
assertTrue(set.contains("testDomain.storage"));
}
Aggregations