View Javadoc
1   /**
2    * Licensed to the Apache Software Foundation (ASF) under one
3    * or more contributor license agreements. See the NOTICE file
4    * distributed with this work for additional information
5    * regarding copyright ownership. The ASF licenses this file
6    * to you under the Apache License, Version 2.0 (the
7    * "License"); you may not use this file except in compliance
8    * with the License. You may obtain a copy of the License at
9    *
10   * http://www.apache.org/licenses/LICENSE-2.0
11   *
12   * Unless required by applicable law or agreed to in writing,
13   * software distributed under the License is distributed on an
14   * "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY
15   * KIND, either express or implied. See the License for the
16   * specific language governing permissions and limitations
17   * under the License.
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.RecipientToken;
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 RecipientTokenTest extends AbstractTestBase {
35  
36      @Test
37      public void testRecipientToken() throws Exception {
38          String fileName = "RecipientToken.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 RecipientToken);
52              RecipientToken recipientToken = (RecipientToken) alternative.get(0);
53              assertFalse(recipientToken.isNormalized());
54              assertTrue(recipientToken.isIgnorable());
55              assertTrue(recipientToken.isOptional());
56              assertEquals(Constants.TYPE_ASSERTION, recipientToken.getType());
57              assertEquals(SP12Constants.RECIPIENT_TOKEN, recipientToken.getName());
58              assertNotNull(recipientToken.getToken());
59              assertEquals(SP12Constants.X509_TOKEN, recipientToken.getToken().getName());
60              count++;
61          }
62          assertEquals(1, count);
63  
64          policy = policy.normalize(true);
65          serializedPolicy = serializePolicy(policy);
66          assertXMLisEqual(serializedPolicy, normalizedPolicyReferenceFile);
67  
68          alternativeIterator = policy.getAlternatives();
69          List<Assertion> alternative = alternativeIterator.next();
70          assertEquals(0, alternative.size());
71  
72          List<PolicyComponent> policyComponents = policy.getPolicyComponents();
73          assertEquals(1, policyComponents.size());
74          PolicyOperator policyOperator = (PolicyOperator) policyComponents.get(0);
75          policyComponents = policyOperator.getPolicyComponents();
76          assertEquals(2, policyComponents.size());
77          All all = (All) policyComponents.get(0);
78          List<PolicyComponent> policyComponentsAll = all.getAssertions();
79          assertEquals(0, policyComponentsAll.size());
80  
81          all = (All) policyComponents.get(1);
82          policyComponentsAll = all.getAssertions();
83          assertEquals(1, policyComponentsAll.size());
84  
85          Iterator<PolicyComponent> policyComponentIterator = policyComponentsAll.iterator();
86          RecipientToken recipientToken = (RecipientToken) policyComponentIterator.next();
87          assertTrue(recipientToken.isNormalized());
88          assertTrue(recipientToken.isIgnorable());
89          assertFalse(recipientToken.isOptional());
90          assertEquals(Constants.TYPE_ASSERTION, recipientToken.getType());
91          assertEquals(SP12Constants.RECIPIENT_TOKEN, recipientToken.getName());
92          assertNotNull(recipientToken.getToken());
93          assertEquals(SP12Constants.X509_TOKEN, recipientToken.getToken().getName());
94      }
95  }