AbstractHttpClientWagonTest

package org.apache.maven.wagon.providers.http;

/*
 * 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.
 */

import static org.hamcrest.CoreMatchers.instanceOf;
import static org.junit.Assert.assertEquals;
import static org.junit.Assert.assertFalse;
import static org.junit.Assert.assertNotNull;
import static org.junit.Assert.assertThat;
import static org.junit.Assert.assertTrue;
import static org.junit.Assert.fail;

import java.io.File;
import java.io.IOException;
import java.io.InterruptedIOException;
import java.lang.reflect.Field;
import java.net.ConnectException;
import java.net.MalformedURLException;
import java.net.URL;
import java.net.URLClassLoader;
import java.net.UnknownHostException;
import java.util.ArrayList;
import java.util.Collection;
import java.util.Set;

import javax.net.ssl.SSLException;

import org.apache.http.client.HttpRequestRetryHandler;
import org.apache.http.impl.client.CloseableHttpClient;
import org.apache.http.impl.client.DefaultHttpRequestRetryHandler;
import org.apache.http.impl.execchain.RedirectExec;
import org.apache.http.impl.execchain.RetryExec;
import org.apache.maven.wagon.InputData;
import org.apache.maven.wagon.repository.Repository;
import org.apache.maven.wagon.resource.Resource;
import org.apache.maven.wagon.shared.http.AbstractHttpClientWagon;
import org.junit.Test;

public class AbstractHttpClientWagonTest
{
    @Test
    public void test()
        throws Exception
    {
        AbstractHttpClientWagon wagon = new AbstractHttpClientWagon()
        {
        };

        Repository repository = new Repository( "central", "http://repo.maven.apache.org/maven2" );

        wagon.connect( repository );

        Resource resource = new Resource();

        resource.setName( "junit/junit/maven-metadata.xml" );

        InputData inputData = new InputData();

        inputData.setResource( resource );

        wagon.fillInputData( inputData );

        wagon.disconnect();
    }

    @Test
    public void retryableConfigurationDefaultTest() throws Exception
    {
        doTestHttpClient( new Runnable()
        {
            @Override
            public void run()
            {
                final HttpRequestRetryHandler handler = getCurrentHandler();
                assertNotNull( handler );
                assertThat( handler, instanceOf( DefaultHttpRequestRetryHandler.class ) );
                final DefaultHttpRequestRetryHandler impl = DefaultHttpRequestRetryHandler.class.cast(handler);
                assertEquals( 3, impl.getRetryCount() );
                assertFalse( impl.isRequestSentRetryEnabled() );
            }
        });
    }

    @Test
    public void retryableConfigurationCountTest() throws Exception
    {
        doTestHttpClient( new Runnable()
        {
            @Override
            public void run()
            {
                System.setProperty( "maven.wagon.http.retryHandler.class", "default" );
                System.setProperty( "maven.wagon.http.retryHandler.count", "5" );

                final HttpRequestRetryHandler handler = getCurrentHandler();
                assertNotNull( handler );
                assertThat( handler, instanceOf( DefaultHttpRequestRetryHandler.class ) );
                final DefaultHttpRequestRetryHandler impl = DefaultHttpRequestRetryHandler.class.cast(handler);
                assertEquals( 5, impl.getRetryCount() );
                assertFalse( impl.isRequestSentRetryEnabled() );
            }
        });
    }

    @Test
    public void retryableConfigurationSentTest() throws Exception
    {
        doTestHttpClient( new Runnable()
        {
            @Override
            public void run()
            {
                System.setProperty( "maven.wagon.http.retryHandler.class", "default" );
                System.setProperty( "maven.wagon.http.retryHandler.requestSentEnabled", "true" );

                final HttpRequestRetryHandler handler = getCurrentHandler();
                assertNotNull( handler );
                assertThat( handler, instanceOf( DefaultHttpRequestRetryHandler.class ) );
                final DefaultHttpRequestRetryHandler impl = DefaultHttpRequestRetryHandler.class.cast(handler);
                assertEquals( 3, impl.getRetryCount() );
                assertTrue( impl.isRequestSentRetryEnabled() );
            }
        });
    }

    @Test
    public void retryableConfigurationExceptionsTest() throws Exception
    {
        doTestHttpClient( new Runnable()
        {
            @Override
            public void run()
            {
                System.setProperty( "maven.wagon.http.retryHandler.class", "default" );
                System.setProperty( "maven.wagon.http.retryHandler.nonRetryableClasses", IOException.class.getName() );

                final HttpRequestRetryHandler handler = getCurrentHandler();
                assertNotNull( handler );
                assertThat( handler, instanceOf( DefaultHttpRequestRetryHandler.class ) );
                final DefaultHttpRequestRetryHandler impl = DefaultHttpRequestRetryHandler.class.cast(handler);
                assertEquals( 3, impl.getRetryCount() );
                assertFalse( impl.isRequestSentRetryEnabled() );

                try
                {
                    final Field nonRetriableClasses = handler.getClass().getSuperclass()
                            .getDeclaredField( "nonRetriableClasses" );
                    if ( !nonRetriableClasses.isAccessible() )
                    {
                        nonRetriableClasses.setAccessible(true);
                    }
                    final Set<?> exceptions = Set.class.cast( nonRetriableClasses.get(handler) );
                    assertEquals( 1, exceptions.size() );
                    assertTrue( exceptions.contains( IOException.class ) );
                }
                catch ( final Exception e )
                {
                    fail( e.getMessage() );
                }
            }
        });
    }

    private HttpRequestRetryHandler getCurrentHandler()
    {
        try
        {
            final Class<?> impl = Thread.currentThread().getContextClassLoader().loadClass(
                        "org.apache.maven.wagon.shared.http.AbstractHttpClientWagon" );

            final CloseableHttpClient httpClient = CloseableHttpClient.class.cast(
                    impl.getMethod("getHttpClient").invoke(null) );

            final Field redirectExec = httpClient.getClass().getDeclaredField( "execChain" );
            if ( !redirectExec.isAccessible() )
            {
                redirectExec.setAccessible( true );
            }
            final RedirectExec redirectExecInstance = RedirectExec.class.cast(
                    redirectExec.get( httpClient ) );

            final Field requestExecutor = redirectExecInstance.getClass().getDeclaredField( "requestExecutor" );
            if ( !requestExecutor.isAccessible() )
            {
                requestExecutor.setAccessible( true );
            }
            final RetryExec requestExecutorInstance = RetryExec.class.cast(
                    requestExecutor.get( redirectExecInstance ) );

            final Field retryHandler = requestExecutorInstance.getClass().getDeclaredField( "retryHandler" );
            if ( !retryHandler.isAccessible() )
            {
                retryHandler.setAccessible( true );
            }
            return HttpRequestRetryHandler.class.cast( retryHandler.get( requestExecutorInstance ) );
        }
        catch ( final Exception e )
        {
            throw new IllegalStateException(e);
        }
    }

    private void doTestHttpClient( final Runnable test ) throws Exception
    {
        final String classpath = System.getProperty( "java.class.path" );
        final String[] paths = classpath.split( File.pathSeparator );
        final Collection<URL> urls = new ArrayList<>( paths.length );
        for ( final String path : paths )
        {
            try
            {
                urls.add( new File( path ).toURI().toURL() );
            }
            catch ( final MalformedURLException e )
            {
                fail( e.getMessage() );
            }
        }
        final URLClassLoader loader = new URLClassLoader( urls.toArray( new URL[ paths.length ] ) , new ClassLoader()
        {
            @Override
            protected Class<?> loadClass( final String name, final boolean resolve ) throws ClassNotFoundException
            {
                if ( name.startsWith( "org.apache.maven.wagon.shared.http" ) )
                {
                    throw new ClassNotFoundException( name );
                }
                return super.loadClass( name, resolve );
            }
        });
        final Thread thread = Thread.currentThread();
        final ClassLoader contextClassLoader = thread.getContextClassLoader();
        thread.setContextClassLoader( loader );

        final String originalClass = System.getProperty( "maven.wagon.http.retryHandler.class", "default" );
        final String originalSentEnabled = System.getProperty(
                "maven.wagon.http.retryHandler.requestSentEnabled", "false" );
        final String originalCount = System.getProperty( "maven.wagon.http.retryHandler.count", "3" );
        final String originalExceptions = System.getProperty( "maven.wagon.http.retryHandler.nonRetryableClasses",
                InterruptedIOException.class.getName() + ","
                    + UnknownHostException.class.getName() + ","
                    + ConnectException.class.getName() + ","
                    + SSLException.class.getName());
        try
        {
            test.run();
        }
        finally
        {
            loader.close();
            thread.setContextClassLoader( contextClassLoader );
            System.setProperty(  "maven.wagon.http.retryHandler.class", originalClass );
            System.setProperty(  "maven.wagon.http.retryHandler.requestSentEnabled", originalSentEnabled );
            System.setProperty(  "maven.wagon.http.retryHandler.count", originalCount );
            System.setProperty(  "maven.wagon.http.retryHandler.nonRetryableClasses", originalExceptions );
        }
    }
}