/**
 * 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.activemq.transport.tcp;

import java.io.IOException;
import java.security.cert.X509Certificate;

import javax.management.remote.JMXPrincipal;
import javax.net.ssl.SSLSocket;

import junit.framework.TestCase;
import org.apache.activemq.command.ConnectionInfo;
import org.apache.activemq.transport.StubTransportListener;
import org.apache.activemq.wireformat.ObjectStreamWireFormat;
import org.junit.experimental.categories.Category;
import org.apache.activemq.test.annotations.ParallelTest;

/**
 * Unit tests for the SslTransport class.
 */
@Category(ParallelTest.class)
public class SslTransportTest extends TestCase {

    SSLSocket sslSocket;
    StubTransportListener stubListener;

    String username;
    String password;
    String certDistinguishedName;

    protected void setUp() throws Exception {
        certDistinguishedName = "ThisNameIsDistinguished";
        username = "SomeUserName";
        password = "SomePassword";
    }

    protected void tearDown() throws Exception {
        super.tearDown();
    }

    private void createTransportAndConsume(boolean wantAuth, boolean needAuth) throws IOException {
        JMXPrincipal principal = new JMXPrincipal(certDistinguishedName);
        X509Certificate cert = new StubX509Certificate(principal);
        StubSSLSession sslSession = new StubSSLSession(cert);

        sslSocket = new StubSSLSocket(sslSession);
        sslSocket.setWantClientAuth(wantAuth);
        sslSocket.setNeedClientAuth(needAuth);

        SslTransport transport = new SslTransport(new ObjectStreamWireFormat(), sslSocket);

        stubListener = new StubTransportListener();

        transport.setTransportListener(stubListener);

        ConnectionInfo sentInfo = new ConnectionInfo();

        sentInfo.setUserName(username);
        sentInfo.setPassword(password);

        transport.doConsume(sentInfo);
    }

    public void testKeepClientUserName() throws IOException {
        createTransportAndConsume(true, true);

        final ConnectionInfo receivedInfo = (ConnectionInfo)stubListener.getCommands().remove();

        X509Certificate receivedCert;

        try {
            receivedCert = ((X509Certificate[])receivedInfo.getTransportContext())[0];
        } catch (Exception e) {
            receivedCert = null;
        }

        if (receivedCert == null) {
            fail("Transmitted certificate chain was not attached to ConnectionInfo.");
        }

        assertEquals("Received certificate distinguished name did not match the one transmitted.", certDistinguishedName, receivedCert.getSubjectDN().getName());

    }
}
