package org.apache.maven.wagon.tck.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.apache.maven.wagon.tck.http.util.TestUtil.getResource;

 import org.apache.maven.wagon.ConnectionException;

 import org.apache.maven.wagon.Wagon;

 import org.apache.maven.wagon.authentication.AuthenticationException;

 import org.apache.maven.wagon.authentication.AuthenticationInfo;

 import org.apache.maven.wagon.proxy.ProxyInfo;

 import org.apache.maven.wagon.repository.Repository;

 import org.apache.maven.wagon.tck.http.fixture.ServerFixture;

 import org.codehaus.plexus.DefaultPlexusContainer;

 import org.codehaus.plexus.PlexusContainer;

 import org.codehaus.plexus.component.configurator.ComponentConfigurationException;

 import org.codehaus.plexus.component.repository.exception.ComponentLifecycleException;

 import org.codehaus.plexus.util.FileUtils;

 import org.junit.After;

 import org.junit.AfterClass;

 import org.junit.Before;

 import org.junit.BeforeClass;

 import java.io.File;

 import java.io.IOException;

 import java.util.HashSet;

 import java.util.Set;

 public abstract class HttpWagonTests

 {

     private ServerFixture serverFixture;

     private static PlexusContainer container;

     private Wagon wagon;

     private static WagonTestCaseConfigurator configurator;

     private String baseUrl;

     private final static Set<File> tmpFiles = new HashSet<File>();

     private Repository repo;

     private final Set<Object> notificationTargets = new HashSet<Object>();

     @Before

     public void beforeEach()

         throws Exception

     {

         serverFixture = new ServerFixture( getPort(), isSsl() );

         wagon = (Wagon) container.lookup( Wagon.ROLE, configurator.getWagonHint() );

     }

     @BeforeClass

     public static void beforeAll()

         throws Exception

     {

         File keystore = getResource( ServerFixture.SERVER_SSL_KEYSTORE_RESOURCE_PATH );

         System.setProperty( "javax.net.ssl.keyStore", keystore.getAbsolutePath() );

         System.setProperty( "javax.net.ssl.keyStorePassword", ServerFixture.SERVER_SSL_KEYSTORE_PASSWORD );

         System.setProperty( "javax.net.ssl.trustStore", keystore.getAbsolutePath() );

         System.setProperty( "javax.net.ssl.trustStorePassword", ServerFixture.SERVER_SSL_KEYSTORE_PASSWORD );

         container = new DefaultPlexusContainer();

         //container.initialize();

         //container.start();

         configurator = (WagonTestCaseConfigurator) container.lookup( WagonTestCaseConfigurator.class.getName() );

     }

     @After

     public void afterEach()

     {

         try

         {

             wagon.disconnect();

         }

         catch ( ConnectionException e )

         {

             e.printStackTrace();

         }

         for ( Object obj : notificationTargets )

         {

             synchronized ( obj )

             {

                 obj.notify();

             }

         }

         if ( serverFixture != null )

         {

             try

             {

                 serverFixture.stop();

             }

             catch ( Exception e )

             {

                 e.printStackTrace();

             }

         }

         try

         {

             container.release( wagon );

         }

         catch ( ComponentLifecycleException e )

         {

             e.printStackTrace();

         }

     }

     @AfterClass

     public static void afterAll()

     {

         for ( File f : tmpFiles )

         {

             if ( f.exists() )

             {

                 try

                 {

                     FileUtils.forceDelete( f );

                 }

                 catch ( IOException e )

                 {

                     e.printStackTrace();

                 }

             }

         }

         if ( container != null )

         {

             try

             {

                 container.release( configurator );

             }

             catch ( ComponentLifecycleException e )

             {

                 e.printStackTrace();

             }

             container.dispose();

         }

     }

     protected void addNotificationTarget( final Object target )

     {

         notificationTargets.add( target );

     }

     protected File newTempFile()

         throws IOException

     {

         File f = File.createTempFile( "wagon-target.", ".file" );

         f.deleteOnExit();

         return f;

     }

     protected boolean isSsl()

     {

         return false;

     }

     protected ProxyInfo newProxyInfo()

     {

         ProxyInfo info = new ProxyInfo();

         info.setType( isSsl() ? "https" : "http" );

         info.setHost( ServerFixture.SERVER_HOST );

         info.setPort( getPort() );

         return info;

     }

     protected boolean isSupported()

     {

         StackTraceElement[] elements = new Throwable().getStackTrace();

         String testCaseId = null;

         String lastMethodName = null;

         for ( StackTraceElement e : elements )

         {

             if ( !e.getClassName().startsWith( getClass().getPackage().getName() ) )

             {

                 testCaseId = lastMethodName;

                 break;

             }

             else

             {

                 lastMethodName = e.getMethodName();

             }

         }

         if ( testCaseId == null || !configurator.isSupported( testCaseId ) )

         {

             System.out.println( "Cannot run test: " + testCaseId

                 + ". Wagon under test does not support this test case." );

             return false;

         }

         return true;

     }

     protected boolean initTest( final AuthenticationInfo auth, final ProxyInfo proxy )

         throws ComponentConfigurationException, ConnectionException, AuthenticationException

     {

         return initTest( getBaseUrl(), auth, proxy );

     }

     protected boolean initTest( final String baseUrl, final AuthenticationInfo auth, final ProxyInfo proxy )

         throws ComponentConfigurationException, ConnectionException, AuthenticationException

     {

         StackTraceElement[] elements = new Throwable().getStackTrace();

         String testCaseId = null;

         String lastMethodName = null;

         for ( StackTraceElement e : elements )

         {

             if ( !e.getClassName().startsWith( getClass().getPackage().getName() ) )

             {

                 testCaseId = lastMethodName;

                 break;

             }

             else

             {

                 lastMethodName = e.getMethodName();

             }

         }

         if ( testCaseId == null || !configurator.configureWagonForTest( wagon, testCaseId ) )

         {

             System.out.println( "Cannot run test: " + testCaseId

                 + ". Wagon under test does not support this test case." );

             return false;

         }

         try

         {

             serverFixture.start();

         }

         catch ( Exception e )

         {

             throw new IllegalStateException( "Failed to start: " + e.getMessage(), e );

         }

         repo = new Repository( "test", baseUrl );

         wagon.connect( repo, auth, proxy );

         return true;

     }

     protected int getPort()

     {

         int port = getPortPropertyValue();

         if ( port < 1 )

         {

             port = getDefaultPort();

         }

         return port;

     }

     protected int getDefaultPort()

     {

         return 9080;

     }

     protected int getPortPropertyValue()

     {

         return Integer.parseInt( System.getProperty( "test.port", "-1" ) );

     }

     protected String getBaseUrl()

     {

         if ( baseUrl == null )

         {

             StringBuilder sb = new StringBuilder();

             sb.append( isSsl() ? "https" : "http" );

             sb.append( "://" + ServerFixture.SERVER_HOST + ":" );

             sb.append( getPort() );

             baseUrl = sb.toString();

         }

         return baseUrl;

     }

     protected ServerFixture getServerFixture()

     {

         return serverFixture;

     }

     protected static PlexusContainer getContainer()

     {

         return container;

     }

     protected Wagon getWagon()

     {

         return wagon;

     }

     protected static WagonTestCaseConfigurator getConfigurator()

     {

         return configurator;

     }

     protected static Set<File> getTmpfiles()

     {

         return tmpFiles;

     }

     protected Repository getRepo()

     {

         return repo;

     }

 }