1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19 package org.apache.wss4j.policy.tests;
20
21 import org.apache.neethi.*;
22 import org.apache.wss4j.policy.SP12Constants;
23 import org.apache.wss4j.policy.model.SamlToken;
24 import org.junit.jupiter.api.Test;
25
26 import java.util.Iterator;
27 import java.util.List;
28
29 import static org.junit.jupiter.api.Assertions.assertEquals;
30 import static org.junit.jupiter.api.Assertions.assertFalse;
31 import static org.junit.jupiter.api.Assertions.assertNotNull;
32 import static org.junit.jupiter.api.Assertions.assertTrue;
33
34 public class SamlTokenTest extends AbstractTestBase {
35
36 @Test
37 public void testSamlToken() throws Exception {
38 String fileName = "SamlToken.xml";
39 String policyFile = loadPolicyFile("policy/model/sp12/" + fileName);
40 String serializedPolicyReferenceFile = loadPolicyFile("policy/model/sp12/serialized/" + fileName);
41 String normalizedPolicyReferenceFile = loadPolicyFile("policy/model/sp12/normalized/" + fileName);
42 Policy policy = loadPolicy(policyFile);
43 String serializedPolicy = serializePolicy(policy);
44 assertXMLisEqual(serializedPolicy, serializedPolicyReferenceFile);
45
46 Iterator<List<Assertion>> alternativeIterator = policy.getAlternatives();
47 int count = 0;
48 while (alternativeIterator.hasNext()) {
49 List<Assertion> alternative = alternativeIterator.next();
50 assertEquals(1, alternative.size());
51 assertTrue(alternative.get(0) instanceof SamlToken);
52 SamlToken samlToken = (SamlToken) alternative.get(0);
53 assertFalse(samlToken.isNormalized());
54 assertTrue(samlToken.isIgnorable());
55 assertTrue(samlToken.isOptional());
56 assertEquals(Constants.TYPE_ASSERTION, samlToken.getType());
57 assertEquals(SP12Constants.SAML_TOKEN, samlToken.getName());
58 assertEquals(SamlToken.SamlTokenType.WssSamlV20Token11, samlToken.getSamlTokenType());
59 assertTrue(samlToken.isRequireKeyIdentifierReference());
60 assertNotNull(samlToken.getClaims());
61 count++;
62 }
63 assertEquals(1, count);
64
65 policy = policy.normalize(true);
66 serializedPolicy = serializePolicy(policy);
67 assertXMLisEqual(serializedPolicy, normalizedPolicyReferenceFile);
68
69 alternativeIterator = policy.getAlternatives();
70 List<Assertion> alternative = alternativeIterator.next();
71 assertEquals(0, alternative.size());
72
73 List<PolicyComponent> policyComponents = policy.getPolicyComponents();
74 assertEquals(1, policyComponents.size());
75 PolicyOperator policyOperator = (PolicyOperator) policyComponents.get(0);
76 policyComponents = policyOperator.getPolicyComponents();
77 assertEquals(2, policyComponents.size());
78 All all = (All) policyComponents.get(0);
79 List<PolicyComponent> policyComponentsAll = all.getAssertions();
80 assertEquals(0, policyComponentsAll.size());
81
82 all = (All) policyComponents.get(1);
83 policyComponentsAll = all.getAssertions();
84 assertEquals(1, policyComponentsAll.size());
85
86 Iterator<PolicyComponent> policyComponentIterator = policyComponentsAll.iterator();
87 SamlToken samlToken = (SamlToken) policyComponentIterator.next();
88 assertTrue(samlToken.isNormalized());
89 assertTrue(samlToken.isIgnorable());
90 assertFalse(samlToken.isOptional());
91 assertEquals(Constants.TYPE_ASSERTION, samlToken.getType());
92 assertEquals(SP12Constants.SAML_TOKEN, samlToken.getName());
93 assertEquals(SamlToken.SamlTokenType.WssSamlV20Token11, samlToken.getSamlTokenType());
94 assertTrue(samlToken.isRequireKeyIdentifierReference());
95 assertNotNull(samlToken.getClaims());
96 }
97 }