001: /*
002: * DO NOT ALTER OR REMOVE COPYRIGHT NOTICES OR THIS HEADER.
003: *
004: * Copyright 1997-2007 Sun Microsystems, Inc. All rights reserved.
005: *
006: * The contents of this file are subject to the terms of either the GNU
007: * General Public License Version 2 only ("GPL") or the Common Development
008: * and Distribution License("CDDL") (collectively, the "License"). You
009: * may not use this file except in compliance with the License. You can obtain
010: * a copy of the License at https://glassfish.dev.java.net/public/CDDL+GPL.html
011: * or glassfish/bootstrap/legal/LICENSE.txt. See the License for the specific
012: * language governing permissions and limitations under the License.
013: *
014: * When distributing the software, include this License Header Notice in each
015: * file and include the License file at glassfish/bootstrap/legal/LICENSE.txt.
016: * Sun designates this particular file as subject to the "Classpath" exception
017: * as provided by Sun in the GPL Version 2 section of the License file that
018: * accompanied this code. If applicable, add the following below the License
019: * Header, with the fields enclosed by brackets [] replaced by your own
020: * identifying information: "Portions Copyrighted [year]
021: * [name of copyright owner]"
022: *
023: * Contributor(s):
024: *
025: * If you wish your version of this file to be governed by only the CDDL or
026: * only the GPL Version 2, indicate your decision by adding "[Contributor]
027: * elects to include this software in this distribution under the [CDDL or GPL
028: * Version 2] license." If you don't indicate a single choice of license, a
029: * recipient has the option to distribute your version of this file under
030: * either the CDDL, the GPL Version 2 or to extend the choice of license to
031: * its licensees as provided above. However, if you add GPL Version 2 code
032: * and therefore, elected the GPL Version 2 license, then the option applies
033: * only if the new code is made subject to such option by the copyright
034: * holder.
035: */
036: /*
037: * SamlTokenTest.java
038: * JUnit based test
039: *
040: * Created on August 24, 2006, 12:27 AM
041: */
042:
043: package com.sun.xml.ws.security.impl.policy;
044:
045: import com.sun.xml.ws.policy.Policy;
046: import com.sun.xml.ws.policy.PolicyException;
047: import com.sun.xml.ws.policy.sourcemodel.PolicyModelTranslator;
048: import com.sun.xml.ws.policy.sourcemodel.PolicyModelUnmarshaller;
049: import com.sun.xml.ws.policy.sourcemodel.PolicySourceModel;
050: import java.io.IOException;
051: import java.io.InputStreamReader;
052: import java.io.Reader;
053:
054: import junit.framework.*;
055: import com.sun.xml.ws.policy.AssertionSet;
056: import com.sun.xml.ws.policy.NestedPolicy;
057: import com.sun.xml.ws.policy.Policy;
058: import com.sun.xml.ws.policy.PolicyAssertion;
059: import com.sun.xml.ws.security.policy.SecurityAssertionValidator;
060: import com.sun.xml.ws.policy.sourcemodel.AssertionData;
061: import java.util.ArrayList;
062: import java.util.Collection;
063: import java.util.Collections;
064: import java.util.Iterator;
065: import java.util.List;
066: import java.util.UUID;
067: import java.util.logging.Level;
068: import javax.xml.namespace.QName;
069: import static com.sun.xml.ws.security.impl.policy.Constants.*;
070:
071: /**
072: *
073: * @author Mayank.Mishra@SUN.com
074: */
075: public class SamlTokenTest extends TestCase {
076:
077: public SamlTokenTest(String testName) {
078: super (testName);
079: }
080:
081: protected void setUp() throws Exception {
082: }
083:
084: protected void tearDown() throws Exception {
085: }
086:
087: public static Test suite() {
088: TestSuite suite = new TestSuite(SamlTokenTest.class);
089:
090: return suite;
091: }
092:
093: private PolicySourceModel unmarshalPolicyResource(String resource)
094: throws PolicyException, IOException {
095: Reader reader = getResourceReader(resource);
096: PolicySourceModel model = PolicyModelUnmarshaller
097: .getXmlUnmarshaller().unmarshalModel(reader);
098: reader.close();
099: return model;
100: }
101:
102: private Reader getResourceReader(String resourceName) {
103: return new InputStreamReader(Thread.currentThread()
104: .getContextClassLoader().getResourceAsStream(
105: resourceName));
106: }
107:
108: public Policy unmarshalPolicy(String xmlFile) throws Exception {
109: PolicySourceModel model = unmarshalPolicyResource(xmlFile);
110: Policy mbp = PolicyModelTranslator.getTranslator().translate(
111: model);
112: return mbp;
113:
114: }
115:
116: public void testSamlToken_Types_5() throws Exception {
117: testSamlToken_Type(
118: "security/SamlTokenAssertions1.xml",
119: com.sun.xml.ws.security.impl.policy.SamlToken.WSS_SAML_V10_TOKEN10);
120: testSamlToken_Type(
121: "security/SamlTokenAssertions2.xml",
122: com.sun.xml.ws.security.impl.policy.SamlToken.WSS_SAML_V11_TOKEN10);
123: testSamlToken_Type(
124: "security/SamlTokenAssertions3.xml",
125: com.sun.xml.ws.security.impl.policy.SamlToken.WSS_SAML_V10_TOKEN11);
126: testSamlToken_Type(
127: "security/SamlTokenAssertions4.xml",
128: com.sun.xml.ws.security.impl.policy.SamlToken.WSS_SAML_V11_TOKEN11);
129: testSamlToken_Type(
130: "security/SamlTokenAssertions5.xml",
131: com.sun.xml.ws.security.impl.policy.SamlToken.WSS_SAML_V20_TOKEN11);
132: }
133:
134: public void testSamlToken_Keys() throws Exception {
135: String fileName = "security/SamlTokenAssertions1.xml";
136: Policy policy = unmarshalPolicy(fileName);
137: Iterator<AssertionSet> itr = policy.iterator();
138: if (itr.hasNext()) {
139: AssertionSet as = itr.next();
140: for (PolicyAssertion assertion : as) {
141: assertEquals("Invalid assertion", "SamlToken",
142: assertion.getName().getLocalPart());
143: SamlToken samlt = (SamlToken) assertion;
144: assertTrue(samlt.isRequireDerivedKeys());
145: }
146: } else {
147: throw new Exception(
148: "No Assertions found!. Unmarshalling of "
149: + fileName + " failed!");
150: }
151: }
152:
153: public void testSamlToken_Reference() throws Exception {
154: String fileName = "security/SamlTokenAssertions2.xml";
155: Policy policy = unmarshalPolicy(fileName);
156: Iterator<AssertionSet> itr = policy.iterator();
157: if (itr.hasNext()) {
158: AssertionSet as = itr.next();
159: for (PolicyAssertion assertion : as) {
160: assertEquals("Invalid assertion", "SamlToken",
161: assertion.getName().getLocalPart());
162: SamlToken samlt = (SamlToken) assertion;
163: Iterator itrst = samlt.getTokenRefernceType();
164: if (itrst.hasNext()) {
165: assertTrue(((String) itrst.next())
166: .equals(com.sun.xml.ws.security.impl.policy.SamlToken.REQUIRE_KEY_IDENTIFIER_REFERENCE));
167: }
168: }
169: } else {
170: throw new Exception(
171: "No Assertions found!. Unmarshalling of "
172: + fileName + " failed!");
173: }
174: }
175:
176: public void testSamlToken_Type(String fileName, String tokenType)
177: throws Exception {
178: Policy policy = unmarshalPolicy(fileName);
179: Iterator<AssertionSet> itr = policy.iterator();
180: if (itr.hasNext()) {
181: AssertionSet as = itr.next();
182: for (PolicyAssertion assertion : as) {
183: assertEquals("Invalid assertion", "SamlToken",
184: assertion.getName().getLocalPart());
185: SamlToken samlt = (SamlToken) assertion;
186: assertTrue(samlt.getTokenType().equals(tokenType));
187: }
188: } else {
189: throw new Exception(
190: "No Assertions found!. Unmarshalling of "
191: + fileName + " failed!");
192: }
193: }
194:
195: }
|