/*
    * ====================================================================
    * 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.
   * ====================================================================
   *
   * This software consists of voluntary contributions made by many
   * individuals on behalf of the Apache Software Foundation.  For more
   * information on the Apache Software Foundation, please see
   * <http://www.apache.org/>.
   *
   */
  
  package org.apache.http.nio.testserver;
  
  import java.net.URL;
  
  import javax.net.ssl.SSLContext;
  
  import org.apache.http.impl.nio.pool.BasicNIOConnFactory;
  import org.apache.http.nio.util.TestingSupport;
  import org.apache.http.ssl.SSLContextBuilder;
  import org.junit.After;
  
  /**
   * Base class for all HttpCore NIO integration tests
   *
   */
  public abstract class HttpCoreNIOTestBase {
  
      private static int JRE_LEVEL = TestingSupport.determineJRELevel();
  
      public enum ProtocolScheme { http, https }
  
      private final ProtocolScheme scheme;
  
      protected HttpServerNio server;
      protected HttpClientNio client;
  
      public HttpCoreNIOTestBase(final ProtocolScheme scheme) {
          this.scheme = scheme;
      }
  
      public HttpCoreNIOTestBase() {
          this(ProtocolScheme.http);
      }
  
      public ProtocolScheme getScheme() {
          return this.scheme;
      }
  
      protected SSLContext createServerSSLContext() throws Exception {
          if (JRE_LEVEL >= 8) {
              final URL keyStoreURL = getClass().getResource("/test-server.p12");
              final String storePassword = "nopassword";
              return SSLContextBuilder.create()
                      .setKeyStoreType("pkcs12")
                      .loadTrustMaterial(keyStoreURL, storePassword.toCharArray())
                      .loadKeyMaterial(keyStoreURL, storePassword.toCharArray(), storePassword.toCharArray())
                      .build();
          } else {
              final URL keyStoreURL = getClass().getResource("/test.keystore");
              final String storePassword = "nopassword";
              return SSLContextBuilder.create()
                      .loadTrustMaterial(keyStoreURL, storePassword.toCharArray())
                      .loadKeyMaterial(keyStoreURL, storePassword.toCharArray(), storePassword.toCharArray())
                      .build();
          }
      }
  
      protected SSLContext createClientSSLContext() throws Exception {
          if (JRE_LEVEL >= 8) {
              final URL keyStoreURL = getClass().getResource("/test-client.p12");
              final String storePassword = "nopassword";
              return SSLContextBuilder.create()
                      .setKeyStoreType("pkcs12")
                      .loadTrustMaterial(keyStoreURL, storePassword.toCharArray())
                      .build();
          } else {
              final URL keyStoreURL = getClass().getResource("/test.keystore");
              final String storePassword = "nopassword";
              return SSLContextBuilder.create()
                      .loadTrustMaterial(keyStoreURL, storePassword.toCharArray())
                      .build();
          }
     }
 
     protected ServerConnectionFactory createServerConnectionFactory() throws Exception {
         return new ServerConnectionFactory(
                 this.scheme.equals(ProtocolScheme.https) ? createServerSSLContext() : null);
     }
 
     protected BasicNIOConnFactory createClientConnectionFactory() throws Exception {
         return new BasicNIOConnFactory(
                 new ClientConnectionFactory(),
                 this.scheme.equals(ProtocolScheme.https) ? new ClientConnectionFactory(createClientSSLContext()) : null);
 
     }
 
     public void initServer() throws Exception {
         this.server = new HttpServerNio();
         this.server.setConnectionFactory(createServerConnectionFactory());
         this.server.setTimeout(5000);
     }
 
     public void initClient() throws Exception {
         this.client = new HttpClientNio(createClientConnectionFactory());
         this.client.setTimeout(5000);
     }
 
     @After
     public void shutDownClient() throws Exception {
         if (this.client != null) {
             this.client.shutdown();
             this.client = null;
         }
     }
 
     @After
     public void shutDownServer() throws Exception {
         if (this.server != null) {
             this.server.shutdown();
             this.server = null;
         }
     }
 
 }