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: package com.sun.xml.ws.policy.testutils;
038:
039: import com.sun.xml.stream.buffer.XMLStreamBuffer;
040: import com.sun.xml.stream.buffer.XMLStreamBufferException;
041: import com.sun.xml.ws.api.model.wsdl.WSDLModel;
042: import com.sun.xml.ws.api.server.SDDocumentSource;
043: import com.sun.xml.ws.api.wsdl.parser.WSDLParserExtension;
044: import com.sun.xml.ws.api.wsdl.parser.XMLEntityResolver.Parser;
045: import com.sun.xml.ws.policy.Policy;
046: import com.sun.xml.ws.policy.PolicyException;
047: import com.sun.xml.ws.policy.PolicyMap;
048: import com.sun.xml.ws.policy.jaxws.PolicyConfigResolver;
049: import com.sun.xml.ws.policy.jaxws.PolicyWSDLParserExtension;
050: import com.sun.xml.ws.policy.jaxws.WSDLPolicyMapWrapper;
051: import com.sun.xml.ws.policy.sourcemodel.PolicyModelTranslator;
052: import com.sun.xml.ws.policy.sourcemodel.PolicyModelUnmarshaller;
053: import com.sun.xml.ws.policy.sourcemodel.PolicySourceModel;
054: import java.io.IOException;
055: import java.io.InputStream;
056: import java.io.InputStreamReader;
057: import java.io.Reader;
058: import java.net.URL;
059: import javax.xml.stream.XMLInputFactory;
060: import javax.xml.stream.XMLStreamException;
061: import org.xml.sax.SAXException;
062:
063: /**
064: * This class provides utility methods to load resources and unmarshall policy source model.
065: *
066: * @author Marek Potociar
067: */
068: public final class PolicyResourceLoader {
069: public static final String POLICY_UNIT_TEST_RESOURCE_ROOT = "policy/";
070: private static final XMLInputFactory inputFactory = XMLInputFactory
071: .newInstance();
072:
073: public static final String[] SINGLE_ALTERNATIVE_POLICY = new String[] {
074: "single_alternative_policy/policy1.xml",
075: "single_alternative_policy/policy2.xml",
076: "single_alternative_policy/policy3.xml",
077: "single_alternative_policy/policy4.xml",
078: "single_alternative_policy/policy5.xml" };
079:
080: private PolicyResourceLoader() {
081: }
082:
083: public static PolicySourceModel unmarshallModel(String resource)
084: throws PolicyException, IOException {
085: Reader resourceReader = getResourceReader(resource);
086: PolicySourceModel model = PolicyModelUnmarshaller
087: .getXmlUnmarshaller().unmarshalModel(resourceReader);
088: resourceReader.close();
089: return model;
090: }
091:
092: public static PolicySourceModel unmarshallModel(
093: Reader resourceReader) throws PolicyException, IOException {
094: PolicySourceModel model = PolicyModelUnmarshaller
095: .getXmlUnmarshaller().unmarshalModel(resourceReader);
096: resourceReader.close();
097: return model;
098: }
099:
100: public static InputStream getResourceStream(String resourceName)
101: throws PolicyException {
102: String fullName = POLICY_UNIT_TEST_RESOURCE_ROOT + resourceName;
103: InputStream input = Thread.currentThread()
104: .getContextClassLoader().getResourceAsStream(fullName);
105: if (input == null) {
106: throw new PolicyException("Failed to find resource \""
107: + fullName + "\"");
108: }
109: return input;
110: }
111:
112: public static Reader getResourceReader(String resourceName)
113: throws PolicyException {
114: return new InputStreamReader(getResourceStream(resourceName));
115: }
116:
117: public static XMLStreamBuffer getResourceXmlBuffer(
118: String resourceName) throws PolicyException {
119: try {
120: return XMLStreamBuffer
121: .createNewBufferFromXMLStreamReader(inputFactory
122: .createXMLStreamReader(getResourceStream(resourceName)));
123: } catch (XMLStreamException ex) {
124: throw new PolicyException(
125: "Failed to create XMLStreamBuffer", ex);
126: }
127: }
128:
129: public static URL getResourceUrl(String resourceName) {
130: return Thread.currentThread().getContextClassLoader()
131: .getResource(
132: POLICY_UNIT_TEST_RESOURCE_ROOT + resourceName);
133: }
134:
135: public static Policy translateModel(PolicySourceModel model)
136: throws PolicyException {
137: return PolicyModelTranslator.getTranslator().translate(model);
138: }
139:
140: public static Policy loadPolicy(String resourceName)
141: throws PolicyException, IOException {
142: return translateModel(unmarshallModel(resourceName));
143: }
144:
145: // reads policy map from given wsdl document
146: public static PolicyMap getPolicyMap(String resourceName)
147: throws PolicyException {
148:
149: WSDLModel model = getWSDLModel(resourceName, true);
150: WSDLPolicyMapWrapper wrapper = model
151: .getExtension(WSDLPolicyMapWrapper.class);
152: return wrapper.getPolicyMap();
153: }
154:
155: public static PolicyMap getPolicyMap(String resourceName,
156: boolean isClient) throws PolicyException {
157:
158: WSDLModel model = getWSDLModel(resourceName, isClient);
159: WSDLPolicyMapWrapper wrapper = model
160: .getExtension(WSDLPolicyMapWrapper.class);
161: return wrapper.getPolicyMap();
162: }
163:
164: public static WSDLModel getWSDLModel(String resourceName)
165: throws PolicyException {
166: return getWSDLModel(resourceName, true);
167: }
168:
169: // reads wsdl model from given wsdl document
170: public static WSDLModel getWSDLModel(String resourceName,
171: boolean isClient) throws PolicyException {
172: URL resourceUrl = getResourceUrl(resourceName);
173: XMLStreamBuffer resourceBuffer = getResourceXmlBuffer(resourceName);
174: SDDocumentSource doc = SDDocumentSource.create(resourceUrl,
175: resourceBuffer);
176: try {
177: Parser parser = new Parser(doc);
178: WSDLModel model = WSDLModel.WSDLParser
179: .parse(
180: parser,
181: new PolicyConfigResolver(),
182: isClient,
183: new WSDLParserExtension[] { new PolicyWSDLParserExtension() });
184: return model;
185: } catch (XMLStreamException ex) {
186: throw new PolicyException("Failed to parse document", ex);
187: } catch (IOException ex) {
188: throw new PolicyException("Failed to parse document", ex);
189: } catch (SAXException ex) {
190: throw new PolicyException("Failed to parse document", ex);
191: }
192: }
193:
194: }
|