unit.test.be.e_contract.mycarenet.ehealth.common.WSSecuritySOAPHandlerTest.java Source code

Introduction

Here is the source code for unit.test.be.e_contract.mycarenet.ehealth.common.WSSecuritySOAPHandlerTest.java

Source

/*
 * Java MyCareNet Project.
 * Copyright (C) 2013 e-Contract.be BVBA.
 *
 * This is free software; you can redistribute it and/or modify it
 * under the terms of the GNU Lesser General Public License version
 * 3.0 as published by the Free Software Foundation.
 *
 * This software is distributed in the hope that it will be useful,
 * but WITHOUT ANY WARRANTY; without even the implied warranty of
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
 * Lesser General Public License for more details.
 *
 * You should have received a copy of the GNU Lesser General Public
 * License along with this software; if not, see 
 * http://www.gnu.org/licenses/.
 */

package unit.test.be.e_contract.mycarenet.ehealth.common;

import java.io.StringWriter;
import java.security.KeyPair;
import java.security.KeyPairGenerator;
import java.security.PrivateKey;

import javax.xml.namespace.QName;
import javax.xml.soap.MessageFactory;
import javax.xml.soap.SOAPBody;
import javax.xml.soap.SOAPEnvelope;
import javax.xml.soap.SOAPMessage;
import javax.xml.soap.SOAPPart;
import javax.xml.transform.Transformer;
import javax.xml.transform.TransformerFactory;
import javax.xml.transform.dom.DOMSource;
import javax.xml.transform.stream.StreamResult;
import javax.xml.ws.handler.MessageContext;
import javax.xml.ws.handler.soap.SOAPMessageContext;

import org.apache.commons.logging.Log;
import org.apache.commons.logging.LogFactory;
import org.easymock.EasyMock;
import org.junit.Test;
import org.w3c.dom.Node;

import be.e_contract.mycarenet.ehealth.common.WSSecuritySOAPHandler;

public class WSSecuritySOAPHandlerTest {

    private static final Log LOG = LogFactory.getLog(WSSecuritySOAPHandlerTest.class);

    @Test
    public void testHandleMessage() throws Exception {
        KeyPairGenerator keyPairGenerator = KeyPairGenerator.getInstance("RSA");
        KeyPair keyPair = keyPairGenerator.generateKeyPair();
        PrivateKey privateKey = keyPair.getPrivate();

        String samlAssertion = "<Assertion xmlns=\"urn:oasis:names:tc:SAML:1.0:assertion\""
                + " AssertionID=\"_42e7a00652420d86ee884f295a3fbf02\">" + "</Assertion>";

        WSSecuritySOAPHandler testedInstance = new WSSecuritySOAPHandler();
        testedInstance.setPrivateKey(privateKey);
        testedInstance.setAssertion(samlAssertion);

        SOAPMessageContext mockSoapMessageContext = EasyMock.createMock(SOAPMessageContext.class);
        EasyMock.expect(mockSoapMessageContext.get(MessageContext.MESSAGE_OUTBOUND_PROPERTY)).andReturn(true);

        MessageFactory messageFactory = MessageFactory.newInstance();
        SOAPMessage soapMessage = messageFactory.createMessage();
        SOAPPart soapPart = soapMessage.getSOAPPart();
        SOAPEnvelope soapEnvelope = soapPart.getEnvelope();
        SOAPBody soapBody = soapEnvelope.getBody();
        soapBody.addBodyElement(new QName("http://www.example.com", "Test"));

        EasyMock.expect(mockSoapMessageContext.getMessage()).andReturn(soapMessage);

        // prepare
        EasyMock.replay(mockSoapMessageContext);

        // operate
        testedInstance.handleMessage(mockSoapMessageContext);

        // verify
        EasyMock.verify(mockSoapMessageContext);
        LOG.debug(toString(soapPart));
    }

    private String toString(Node node) throws Exception {
        TransformerFactory transformerFactory = TransformerFactory.newInstance();
        Transformer transformer = transformerFactory.newTransformer();
        StringWriter stringWriter = new StringWriter();
        transformer.transform(new DOMSource(node), new StreamResult(stringWriter));
        return stringWriter.toString();
    }
}