package org.apache.maven.plugins.dependency.utils.translators;
   
   /* 
    * 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 java.util.HashMap;
  import java.util.HashSet;
  import java.util.Iterator;
  import java.util.Set;
  
  import org.apache.maven.artifact.Artifact;
  import org.apache.maven.artifact.factory.ArtifactFactory;
  import org.apache.maven.artifact.factory.DefaultArtifactFactory;
  import org.apache.maven.artifact.handler.manager.ArtifactHandlerManager;
  import org.apache.maven.artifact.handler.manager.DefaultArtifactHandlerManager;
  import org.apache.maven.artifact.repository.ArtifactRepository;
  import org.apache.maven.execution.MavenSession;
  import org.apache.maven.plugin.logging.Log;
  import org.apache.maven.plugin.testing.SilentLog;
  import org.apache.maven.plugin.testing.stubs.MavenProjectStub;
  import org.apache.maven.plugin.testing.stubs.StubArtifactRepository;
  import org.apache.maven.plugins.dependency.AbstractDependencyMojoTestCase;
  import org.apache.maven.plugins.dependency.testUtils.DependencyArtifactStubFactory;
  import org.apache.maven.project.ProjectBuildingRequest;
  import org.apache.maven.shared.artifact.ArtifactCoordinate;
  import org.apache.maven.shared.repository.RepositoryManager;
  import org.sonatype.aether.impl.internal.SimpleLocalRepositoryManager;
  import org.sonatype.aether.util.DefaultRepositorySystemSession;
  
  /**
   * @author brianf
   */
  public class TestClassifierTypeTranslator
      extends AbstractDependencyMojoTestCase
  {
      Set<Artifact> artifacts = new HashSet<Artifact>();
  
      ArtifactFactory artifactFactory;
  
      ArtifactRepository artifactRepository;
  
      Log log = new SilentLog();
  
      private RepositoryManager repoManager;
  
      private ProjectBuildingRequest buildingRequest;
  
      private ArtifactHandlerManager artifactHandlerManager;
  
      protected void setUp()
          throws Exception
      {
          super.setUp( "classifiertype-translator", false );
  
          artifactHandlerManager = new DefaultArtifactHandlerManager();
          this.setVariableValueToObject( artifactHandlerManager, "artifactHandlers", new HashMap() );
  
          artifactFactory = new DefaultArtifactFactory();
          this.setVariableValueToObject( artifactFactory, "artifactHandlerManager", artifactHandlerManager );
  
          artifactRepository = new StubArtifactRepository( null );
  
          DependencyArtifactStubFactory factory = new DependencyArtifactStubFactory( null, false );
          artifacts = factory.getMixedArtifacts();
  
          repoManager = lookup( RepositoryManager.class );
  
          MavenSession session = newMavenSession( new MavenProjectStub() );
          buildingRequest = session.getProjectBuildingRequest();
  
          DefaultRepositorySystemSession repoSession = (DefaultRepositorySystemSession) session.getRepositorySession();
          repoSession.setLocalRepositoryManager( new SimpleLocalRepositoryManager( stubFactory.getWorkingDir() ) );
  
      }
  
      public void testNullClassifier()
      {
          doTestNullEmptyClassifier( null );
      }
  
      public void testEmptyClassifier()
      {
          doTestNullEmptyClassifier( "" );
     }
 
     public void doTestNullEmptyClassifier( String classifier )
     {
         String type = "zip";
 
         ArtifactTranslator at = new ClassifierTypeTranslator( artifactHandlerManager, classifier, type );
         Set<ArtifactCoordinate> results = at.translate( artifacts, log );
 
         for ( Artifact artifact : artifacts )
         {
             Iterator<ArtifactCoordinate> resultIter = results.iterator();
             boolean found = false;
             while ( resultIter.hasNext() )
             {
                 ArtifactCoordinate translatedArtifact = resultIter.next();
                 if ( artifact.getArtifactId().equals( translatedArtifact.getArtifactId() )
                     && artifact.getGroupId().equals( translatedArtifact.getGroupId() )
                 /* && artifact.getScope().equals(translatedArtifact.getScope()) */ )
                 {
                     // classifier is null, should be the same as the artifact
                     assertEquals( artifact.getClassifier(), translatedArtifact.getClassifier() );
                     assertEquals( type, translatedArtifact.getExtension() );
 
                     found = true;
                     break;
                 }
             }
             assertTrue( found );
         }
     }
 
     public void testNullType()
     {
         doTestNullEmptyType( null );
     }
 
     public void testEmptyType()
     {
         doTestNullEmptyType( "" );
     }
 
     public void doTestNullEmptyType( String type )
     {
         String classifier = "jdk5";
 
         ArtifactTranslator at = new ClassifierTypeTranslator( artifactHandlerManager, classifier, type );
         Set<ArtifactCoordinate> results = at.translate( artifacts, log );
 
         for ( Artifact artifact : artifacts )
         {
             Iterator<ArtifactCoordinate> resultIter = results.iterator();
             boolean found = false;
             while ( !found && resultIter.hasNext() )
             {
                 ArtifactCoordinate translatedArtifact = resultIter.next();
                 if ( artifact.getArtifactId() == translatedArtifact.getArtifactId()
                     && artifact.getGroupId() == translatedArtifact.getGroupId()
                 /* && artifact.getScope() == translatedArtifact.getScope() */ )
                 {
                     // classifier is null, should be the same as the artifact
                     assertEquals( classifier, translatedArtifact.getClassifier() );
                     assertEquals( artifact.getType(), translatedArtifact.getExtension() );
 
                     found = true;
                     break;
                 }
             }
             assertTrue( found );
         }
     }
 
     public void testClassifierAndType()
     {
         String classifier = "jdk14";
         String type = "sources";
         ArtifactTranslator at = new ClassifierTypeTranslator( artifactHandlerManager, classifier, type );
         Set<ArtifactCoordinate> results = at.translate( artifacts, log );
 
         for ( Artifact artifact : artifacts )
         {
             Iterator<ArtifactCoordinate> resultIter = results.iterator();
             boolean found = false;
             while ( !found && resultIter.hasNext() )
             {
                 ArtifactCoordinate translatedArtifact = resultIter.next();
                 if ( artifact.getArtifactId() == translatedArtifact.getArtifactId()
                     && artifact.getGroupId() == translatedArtifact.getGroupId()
                 /* && artifact.getScope() == translatedArtifact.getScope() */ )
                 {
                     assertEquals( translatedArtifact.getClassifier(), classifier );
                     assertEquals( translatedArtifact.getExtension(), type );
 
                     found = true;
                     break;
                 }
             }
             assertTrue( found );
         }
     }
 
     public void testGetterSetter()
     {
         String classifier = "class";
         String type = "type";
         ClassifierTypeTranslator at = new ClassifierTypeTranslator( artifactHandlerManager, classifier, type );
 
         assertEquals( classifier, at.getClassifier() );
         assertEquals( type, at.getType() );
 
         at.setClassifier( type );
         at.setType( classifier );
 
         assertEquals( type, at.getClassifier() );
         assertEquals( classifier, at.getType() );
 
     }
 }