コード例 #1
0
  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;
  }
コード例 #2
0
/**
 * 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;
  }
}