private EdOrgContextRightsCache createEdOrgContextRights() {
EdOrgContextRightsCache edOrgContextRights = new EdOrgContextRightsCache();
Collection<GrantedAuthority> authorities2S =
new HashSet<GrantedAuthority>(
Arrays.asList(READ_GENERAL, READ_RESTRICTED, WRITE_GENERAL, STAFF_CONTEXT));
Collection<GrantedAuthority> authorities2T =
new HashSet<GrantedAuthority>(
Arrays.asList(READ_PUBLIC, READ_GENERAL, WRITE_GENERAL, TEACHER_CONTEXT));
Collection<GrantedAuthority> authorities4T =
new HashSet<GrantedAuthority>(Arrays.asList(READ_PUBLIC, WRITE_GENERAL, TEACHER_CONTEXT));
Collection<GrantedAuthority> authorities5S =
new HashSet<GrantedAuthority>(
Arrays.asList(
READ_PUBLIC, AGGREGATE_READ, WRITE_PUBLIC, AGGREGATE_WRITE, STAFF_CONTEXT));
Map<String, Collection<GrantedAuthority>> contextRights2 =
new HashMap<String, Collection<GrantedAuthority>>();
contextRights2.put(Right.STAFF_CONTEXT.name(), authorities2S);
contextRights2.put(Right.TEACHER_CONTEXT.name(), authorities2T);
Map<String, Collection<GrantedAuthority>> contextRights3 =
new HashMap<String, Collection<GrantedAuthority>>();
contextRights3.put(Right.STAFF_CONTEXT.name(), new HashSet<GrantedAuthority>());
contextRights3.put(Right.TEACHER_CONTEXT.name(), new HashSet<GrantedAuthority>());
Map<String, Collection<GrantedAuthority>> contextRights4 =
new HashMap<String, Collection<GrantedAuthority>>();
contextRights4.put(Right.STAFF_CONTEXT.name(), new HashSet<GrantedAuthority>());
contextRights4.put(Right.TEACHER_CONTEXT.name(), authorities4T);
Map<String, Collection<GrantedAuthority>> contextRights5 =
new HashMap<String, Collection<GrantedAuthority>>();
contextRights5.put(Right.STAFF_CONTEXT.name(), authorities5S);
contextRights5.put(Right.TEACHER_CONTEXT.name(), new HashSet<GrantedAuthority>());
edOrgContextRights.put("edOrg2", contextRights2);
edOrgContextRights.put("edOrg3", contextRights3);
edOrgContextRights.put("edOrg4", contextRights4);
edOrgContextRights.put("edOrg5", contextRights5);
return edOrgContextRights;
}
/**
* Test EntityEdOrgRightBuilder class.
*
* @author - tshewchuk
*/
@RunWith(SpringJUnit4ClassRunner.class)
@ContextConfiguration(locations = {"/spring/applicationContext-test.xml"})
@TestExecutionListeners({
WebContextTestExecutionListener.class,
DependencyInjectionTestExecutionListener.class,
DirtiesContextTestExecutionListener.class
})
public class EntityEdOrgRightBuilderTest {
@Autowired @InjectMocks private EntityEdOrgRightBuilder entityEdOrgRightBuilder;
private EdOrgOwnershipArbiter edOrgOwnershipArbiter;
private Map<String, Collection<GrantedAuthority>> edOrgRights;
private EdOrgContextRightsCache edOrgContextRights;
private final GrantedAuthority READ_PUBLIC = new GrantedAuthorityImpl("READ_PUBLIC");
private final GrantedAuthority READ_GENERAL = new GrantedAuthorityImpl("READ_GENERAL");
private final GrantedAuthority READ_RESTRICTED = new GrantedAuthorityImpl("READ_RESTRICTED");
private final GrantedAuthority AGGREGATE_READ = new GrantedAuthorityImpl("AGGREGATE_READ");
private final GrantedAuthority WRITE_PUBLIC = new GrantedAuthorityImpl("WRITE_PUBLIC");
private final GrantedAuthority WRITE_GENERAL = new GrantedAuthorityImpl("WRITE_GENERAL");
private final GrantedAuthority WRITE_RESTRICTED = new GrantedAuthorityImpl("WRITE_RESTRICTED");
private final GrantedAuthority AGGREGATE_WRITE = new GrantedAuthorityImpl("AGGREGATE_WRITE");
private final GrantedAuthority STAFF_CONTEXT =
new GrantedAuthorityImpl(Right.STAFF_CONTEXT.name());
private final GrantedAuthority TEACHER_CONTEXT =
new GrantedAuthorityImpl(Right.TEACHER_CONTEXT.name());
@Before
public void setup() {
edOrgOwnershipArbiter = Mockito.mock(EdOrgOwnershipArbiter.class);
entityEdOrgRightBuilder.setEdOrgOwnershipArbiter(edOrgOwnershipArbiter);
edOrgRights = createEdOrgRights();
edOrgContextRights = createEdOrgContextRights();
}
@SuppressWarnings("unchecked")
@Test
public void testBuildEntityEdOrgRights() {
Set<String> edOrgs = new HashSet<String>();
edOrgs.add("edOrg1");
edOrgs.add("edOrg2");
edOrgs.add("edOrg3");
edOrgs.add("edOrg4");
Entity entity = Mockito.mock(Entity.class);
Mockito.when(entity.getType()).thenReturn("student");
Mockito.when(
edOrgOwnershipArbiter.determineHierarchicalEdorgs(
Matchers.anyList(), Matchers.anyString()))
.thenReturn(edOrgs);
Collection<GrantedAuthority> grantedAuthorities =
entityEdOrgRightBuilder.buildEntityEdOrgRights(edOrgRights, entity, false);
Assert.assertEquals(5, grantedAuthorities.size());
Assert.assertTrue(grantedAuthorities.contains(READ_PUBLIC));
Assert.assertTrue(grantedAuthorities.contains(READ_GENERAL));
Assert.assertTrue(grantedAuthorities.contains(READ_RESTRICTED));
Assert.assertTrue(grantedAuthorities.contains(WRITE_GENERAL));
Assert.assertTrue(grantedAuthorities.contains(WRITE_RESTRICTED));
Assert.assertFalse(grantedAuthorities.contains(AGGREGATE_READ));
Assert.assertFalse(grantedAuthorities.contains(WRITE_PUBLIC));
Assert.assertFalse(grantedAuthorities.contains(AGGREGATE_WRITE));
}
@SuppressWarnings("unchecked")
@Test
public void testBuildEntityEdOrgRightsWithNoEntityEdOrgs() {
EdOrgOwnershipArbiter edOrgOwnershipArbiter = Mockito.mock(EdOrgOwnershipArbiter.class);
entityEdOrgRightBuilder.setEdOrgOwnershipArbiter(edOrgOwnershipArbiter);
Entity entity = Mockito.mock(Entity.class);
Mockito.when(entity.getType()).thenReturn("student");
Mockito.when(
edOrgOwnershipArbiter.determineHierarchicalEdorgs(
Matchers.anyList(), Matchers.anyString()))
.thenReturn(new HashSet<String>());
Collection<GrantedAuthority> grantedAuthorities =
entityEdOrgRightBuilder.buildEntityEdOrgRights(edOrgRights, entity, false);
Assert.assertTrue(grantedAuthorities.isEmpty());
}
@SuppressWarnings("unchecked")
@Test
public void testBuildEntityEdOrgRightsWithNoEntityEdOrgsNoMatchingEdOrgs() {
EdOrgOwnershipArbiter edOrgOwnershipArbiter = Mockito.mock(EdOrgOwnershipArbiter.class);
entityEdOrgRightBuilder.setEdOrgOwnershipArbiter(edOrgOwnershipArbiter);
Set<String> edOrgs = new HashSet<String>();
edOrgs.add("edOrg1");
edOrgs.add("edOrg6");
edOrgs.add("edOrg7");
Entity entity = Mockito.mock(Entity.class);
Mockito.when(entity.getType()).thenReturn("student");
Mockito.when(
edOrgOwnershipArbiter.determineHierarchicalEdorgs(
Matchers.anyList(), Matchers.anyString()))
.thenReturn(edOrgs);
Collection<GrantedAuthority> grantedAuthorities =
entityEdOrgRightBuilder.buildEntityEdOrgRights(edOrgRights, entity, false);
Assert.assertTrue(grantedAuthorities.isEmpty());
}
@SuppressWarnings("unchecked")
@Test
public void testBuildEntityEdOrgDualContextRights() {
Set<String> edOrgs = new HashSet<String>();
edOrgs.add("edOrg1");
edOrgs.add("edOrg2");
edOrgs.add("edOrg3");
edOrgs.add("edOrg4");
Entity entity = Mockito.mock(Entity.class);
Mockito.when(entity.getType()).thenReturn("student");
Mockito.when(
edOrgOwnershipArbiter.determineHierarchicalEdorgs(
Matchers.anyList(), Matchers.anyString()))
.thenReturn(edOrgs);
Collection<GrantedAuthority> grantedAuthorities =
entityEdOrgRightBuilder.buildEntityEdOrgContextRights(
edOrgContextRights, entity, SecurityUtil.UserContext.DUAL_CONTEXT, false);
Assert.assertEquals(6, grantedAuthorities.size());
Assert.assertTrue(grantedAuthorities.contains(READ_PUBLIC));
Assert.assertTrue(grantedAuthorities.contains(READ_GENERAL));
Assert.assertTrue(grantedAuthorities.contains(READ_RESTRICTED));
Assert.assertTrue(grantedAuthorities.contains(WRITE_GENERAL));
Assert.assertTrue(grantedAuthorities.contains(STAFF_CONTEXT));
Assert.assertTrue(grantedAuthorities.contains(TEACHER_CONTEXT));
Assert.assertFalse(grantedAuthorities.contains(WRITE_RESTRICTED));
Assert.assertFalse(grantedAuthorities.contains(AGGREGATE_READ));
Assert.assertFalse(grantedAuthorities.contains(WRITE_PUBLIC));
Assert.assertFalse(grantedAuthorities.contains(AGGREGATE_WRITE));
}
@SuppressWarnings("unchecked")
@Test
public void testBuildEntityEdOrgSingleContextRights() {
Set<String> edOrgs = new HashSet<String>();
edOrgs.add("edOrg1");
edOrgs.add("edOrg2");
edOrgs.add("edOrg3");
edOrgs.add("edOrg4");
Entity entity = Mockito.mock(Entity.class);
Mockito.when(entity.getType()).thenReturn("student");
Mockito.when(
edOrgOwnershipArbiter.determineHierarchicalEdorgs(
Matchers.anyList(), Matchers.anyString()))
.thenReturn(edOrgs);
Collection<GrantedAuthority> grantedAuthorities =
entityEdOrgRightBuilder.buildEntityEdOrgContextRights(
edOrgContextRights, entity, SecurityUtil.UserContext.TEACHER_CONTEXT, false);
Assert.assertEquals(4, grantedAuthorities.size());
Assert.assertTrue(grantedAuthorities.contains(READ_PUBLIC));
Assert.assertTrue(grantedAuthorities.contains(READ_GENERAL));
Assert.assertTrue(grantedAuthorities.contains(WRITE_GENERAL));
Assert.assertTrue(grantedAuthorities.contains(TEACHER_CONTEXT));
Assert.assertFalse(grantedAuthorities.contains(READ_RESTRICTED));
Assert.assertFalse(grantedAuthorities.contains(WRITE_RESTRICTED));
Assert.assertFalse(grantedAuthorities.contains(AGGREGATE_READ));
Assert.assertFalse(grantedAuthorities.contains(WRITE_PUBLIC));
Assert.assertFalse(grantedAuthorities.contains(AGGREGATE_WRITE));
Assert.assertFalse(grantedAuthorities.contains(STAFF_CONTEXT));
}
@SuppressWarnings("unchecked")
@Test
public void testBuildEntityEdOrgContextRightsWithNoEntityEdOrgs() {
EdOrgOwnershipArbiter edOrgOwnershipArbiter = Mockito.mock(EdOrgOwnershipArbiter.class);
entityEdOrgRightBuilder.setEdOrgOwnershipArbiter(edOrgOwnershipArbiter);
Entity entity = Mockito.mock(Entity.class);
Mockito.when(entity.getType()).thenReturn("student");
Mockito.when(
edOrgOwnershipArbiter.determineHierarchicalEdorgs(
Matchers.anyList(), Matchers.anyString()))
.thenReturn(new HashSet<String>());
Collection<GrantedAuthority> grantedAuthorities =
entityEdOrgRightBuilder.buildEntityEdOrgContextRights(
edOrgContextRights, entity, SecurityUtil.UserContext.DUAL_CONTEXT, false);
Assert.assertTrue(grantedAuthorities.isEmpty());
}
@SuppressWarnings("unchecked")
@Test
public void testBuildEntityEdOrgContextRightsWithNoEntityEdOrgsNoMatchingEdOrgs() {
EdOrgOwnershipArbiter edOrgOwnershipArbiter = Mockito.mock(EdOrgOwnershipArbiter.class);
entityEdOrgRightBuilder.setEdOrgOwnershipArbiter(edOrgOwnershipArbiter);
Set<String> edOrgs = new HashSet<String>();
edOrgs.add("edOrg1");
edOrgs.add("edOrg6");
edOrgs.add("edOrg7");
Entity entity = Mockito.mock(Entity.class);
Mockito.when(entity.getType()).thenReturn("student");
Mockito.when(
edOrgOwnershipArbiter.determineHierarchicalEdorgs(
Matchers.anyList(), Matchers.anyString()))
.thenReturn(edOrgs);
Collection<GrantedAuthority> grantedAuthorities =
entityEdOrgRightBuilder.buildEntityEdOrgContextRights(
edOrgContextRights, entity, SecurityUtil.UserContext.DUAL_CONTEXT, false);
Assert.assertTrue(grantedAuthorities.isEmpty());
}
private Map<String, Collection<GrantedAuthority>> createEdOrgRights() {
Map<String, Collection<GrantedAuthority>> edOrgRights =
new HashMap<String, Collection<GrantedAuthority>>();
Collection<GrantedAuthority> authorities2 =
new HashSet<GrantedAuthority>(Arrays.asList(READ_PUBLIC, READ_GENERAL, WRITE_GENERAL));
Collection<GrantedAuthority> authorities3 =
new HashSet<GrantedAuthority>(
Arrays.asList(READ_PUBLIC, READ_RESTRICTED, WRITE_RESTRICTED));
Collection<GrantedAuthority> authorities4 =
new HashSet<GrantedAuthority>(Arrays.asList(READ_PUBLIC, WRITE_GENERAL));
Collection<GrantedAuthority> authorities5 =
new HashSet<GrantedAuthority>(
Arrays.asList(READ_PUBLIC, AGGREGATE_READ, WRITE_PUBLIC, AGGREGATE_WRITE));
edOrgRights.put("edOrg2", authorities2);
edOrgRights.put("edOrg3", authorities3);
edOrgRights.put("edOrg4", authorities4);
edOrgRights.put("edOrg5", authorities5);
return edOrgRights;
}
private EdOrgContextRightsCache createEdOrgContextRights() {
EdOrgContextRightsCache edOrgContextRights = new EdOrgContextRightsCache();
Collection<GrantedAuthority> authorities2S =
new HashSet<GrantedAuthority>(
Arrays.asList(READ_GENERAL, READ_RESTRICTED, WRITE_GENERAL, STAFF_CONTEXT));
Collection<GrantedAuthority> authorities2T =
new HashSet<GrantedAuthority>(
Arrays.asList(READ_PUBLIC, READ_GENERAL, WRITE_GENERAL, TEACHER_CONTEXT));
Collection<GrantedAuthority> authorities4T =
new HashSet<GrantedAuthority>(Arrays.asList(READ_PUBLIC, WRITE_GENERAL, TEACHER_CONTEXT));
Collection<GrantedAuthority> authorities5S =
new HashSet<GrantedAuthority>(
Arrays.asList(
READ_PUBLIC, AGGREGATE_READ, WRITE_PUBLIC, AGGREGATE_WRITE, STAFF_CONTEXT));
Map<String, Collection<GrantedAuthority>> contextRights2 =
new HashMap<String, Collection<GrantedAuthority>>();
contextRights2.put(Right.STAFF_CONTEXT.name(), authorities2S);
contextRights2.put(Right.TEACHER_CONTEXT.name(), authorities2T);
Map<String, Collection<GrantedAuthority>> contextRights3 =
new HashMap<String, Collection<GrantedAuthority>>();
contextRights3.put(Right.STAFF_CONTEXT.name(), new HashSet<GrantedAuthority>());
contextRights3.put(Right.TEACHER_CONTEXT.name(), new HashSet<GrantedAuthority>());
Map<String, Collection<GrantedAuthority>> contextRights4 =
new HashMap<String, Collection<GrantedAuthority>>();
contextRights4.put(Right.STAFF_CONTEXT.name(), new HashSet<GrantedAuthority>());
contextRights4.put(Right.TEACHER_CONTEXT.name(), authorities4T);
Map<String, Collection<GrantedAuthority>> contextRights5 =
new HashMap<String, Collection<GrantedAuthority>>();
contextRights5.put(Right.STAFF_CONTEXT.name(), authorities5S);
contextRights5.put(Right.TEACHER_CONTEXT.name(), new HashSet<GrantedAuthority>());
edOrgContextRights.put("edOrg2", contextRights2);
edOrgContextRights.put("edOrg3", contextRights3);
edOrgContextRights.put("edOrg4", contextRights4);
edOrgContextRights.put("edOrg5", contextRights5);
return edOrgContextRights;
}
}