TestTAsyncSSLClientManagerCustomClient.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.thrift.async;

import java.io.IOException;
import org.apache.thrift.server.ServerTestBase;
import org.apache.thrift.transport.TNonblockingTransport;
import org.apache.thrift.transport.TSSLTransportFactory;
import org.apache.thrift.transport.TTransportException;

public class TestTAsyncSSLClientManagerCustomClient extends TestTAsyncSSLClientManager {

  private static final String trustStoreLocation = System.getProperty("javax.net.ssl.trustStore");
  private static final String trustStorePassword =
      System.getProperty("javax.net.ssl.trustStorePassword");

  @Override
  public TNonblockingTransport getClientTransport() throws TTransportException, IOException {
    TSSLTransportFactory.TSSLTransportParameters params =
        new TSSLTransportFactory.TSSLTransportParameters();
    params.setTrustStore(trustStoreLocation, trustStorePassword);
    TNonblockingTransport clientTransport =
        TSSLTransportFactory.getNonblockingClientSocket(
            ServerTestBase.HOST, ServerTestBase.PORT, 0, params);
    clientTransportList.add(clientTransport);
    return clientTransport;
  }
}