use of com.sun.identity.entitlement.EntitlementSubject in project OpenAM by OpenRock.
the class EntitlementRegistryTest method shouldReturnCorrectSubjectName.
@Test
public void shouldReturnCorrectSubjectName() {
// Given
String name = "test";
EntitlementSubject testSubject = new JwtClaimSubject();
testRegistry.registerSubjectType(EntitlementSubject.class);
testRegistry.registerSubjectType(name, JwtClaimSubject.class);
// When
String result = testRegistry.getSubjectName(testSubject);
// Then
assertThat(result).isEqualTo(name);
}
use of com.sun.identity.entitlement.EntitlementSubject in project OpenAM by OpenRock.
the class PolicyConditionUpgrader method isSubjectConditionUpgradable.
private boolean isSubjectConditionUpgradable(EntitlementSubject subject) {
if (subject == null) {
return true;
}
if (subject instanceof NoSubject) {
return true;
}
if (subject instanceof LogicalSubject) {
LogicalSubject logicalSubject = (LogicalSubject) subject;
boolean upgradable = true;
for (EntitlementSubject sub : logicalSubject.getESubjects()) {
upgradable &= isUpgradablePolicySubject(sub);
}
return upgradable;
}
return isUpgradablePolicySubject(subject);
}
use of com.sun.identity.entitlement.EntitlementSubject in project OpenAM by OpenRock.
the class PolicyConditionUpgraderTest method isPolicyWithAndSubjectConditionUpgradable.
@Test(dataProvider = "isPolicyWithAndSubjectConditionUpgradableDataProvider")
public void isPolicyWithAndSubjectConditionUpgradable(Class<? extends EntitlementSubject> sub1, boolean sub1InMap, Class<? extends EntitlementSubject> sub2, boolean sub2InMap, boolean expectedResult) {
//Given
Privilege policy = mock(Privilege.class);
AndSubject andSubject = mock(AndSubject.class);
Set<EntitlementSubject> andSubjects = new HashSet<EntitlementSubject>();
EntitlementSubject subject1 = mock(sub1);
EntitlementSubject subject2 = mock(sub2);
andSubjects.add(subject1);
andSubjects.add(subject2);
given(policy.getSubject()).willReturn(andSubject);
given(andSubject.getESubjects()).willReturn(andSubjects);
if (subject1 instanceof PolicySubject) {
given(((PolicySubject) subject1).getClassName()).willReturn("SUBJECT1_CLASS_NAME");
}
if (subject2 instanceof PolicySubject) {
given(((PolicySubject) subject2).getClassName()).willReturn("SUBJECT2_CLASS_NAME");
}
given(conditionUpgradeMap.containsSubjectCondition("SUBJECT1_CLASS_NAME")).willReturn(sub1InMap);
given(conditionUpgradeMap.containsSubjectCondition("SUBJECT2_CLASS_NAME")).willReturn(sub2InMap);
//When
boolean upgradable = conditionUpgrader.isPolicyUpgradable(policy);
//Then
assertThat(upgradable).isEqualTo(expectedResult);
}
use of com.sun.identity.entitlement.EntitlementSubject in project OpenAM by OpenRock.
the class PolicyConditionUpgraderTest method shouldMigratePolicyWithOrSubjectCondition.
@SuppressWarnings("unchecked")
@Test
public void shouldMigratePolicyWithOrSubjectCondition() throws EntitlementException, UpgradeException {
//Given
Privilege policy = mock(Privilege.class);
OrSubject orSubject = mock(OrSubject.class);
Set<EntitlementSubject> orSubjects = new HashSet<EntitlementSubject>();
PolicySubject subject1 = mock(PolicySubject.class);
PolicySubject subject2 = mock(PolicySubject.class);
orSubjects.add(subject1);
orSubjects.add(subject2);
EntitlementSubject migratedSubject1 = mock(EntitlementSubject.class);
EntitlementSubject migratedSubject2 = mock(EntitlementSubject.class);
given(policy.getSubject()).willReturn(orSubject);
given(orSubject.getESubjects()).willReturn(orSubjects);
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(orSubject).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.EntitlementSubject in project OpenAM by OpenRock.
the class PolicyConditionUpgraderTest method isPolicyWithSingleSubjectAndEnvironmentConditionUpgradable.
@Test(dataProvider = "isPolicyWithSingleSubjectAndEnvironmentConditionUpgradableDataProvider")
public void isPolicyWithSingleSubjectAndEnvironmentConditionUpgradable(Class<? extends EntitlementSubject> sub, boolean subInMap, Class<? extends EntitlementCondition> con, boolean conInMap, boolean expectedResult) {
//Given
Privilege policy = mock(Privilege.class);
EntitlementSubject subject = null;
if (sub != null) {
subject = mock(sub);
}
EntitlementCondition condition = null;
if (con != null) {
condition = mock(con);
}
given(policy.getSubject()).willReturn(subject);
given(policy.getCondition()).willReturn(condition);
if (subject instanceof PolicySubject) {
given(((PolicySubject) subject).getClassName()).willReturn("SUBJECT_CLASS_NAME");
}
if (condition instanceof PolicyCondition) {
given(((PolicyCondition) condition).getClassName()).willReturn("CONDITION_CLASS_NAME");
}
given(conditionUpgradeMap.containsSubjectCondition("SUBJECT_CLASS_NAME")).willReturn(subInMap);
given(conditionUpgradeMap.containsEnvironmentCondition("CONDITION_CLASS_NAME")).willReturn(conInMap);
//When
boolean upgradable = conditionUpgrader.isPolicyUpgradable(policy);
//Then
assertThat(upgradable).isEqualTo(expectedResult);
}
Aggregations