package org.apache.archiva.consumers.lucene;
   
   /*
    * 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.io.File;
  import java.util.List;
  
  import org.apache.commons.io.FileUtils;
  import org.apache.maven.archiva.configuration.ManagedRepositoryConfiguration;
  import org.apache.maven.archiva.model.ArchivaArtifact;
  import org.apache.maven.archiva.repository.ManagedRepositoryContent;
  import org.apache.maven.archiva.repository.RepositoryContentFactory;
  import org.apache.maven.archiva.repository.content.ManagedDefaultRepositoryContent;
  import org.apache.maven.archiva.scheduled.ArchivaTaskScheduler;
  import org.apache.maven.archiva.scheduled.tasks.ArtifactIndexingTask;
  import org.apache.maven.archiva.scheduled.tasks.TaskCreator;
  import org.codehaus.plexus.spring.PlexusInSpringTestCase;
  import org.codehaus.plexus.taskqueue.TaskQueue;
  import org.easymock.MockControl;
  import org.easymock.classextension.MockClassControl;
  
  /**
   * LuceneCleanupRemoveIndexedConsumerTest
   */
  public class LuceneCleanupRemoveIndexedConsumerTest
      extends PlexusInSpringTestCase
  {
      private LuceneCleanupRemoveIndexedConsumer consumer;
  
      private RepositoryContentFactory repoFactory;
  
      private MockControl repoFactoryControl;
  
      private ManagedRepositoryConfiguration repositoryConfig;
  
      private TaskQueue indexingQueue;
  
      public void setUp()
          throws Exception
      {
          super.setUp();
  
          repoFactoryControl = MockClassControl.createControl( RepositoryContentFactory.class );
          repoFactory = (RepositoryContentFactory) repoFactoryControl.getMock();
  
          ArchivaTaskScheduler scheduler = (ArchivaTaskScheduler) lookup( ArchivaTaskScheduler.class );
  
          indexingQueue = (TaskQueue) lookup( TaskQueue.ROLE, "indexing" );
  
          consumer = new LuceneCleanupRemoveIndexedConsumer( repoFactory, scheduler );
  
          repositoryConfig = new ManagedRepositoryConfiguration();
          repositoryConfig.setId( "test-repo" );
          repositoryConfig.setLocation( getBasedir() + "/target/test-classes/test-repo" );
          repositoryConfig.setLayout( "default" );
          repositoryConfig.setName( "Test Repository" );
          repositoryConfig.setScanned( true );
          repositoryConfig.setSnapshots( false );
          repositoryConfig.setReleases( true );
          repositoryConfig.setIndexDir( getBasedir() + "/target/test-classes/test-repo/.cleanup-index" );
      }
  
      public void tearDown()
          throws Exception
      {
          FileUtils.deleteDirectory( new File( repositoryConfig.getIndexDir() ) );
  
          super.tearDown();
      }
  
      @SuppressWarnings( "unchecked" )
      public void testProcessArtifactArtifactDoesNotExist()
          throws Exception
      {
          assertTrue( indexingQueue.getQueueSnapshot().isEmpty() );
  
          ArchivaArtifact artifact =
              new ArchivaArtifact( "org.apache.archiva", "archiva-lucene-consumers", "1.2", null, "jar", "test-repo" );
  
          ManagedRepositoryContent repoContent = new ManagedDefaultRepositoryContent();
          repoContent.setRepository( repositoryConfig );
  
         File artifactFile = new File( repoContent.getRepoRoot(), repoContent.toPath( artifact ) );
 
         repoFactoryControl.expectAndReturn( repoFactory.getManagedRepositoryContent( repositoryConfig.getId() ),
                                             repoContent );
 
         repoFactoryControl.replay();
 
         consumer.processArchivaArtifact( artifact );
 
         repoFactoryControl.verify();
 
         List<ArtifactIndexingTask> queue = indexingQueue.getQueueSnapshot();
         assertEquals( 2, queue.size() );
         ArtifactIndexingTask task =
             TaskCreator.createIndexingTask( repositoryConfig, artifactFile, ArtifactIndexingTask.Action.DELETE, null );
         assertEquals( task, queue.get( 0 ) );
         task =
             TaskCreator.createIndexingTask( repositoryConfig, artifactFile, ArtifactIndexingTask.Action.FINISH, null );
         assertEquals( task, queue.get( 1 ) );
     }
 
     public void testProcessArtifactArtifactExists()
         throws Exception
     {
         assertTrue( indexingQueue.getQueueSnapshot().isEmpty() );
 
         ArchivaArtifact artifact =
             new ArchivaArtifact( "org.apache.maven.archiva", "archiva-lucene-cleanup", "1.0", null, "jar", "test-repo" );
         ManagedRepositoryContent repoContent = new ManagedDefaultRepositoryContent();
         repoContent.setRepository( repositoryConfig );
 
         repoFactoryControl.expectAndReturn( repoFactory.getManagedRepositoryContent( repositoryConfig.getId() ),
                                             repoContent );
 
         repoFactoryControl.replay();
 
         consumer.processArchivaArtifact( artifact );
 
         repoFactoryControl.verify();
 
         assertTrue( indexingQueue.getQueueSnapshot().isEmpty() );
     }
 
     @Override
     protected String getPlexusConfigLocation()
     {
         return "/org/apache/archiva/consumers/lucene/LuceneConsumersTest.xml";
     }
 }