use of com.sun.identity.entitlement.opensso.PolicyCondition in project OpenAM by OpenRock.
the class PolicyConditionUpgraderTest method shouldMigratePolicyWithOrEnvironmentCondition.
@SuppressWarnings("unchecked")
@Test
public void shouldMigratePolicyWithOrEnvironmentCondition() throws EntitlementException, UpgradeException {
//Given
Privilege policy = mock(Privilege.class);
OrCondition orCondition = mock(OrCondition.class);
Set<EntitlementCondition> orConditions = new HashSet<EntitlementCondition>();
PolicyCondition condition1 = mock(PolicyCondition.class);
PolicyCondition condition2 = mock(PolicyCondition.class);
orConditions.add(condition1);
orConditions.add(condition2);
EntitlementCondition migratedCondition1 = mock(EntitlementCondition.class);
EntitlementCondition migratedCondition2 = mock(EntitlementCondition.class);
given(policy.getCondition()).willReturn(orCondition);
given(orCondition.getEConditions()).willReturn(orConditions);
given(condition1.getClassName()).willReturn("CONDITION1_CLASS_NAME");
given(condition2.getClassName()).willReturn("CONDITION2_CLASS_NAME");
given(conditionUpgradeMap.migrateEnvironmentCondition(eq("CONDITION1_CLASS_NAME"), eq(condition1), Matchers.<MigrationReport>anyObject())).willReturn(migratedCondition1);
given(conditionUpgradeMap.migrateEnvironmentCondition(eq("CONDITION2_CLASS_NAME"), eq(condition2), Matchers.<MigrationReport>anyObject())).willReturn(migratedCondition2);
//When
conditionUpgrader.dryRunPolicyUpgrade(policy);
//Then
ArgumentCaptor<Set> conditionsCaptor = ArgumentCaptor.forClass(Set.class);
verify(orCondition).setEConditions(conditionsCaptor.capture());
assertThat(conditionsCaptor.getValue()).hasSize(2).contains(migratedCondition1, migratedCondition2);
verify(policy, never()).setSubject(Matchers.<EntitlementSubject>anyObject());
verify(policy, never()).setCondition(Matchers.<EntitlementCondition>anyObject());
}
use of com.sun.identity.entitlement.opensso.PolicyCondition 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.PolicyCondition in project OpenAM by OpenRock.
the class PolicyConditionUpgraderTest method shouldMigratePolicyWithAndEnvironmentCondition.
@SuppressWarnings("unchecked")
@Test
public void shouldMigratePolicyWithAndEnvironmentCondition() throws EntitlementException, UpgradeException {
//Given
Privilege policy = mock(Privilege.class);
AndCondition andCondition = mock(AndCondition.class);
Set<EntitlementCondition> andConditions = new HashSet<EntitlementCondition>();
PolicyCondition condition1 = mock(PolicyCondition.class);
PolicyCondition condition2 = mock(PolicyCondition.class);
andConditions.add(condition1);
andConditions.add(condition2);
EntitlementCondition migratedCondition1 = mock(EntitlementCondition.class);
EntitlementCondition migratedCondition2 = mock(EntitlementCondition.class);
given(policy.getCondition()).willReturn(andCondition);
given(andCondition.getEConditions()).willReturn(andConditions);
given(condition1.getClassName()).willReturn("CONDITION1_CLASS_NAME");
given(condition2.getClassName()).willReturn("CONDITION2_CLASS_NAME");
given(conditionUpgradeMap.migrateEnvironmentCondition(eq("CONDITION1_CLASS_NAME"), eq(condition1), Matchers.<MigrationReport>anyObject())).willReturn(migratedCondition1);
given(conditionUpgradeMap.migrateEnvironmentCondition(eq("CONDITION2_CLASS_NAME"), eq(condition2), Matchers.<MigrationReport>anyObject())).willReturn(migratedCondition2);
//When
conditionUpgrader.dryRunPolicyUpgrade(policy);
//Then
ArgumentCaptor<Set> conditionCaptor = ArgumentCaptor.forClass(Set.class);
verify(andCondition).setEConditions(conditionCaptor.capture());
assertThat(conditionCaptor.getValue()).hasSize(2).contains(migratedCondition1, migratedCondition2);
verify(policy, never()).setSubject(Matchers.<EntitlementSubject>anyObject());
verify(policy, never()).setCondition(Matchers.<EntitlementCondition>anyObject());
}
use of com.sun.identity.entitlement.opensso.PolicyCondition in project OpenAM by OpenRock.
the class PolicyConditionUpgraderTest method isPolicyWithNotEnvironmentConditionUpgradable.
@Test(dataProvider = "isPolicyWithNotEnvironmentConditionUpgradableDataProvider")
public void isPolicyWithNotEnvironmentConditionUpgradable(Class<? extends EntitlementCondition> condition, boolean conditionInMap, boolean expectedResult) {
//Given
Privilege policy = mock(Privilege.class);
NotCondition notCondition = mock(NotCondition.class);
Set<EntitlementCondition> notConditions = new HashSet<EntitlementCondition>();
EntitlementCondition con = mock(condition);
notConditions.add(con);
given(policy.getCondition()).willReturn(notCondition);
given(notCondition.getEConditions()).willReturn(notConditions);
if (con instanceof PolicyCondition) {
given(((PolicyCondition) con).getClassName()).willReturn("CONDITION_CLASS_NAME");
}
given(conditionUpgradeMap.containsEnvironmentCondition("CONDITION_CLASS_NAME")).willReturn(conditionInMap);
//When
boolean upgradable = conditionUpgrader.isPolicyUpgradable(policy);
//Then
assertThat(upgradable).isEqualTo(expectedResult);
}
use of com.sun.identity.entitlement.opensso.PolicyCondition in project OpenAM by OpenRock.
the class JsonPolicyParserTest method shouldPrintComplexConditions.
@Test
public void shouldPrintComplexConditions() throws Exception {
// Given
Privilege policy = new StubPrivilege();
AndCondition and = new AndCondition();
Set<EntitlementCondition> subConditions = new LinkedHashSet<EntitlementCondition>();
Map<String, Set<String>> props = new HashMap<String, Set<String>>();
props.put("AuthenticateToRealm", Collections.singleton("REALM"));
PolicyCondition policyCondition = new PolicyCondition("test", AuthenticateToRealmCondition.class.getName(), props);
NotCondition not = new NotCondition(policyCondition);
subConditions.add(not);
and.setEConditions(subConditions);
policy.setCondition(and);
// When
JsonValue result = parser.printPolicy(policy);
// Then
assertThat(result.get(new JsonPointer("condition/type")).asString()).isEqualTo("AND");
assertThat(result.get(new JsonPointer("condition/conditions/0/type")).asString()).isEqualTo("NOT");
assertThat(result.get(new JsonPointer("condition/conditions/0/condition/type")).asString()).isEqualTo("Policy");
assertThat(result.get(new JsonPointer("condition/conditions/0/condition/className")).asString()).isEqualTo(AuthenticateToRealmCondition.class.getName());
assertThat(result.get(new JsonPointer("condition/conditions/0/condition/properties")).asMapOfList(String.class)).includes(entry("AuthenticateToRealm", Arrays.asList("REALM")));
}
Aggregations