src/test/java/org/apache/neethi/BasicTestCases.java

/**
 * Licensed to the Apache Software Foundation (ASF) under one
 * or more contributor license agreements. See the NOTICE file
 * distributed with this work for additional information
 * regarding copyright ownership. The ASF licenses this file
 * to you under the Apache License, Version 2.0 (the
 * "License"); you may not use this file except in compliance
 * with the License. You may obtain a copy of the License at
 *
 * http://www.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing,
 * software distributed under the License is distributed on an
 * "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY
 * KIND, either express or implied. See the License for the
 * specific language governing permissions and limitations
 * under the License.
 */

package org.apache.neethi;

import java.io.IOException;
import java.io.StringReader;
import java.io.StringWriter;

import javax.xml.namespace.QName;
import javax.xml.parsers.DocumentBuilder;
import javax.xml.parsers.DocumentBuilderFactory;
import javax.xml.parsers.ParserConfigurationException;
import javax.xml.stream.XMLOutputFactory;
import javax.xml.stream.XMLStreamWriter;

import org.w3c.dom.Element;
import org.w3c.dom.Node;

import org.xml.sax.InputSource;
import org.xml.sax.SAXException;

import org.apache.neethi.builders.PrimitiveAssertion;
import org.apache.neethi.builders.xml.XMLPrimitiveAssertionBuilder;

import org.junit.Test;

/**
 * 
 */
public class BasicTestCases extends PolicyTestCase {

    @Test
    public void testPrimitiveBuilder() throws Exception {
        String text = "<ns1:MaximumRetransmissionCount FooAtt=\"blah\" xmlns:ns1=\"http://foo\">10"
            + "</ns1:MaximumRetransmissionCount>";
        
        
        Assertion as = new XMLPrimitiveAssertionBuilder().build(getElementFromString(text),
                                                                policyEngine.getAssertionBuilderFactory());
        assertNotNull(as);
        PrimitiveAssertion pas = (PrimitiveAssertion)as;
        assertEquals("10", pas.getTextValue());
        assertEquals("blah", pas.getAttribute(new QName("FooAtt")));
        
        StringWriter writer = new StringWriter();
        XMLStreamWriter xwriter = XMLOutputFactory.newInstance().createXMLStreamWriter(writer);
        pas.serialize(xwriter);
        xwriter.flush();
        
        Element el = getElementFromString(writer.getBuffer().toString());
        assertEquals("blah", el.getAttribute("FooAtt"));
        assertEquals("10", el.getTextContent());
    }
    @Test
    public void testPolicyChildren() throws Exception {
        String text = "<sp:Wss11 xmlns:sp=\"http://schemas.xmlsoap.org/ws/2005/07/securitypolicy\">"
                      + "<wsp:Policy xmlns:wsp=\"http://schemas.xmlsoap.org/ws/2004/09/policy\">"
                      + "<sp:MustSupportRefKeyIdentifier/>"
                      + "<sp:MustSupportRefIssuerSerial/>"
                      + "<sp:MustSupportRefThumbprint/>"
                      + "<sp:MustSupportRefEncryptedKey/>"
                      + "</wsp:Policy></sp:Wss11>";
                                                                                        

        Assertion as = new XMLPrimitiveAssertionBuilder().build(getElementFromString(text),
                                                                policyEngine.getAssertionBuilderFactory());
        assertNotNull(as);
        
        
        StringWriter writer = new StringWriter();
        XMLStreamWriter xwriter = XMLOutputFactory.newInstance().createXMLStreamWriter(writer);
        as.serialize(xwriter);
        xwriter.flush();
        xwriter.close();
        
        Element el = getElementFromString(writer.getBuffer().toString());
        assertEquals("Wss11", el.getLocalName());
        el = (Element)el.getFirstChild();
        assertEquals("Policy", el.getLocalName());
        el = (Element)el.getFirstChild();
        assertEquals("MustSupportRefKeyIdentifier", el.getLocalName());
    }

    
    @Test
    public void testMultiChildren() throws Exception {
        String text = "<sp:SignedParts xmlns:sp=\"http://schemas.xmlsoap.org/ws/2005/07/securitypolicy\">"
            + "<sp:Body />"
            + "<sp:Header Name=\"To\" Namespace=\"http://schemas.xmlsoap.org/ws/2004/08/addressing\"/>"
            + "<sp:Header Name=\"From\" Namespace=\"http://schemas.xmlsoap.org/ws/2004/08/addressing\" />"
            + "</sp:SignedParts>";

        Assertion as = new XMLPrimitiveAssertionBuilder().build(getElementFromString(text),
                                                                policyEngine.getAssertionBuilderFactory());
        assertNotNull(as);
        
        
        StringWriter writer = new StringWriter();
        XMLStreamWriter xwriter = XMLOutputFactory.newInstance().createXMLStreamWriter(writer);
        as.serialize(xwriter);
        xwriter.flush();
        xwriter.close();
        
        Element el = getElementFromString(writer.getBuffer().toString());
        assertEquals("SignedParts", el.getLocalName());
        el = (Element)el.getFirstChild();
        assertEquals("Body", el.getLocalName());
        el = (Element)el.getNextSibling();
        assertEquals("Header", el.getLocalName());
    }

    
    private Element getElementFromString(String s) 
        throws ParserConfigurationException, SAXException, IOException {
        
        DocumentBuilderFactory dbf = DocumentBuilderFactory.newInstance();

        dbf.setValidating(false);
        dbf.setIgnoringComments(false);
        dbf.setIgnoringElementContentWhitespace(true);
        dbf.setNamespaceAware(true);

        DocumentBuilder db = null;
        db = dbf.newDocumentBuilder();

        
        return db.parse(new InputSource(new StringReader(s))).getDocumentElement();
    }
    
    @Test
    public void testInvalidPolicyElementName() throws Exception {
        for (int x = 0; x < 4; x++) {
            try {
                getPolicy("samples/test28.xml", x);
            } catch (IllegalArgumentException ex) {
                //expected
                assertTrue(ex.getMessage().contains("Policies"));
            }
        }
        Element d = getResourceAsDOM("samples/test28.xml");
        assertNotNull(d);
        Node nd = d.getFirstChild();
        Policy first = null;
        while (nd != null) {
            if (nd instanceof Element) {
                Policy p = policyEngine.getPolicy(nd);
                if (p.getId() != null) {
                    policyEngine.getPolicyRegistry().register(p.getId(), p);
                }
                if (first == null) {
                    first = p;
                }
            }
            nd = nd.getNextSibling();
        }
        first.normalize(true);
        
    }
}