package org.apache.maven.plugin.assembly.archive.task.testutils;
   
   /*
    * 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 org.apache.maven.artifact.repository.ArtifactRepository;
  import org.apache.maven.execution.MavenSession;
  import org.apache.maven.plugin.assembly.AssemblerConfigurationSource;
  import org.apache.maven.plugin.assembly.testutils.MockManager;
  import org.apache.maven.project.MavenProject;
  import org.apache.maven.project.MavenProjectBuilder;
  import org.apache.maven.project.ProjectBuildingException;
  import org.codehaus.plexus.archiver.Archiver;
  import org.codehaus.plexus.archiver.ArchiverException;
  import org.codehaus.plexus.archiver.manager.ArchiverManager;
  import org.easymock.MockControl;
  
  import java.io.File;
  import java.util.List;
  
  import junit.framework.Assert;
  
  public class MockAndControlForAddDependencySetsTask
  {
  
      public Archiver archiver;
  
      public MockControl archiverCtl;
  
      public AssemblerConfigurationSource configSource;
  
      public MockControl configSourceCtl;
  
      public MockControl dependencyResolverCtl;
  
      public MavenProjectBuilder projectBuilder;
  
      public MockControl archiverManagerCtl;
  
      public ArchiverManager archiverManager;
  
      public MockControl projectBuilderCtl;
  
      private final MavenProject project;
  
      public MockAndControlForAddDependencySetsTask( final MockManager mockManager )
      {
          this( mockManager, null );
      }
  
      public MockAndControlForAddDependencySetsTask( final MockManager mockManager, final MavenProject project )
      {
          this.project = project;
  
          archiverCtl = MockControl.createControl( Archiver.class );
          mockManager.add( archiverCtl );
  
          archiver = (Archiver) archiverCtl.getMock();
  
          configSourceCtl = MockControl.createControl( AssemblerConfigurationSource.class );
          mockManager.add( configSourceCtl );
  
          configSource = (AssemblerConfigurationSource) configSourceCtl.getMock();
  
          projectBuilderCtl = MockControl.createControl( MavenProjectBuilder.class );
          mockManager.add( projectBuilderCtl );
  
          projectBuilder = (MavenProjectBuilder) projectBuilderCtl.getMock();
  
          archiverManagerCtl = MockControl.createControl( ArchiverManager.class );
          mockManager.add( archiverManagerCtl );
  
          archiverManager = (ArchiverManager) archiverManagerCtl.getMock();
  
          enableDefaultExpectations();
      }
  
      private void enableDefaultExpectations()
      {
          configSource.getProject();
          configSourceCtl.setReturnValue( project, MockControl.ZERO_OR_MORE );
      }
  
     public void expectAddArchivedFileSet( final File file, final String outputLocation, final String[] includes,
                                           final String[] excludes )
     {
         try
         {
             archiver.addArchivedFileSet( file, outputLocation, includes, excludes );
 
             if ( ( includes != null ) || ( excludes != null ) )
             {
                 archiverCtl.setMatcher( MockControl.ARRAY_MATCHER );
             }
 
             archiverCtl.setVoidCallable( MockControl.ONE_OR_MORE );
         }
         catch ( final ArchiverException e )
         {
             Assert.fail( "Should never happen." );
         }
     }
 
     public void expectModeChange( final int originalDirMode, final int originalFileMode, final int dirMode,
                                   final int fileMode, final int numberOfChanges )
     {
         archiver.getOverrideDirectoryMode();
         archiverCtl.setReturnValue( originalDirMode );
 
         archiver.getOverrideFileMode();
         archiverCtl.setReturnValue( originalFileMode );
 
         // one of the changes will occur below, when we restore the original mode.
         if ( numberOfChanges > 1 )
         {
             for ( int i = 1; i < numberOfChanges; i++ )
             {
                 archiver.setDirectoryMode( dirMode );
                 archiver.setFileMode( fileMode );
             }
         }
 
         archiver.setDirectoryMode( originalDirMode );
         archiver.setFileMode( originalFileMode );
     }
 
     public void expectAddFile( final File file, final String outputLocation )
     {
         try
         {
             archiver.addFile( file, outputLocation );
         }
         catch ( final ArchiverException e )
         {
             Assert.fail( "Should never happen." );
         }
     }
 
     public void expectAddFile( final File file, final String outputLocation, final int fileMode )
     {
         try
         {
             archiver.addFile( file, outputLocation, fileMode );
         }
         catch ( final ArchiverException e )
         {
             Assert.fail( "Should never happen." );
         }
     }
 
     public void expectGetReactorProjects( final List<MavenProject> projects )
     {
         configSource.getReactorProjects();
         configSourceCtl.setReturnValue( projects, MockControl.ONE_OR_MORE );
     }
 
     public void expectCSGetFinalName( final String finalName )
     {
         configSource.getFinalName();
         configSourceCtl.setReturnValue( finalName, MockControl.ONE_OR_MORE );
     }
 
     public void expectGetDestFile( final File destFile )
     {
         archiver.getDestFile();
         archiverCtl.setReturnValue( destFile, MockControl.ZERO_OR_MORE );
     }
 
     public void expectCSGetRepositories( final ArtifactRepository localRepo, final List<ArtifactRepository> remoteRepos )
     {
         configSource.getLocalRepository();
         configSourceCtl.setReturnValue( localRepo, MockControl.ONE_OR_MORE );
 
         configSource.getRemoteRepositories();
         configSourceCtl.setReturnValue( remoteRepos, MockControl.ONE_OR_MORE );
     }
 
     public void expectBuildFromRepository( final ProjectBuildingException error )
     {
         try
         {
             projectBuilder.buildFromRepository( null, null, null );
             projectBuilderCtl.setMatcher( MockControl.ALWAYS_MATCHER );
             projectBuilderCtl.setThrowable( error, MockControl.ONE_OR_MORE );
         }
         catch ( final ProjectBuildingException e )
         {
             Assert.fail( "should never happen" );
         }
     }
 
     public void expectBuildFromRepository( final MavenProject project )
     {
         try
         {
             projectBuilder.buildFromRepository( null, null, null );
             projectBuilderCtl.setMatcher( MockControl.ALWAYS_MATCHER );
             projectBuilderCtl.setReturnValue( project, MockControl.ONE_OR_MORE );
         }
         catch ( final ProjectBuildingException e )
         {
             Assert.fail( "should never happen" );
         }
     }
 
     public void expectGetSession( final MavenSession session )
     {
         configSource.getMavenSession();
         configSourceCtl.setReturnValue( session, MockControl.ZERO_OR_MORE );
     }
 
 }