use of com.sun.identity.entitlement.opensso.PolicySubject in project OpenAM by OpenRock.
the class PolicyConditionUpgrader method migrateSubjectConditions.
private void migrateSubjectConditions(Privilege privilege, MigrationReport migrationReport) throws UpgradeException, EntitlementException {
if (privilege.getSubject() == null) {
return;
}
if (privilege.getSubject() instanceof NoSubject) {
return;
}
if (privilege.getSubject() instanceof LogicalSubject) {
LogicalSubject logicalSubject = (LogicalSubject) privilege.getSubject();
Set<EntitlementSubject> subjects = logicalSubject.getESubjects();
Set<EntitlementSubject> migratedSubjects = new HashSet<EntitlementSubject>();
for (EntitlementSubject subject : subjects) {
if (subject instanceof NoSubject) {
//pass this through directly
migratedSubjects.add(subject);
} else if (!(subject instanceof PolicySubject)) {
//This should never happen due to check in initialise
throw new UpgradeException("Cannot upgrade a subject condition that is not of PolicySubject type!");
} else {
migratedSubjects.add(migrateSubjectCondition((PolicySubject) subject, migrationReport));
}
}
logicalSubject.setESubjects(migratedSubjects);
} else if (privilege.getSubject() instanceof PolicySubject) {
privilege.setSubject(migrateSubjectCondition((PolicySubject) privilege.getSubject(), migrationReport));
} else {
//This should never happen due to check in initialise
throw new UpgradeException("Cannot upgrade a subject condition that is not of PolicySubject type!");
}
}
use of com.sun.identity.entitlement.opensso.PolicySubject in project OpenAM by OpenRock.
the class PolicyConditionUpgraderTest method shouldMigratePolicyWithSingleSubjectAndEnvironmentCondition.
@Test
public void shouldMigratePolicyWithSingleSubjectAndEnvironmentCondition() throws EntitlementException, UpgradeException {
//Given
Privilege policy = mock(Privilege.class);
PolicySubject subject = mock(PolicySubject.class);
PolicyCondition condition = mock(PolicyCondition.class);
EntitlementSubject migratedSubject = mock(EntitlementSubject.class);
EntitlementCondition migratedCondition = mock(EntitlementCondition.class);
given(policy.getSubject()).willReturn(subject);
given(policy.getCondition()).willReturn(condition);
given(subject.getClassName()).willReturn("SUBJECT_CLASS_NAME");
given(condition.getClassName()).willReturn("CONDITION_CLASS_NAME");
given(conditionUpgradeMap.migrateSubjectCondition(eq("SUBJECT_CLASS_NAME"), eq(subject), Matchers.<MigrationReport>anyObject())).willReturn(migratedSubject);
given(conditionUpgradeMap.migrateEnvironmentCondition(eq("CONDITION_CLASS_NAME"), eq(condition), Matchers.<MigrationReport>anyObject())).willReturn(migratedCondition);
//When
conditionUpgrader.dryRunPolicyUpgrade(policy);
//Then
ArgumentCaptor<EntitlementSubject> subjectCaptor = ArgumentCaptor.forClass(EntitlementSubject.class);
verify(policy).setSubject(subjectCaptor.capture());
assertThat(subjectCaptor.getValue()).isEqualTo(migratedSubject);
ArgumentCaptor<EntitlementCondition> conditionCaptor = ArgumentCaptor.forClass(EntitlementCondition.class);
verify(policy).setCondition(conditionCaptor.capture());
assertThat(conditionCaptor.getValue()).isEqualTo(migratedCondition);
}
use of com.sun.identity.entitlement.opensso.PolicySubject in project OpenAM by OpenRock.
the class PolicyConditionUpgraderTest method shouldMigratePolicyWithAndSubjectCondition.
@SuppressWarnings("unchecked")
@Test
public void shouldMigratePolicyWithAndSubjectCondition() throws EntitlementException, UpgradeException {
//Given
Privilege policy = mock(Privilege.class);
AndSubject andSubject = mock(AndSubject.class);
Set<EntitlementSubject> andSubjects = new HashSet<EntitlementSubject>();
PolicySubject subject1 = mock(PolicySubject.class);
PolicySubject subject2 = mock(PolicySubject.class);
andSubjects.add(subject1);
andSubjects.add(subject2);
EntitlementSubject migratedSubject1 = mock(EntitlementSubject.class);
EntitlementSubject migratedSubject2 = mock(EntitlementSubject.class);
given(policy.getSubject()).willReturn(andSubject);
given(andSubject.getESubjects()).willReturn(andSubjects);
given(subject1.getClassName()).willReturn("SUBJECT1_CLASS_NAME");
given(subject2.getClassName()).willReturn("SUBJECT2_CLASS_NAME");
given(conditionUpgradeMap.migrateSubjectCondition(eq("SUBJECT1_CLASS_NAME"), eq(subject1), Matchers.<MigrationReport>anyObject())).willReturn(migratedSubject1);
given(conditionUpgradeMap.migrateSubjectCondition(eq("SUBJECT2_CLASS_NAME"), eq(subject2), Matchers.<MigrationReport>anyObject())).willReturn(migratedSubject2);
//When
conditionUpgrader.dryRunPolicyUpgrade(policy);
//Then
ArgumentCaptor<Set> subjectCaptor = ArgumentCaptor.forClass(Set.class);
verify(andSubject).setESubjects(subjectCaptor.capture());
assertThat(subjectCaptor.getValue()).hasSize(2).contains(migratedSubject1, migratedSubject2);
verify(policy, never()).setSubject(Matchers.<EntitlementSubject>anyObject());
verify(policy, never()).setCondition(Matchers.<EntitlementCondition>anyObject());
}
use of com.sun.identity.entitlement.opensso.PolicySubject in project OpenAM by OpenRock.
the class PolicyConditionUpgraderTest method shouldMigratePolicyWithNotSubjectCondition.
@Test
public void shouldMigratePolicyWithNotSubjectCondition() throws EntitlementException, UpgradeException {
//Given
Privilege policy = mock(Privilege.class);
NotSubject notSubject = mock(NotSubject.class);
Set<EntitlementSubject> notSubjects = new HashSet<EntitlementSubject>();
PolicySubject subject = mock(PolicySubject.class);
notSubjects.add(subject);
EntitlementSubject migratedSubject = mock(EntitlementSubject.class);
given(policy.getSubject()).willReturn(notSubject);
given(notSubject.getESubjects()).willReturn(notSubjects);
given(subject.getClassName()).willReturn("SUBJECT_CLASS_NAME");
given(conditionUpgradeMap.migrateSubjectCondition(eq("SUBJECT_CLASS_NAME"), eq(subject), Matchers.<MigrationReport>anyObject())).willReturn(migratedSubject);
//When
conditionUpgrader.dryRunPolicyUpgrade(policy);
//Then
ArgumentCaptor<Set> subjectCaptor = ArgumentCaptor.forClass(Set.class);
verify(notSubject).setESubjects(subjectCaptor.capture());
assertThat(subjectCaptor.getValue()).hasSize(1).contains(migratedSubject);
verify(policy, never()).setSubject(Matchers.<EntitlementSubject>anyObject());
verify(policy, never()).setCondition(Matchers.<EntitlementCondition>anyObject());
}
Aggregations