source.dussan.org Git - archiva.git/blob

   1 package org.apache.archiva.indexer.search;
   2
   3 /*
   4 * Licensed to the Apache Software Foundation (ASF) under one
   5 * or more contributor license agreements.  See the NOTICE file
   6 * distributed with this work for additional information
   7 * regarding copyright ownership.  The ASF licenses this file
   8 * to you under the Apache License, Version 2.0 (the
   9 * "License"); you may not use this file except in compliance
  10 * with the License.  You may obtain a copy of the License at
  11 *
  12 *  http://www.apache.org/licenses/LICENSE-2.0
  13 *
  14 * Unless required by applicable law or agreed to in writing,
  15 * software distributed under the License is distributed on an
  16 * "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY
  17 * KIND, either express or implied.  See the License for the
  18 * specific language governing permissions and limitations
  19 * under the License.
  20 */
  21
  22 import junit.framework.TestCase;
  23 import org.apache.archiva.admin.repository.managed.DefaultManagedRepositoryAdmin;
  24 import org.apache.archiva.admin.repository.proxyconnector.DefaultProxyConnectorAdmin;
  25 import org.apache.archiva.common.plexusbridge.PlexusSisuBridge;
  26 import org.apache.archiva.configuration.ArchivaConfiguration;
  27 import org.apache.archiva.configuration.Configuration;
  28 import org.apache.archiva.configuration.ManagedRepositoryConfiguration;
  29 import org.apache.archiva.test.utils.ArchivaSpringJUnit4ClassRunner;
  30 import org.apache.commons.io.FileUtils;
  31 import org.apache.commons.lang.SystemUtils;
  32 import org.apache.maven.index.ArtifactContext;
  33 import org.apache.maven.index.ArtifactContextProducer;
  34 import org.apache.maven.index.ArtifactScanningListener;
  35 import org.apache.maven.index.NexusIndexer;
  36 import org.apache.maven.index.QueryCreator;
  37 import org.apache.maven.index.ScanningResult;
  38 import org.apache.maven.index.context.IndexCreator;
  39 import org.apache.maven.index.context.IndexingContext;
  40 import org.easymock.EasyMock;
  41 import org.easymock.IMocksControl;
  42 import org.junit.After;
  43 import org.junit.Before;
  44 import org.junit.runner.RunWith;
  45 import org.slf4j.Logger;
  46 import org.slf4j.LoggerFactory;
  47 import org.springframework.test.context.ContextConfiguration;
  48
  49 import javax.inject.Inject;
  50 import java.io.File;
  51 import java.util.List;
  52
  53 /**
  54  * @author Olivier Lamy
  55  */
  56 @RunWith( ArchivaSpringJUnit4ClassRunner.class )
  57 @ContextConfiguration( locations = { "classpath*:/META-INF/spring-context.xml", "classpath:/spring-context.xml" } )
  58 public abstract class AbstractMavenRepositorySearch
  59     extends TestCase
  60 {
  61
  62     protected Logger log = LoggerFactory.getLogger( getClass() );
  63
  64     public static String TEST_REPO_1 = "maven-search-test-repo";
  65
  66     public static String TEST_REPO_2 = "maven-search-test-repo-2";
  67
  68
  69     public static String REPO_RELEASE = "repo-release";
  70
  71     MavenRepositorySearch search;
  72
  73     ArchivaConfiguration archivaConfig;
  74
  75     @Inject
  76     ArtifactContextProducer artifactContextProducer;
  77
  78     IMocksControl archivaConfigControl;
  79
  80     Configuration config;
  81
  82     @Inject
  83     PlexusSisuBridge plexusSisuBridge;
  84
  85     @Inject
  86     List<IndexCreator> indexCreators;
  87
  88     @Inject
  89     NexusIndexer nexusIndexer;
  90
  91     @Inject
  92     QueryCreator queryCreator;
  93
  94     @Before
  95     @Override
  96     public void setUp()
  97         throws Exception
  98     {
  99         super.setUp();
 100
 101         FileUtils.deleteDirectory( new File( org.apache.archiva.common.utils.FileUtils.getBasedir(), "/target/repos/" + TEST_REPO_1 + "/.indexer" ) );
 102         assertFalse( new File( org.apache.archiva.common.utils.FileUtils.getBasedir(), "/target/repos/" + TEST_REPO_1 + "/.indexer" ).exists() );
 103
 104         FileUtils.deleteDirectory( new File( org.apache.archiva.common.utils.FileUtils.getBasedir(), "/target/repos/" + TEST_REPO_2 + "/.indexer" ) );
 105         assertFalse( new File( org.apache.archiva.common.utils.FileUtils.getBasedir(), "/target/repos/" + TEST_REPO_2 + "/.indexer" ).exists() );
 106
 107         archivaConfigControl = EasyMock.createControl();
 108
 109         archivaConfig = archivaConfigControl.createMock( ArchivaConfiguration.class );
 110
 111         DefaultManagedRepositoryAdmin defaultManagedRepositoryAdmin = new DefaultManagedRepositoryAdmin();
 112         defaultManagedRepositoryAdmin.setArchivaConfiguration( archivaConfig );
 113
 114         DefaultProxyConnectorAdmin defaultProxyConnectorAdmin = new DefaultProxyConnectorAdmin();
 115         defaultProxyConnectorAdmin.setArchivaConfiguration( archivaConfig );
 116
 117         search = new MavenRepositorySearch( nexusIndexer, defaultManagedRepositoryAdmin, defaultProxyConnectorAdmin,
 118                                             queryCreator );
 119
 120         defaultManagedRepositoryAdmin.setIndexer( nexusIndexer );
 121         defaultManagedRepositoryAdmin.setIndexCreators( indexCreators );
 122
 123         config = new Configuration();
 124         config.addManagedRepository( createRepositoryConfig( TEST_REPO_1 ) );
 125         config.addManagedRepository( createRepositoryConfig( TEST_REPO_2 ) );
 126         config.addManagedRepository( createRepositoryConfig( REPO_RELEASE ) );
 127     }
 128
 129     @After
 130     @Override
 131     public void tearDown()
 132         throws Exception
 133     {
 134         for ( IndexingContext indexingContext : nexusIndexer.getIndexingContexts().values() )
 135         {
 136             nexusIndexer.removeIndexingContext( indexingContext, true );
 137         }
 138
 139         FileUtils.deleteDirectory( new File( org.apache.archiva.common.utils.FileUtils.getBasedir(), "/target/repos/" + TEST_REPO_1 ) );
 140         assertFalse( new File( org.apache.archiva.common.utils.FileUtils.getBasedir(), "/target/repos/" + TEST_REPO_1 ).exists() );
 141
 142         FileUtils.deleteDirectory( new File( org.apache.archiva.common.utils.FileUtils.getBasedir(), "/target/repos/" + TEST_REPO_2 ) );
 143         assertFalse( new File( org.apache.archiva.common.utils.FileUtils.getBasedir(), "/target/repos/" + TEST_REPO_2 ).exists() );
 144
 145         super.tearDown();
 146     }
 147
 148     protected ManagedRepositoryConfiguration createRepositoryConfig( String repository )
 149     {
 150         ManagedRepositoryConfiguration repositoryConfig = new ManagedRepositoryConfiguration();
 151         repositoryConfig.setId( repository );
 152         repositoryConfig.setLocation( org.apache.archiva.common.utils.FileUtils.getBasedir() + "/target/repos/" + repository );
 153         File f = new File( repositoryConfig.getLocation() );
 154         if ( !f.exists() )
 155         {
 156             f.mkdirs();
 157         }
 158         repositoryConfig.setLayout( "default" );
 159         repositoryConfig.setName( repository );
 160         repositoryConfig.setScanned( true );
 161         repositoryConfig.setSnapshots( false );
 162         repositoryConfig.setReleases( true );
 163
 164         return repositoryConfig;
 165     }
 166
 167     protected void createIndex( String repository, List<File> filesToBeIndexed, boolean scan )
 168         throws Exception
 169     {
 170
 171         IndexingContext context = nexusIndexer.getIndexingContexts().get( repository );
 172
 173         if ( context != null )
 174         {
 175             nexusIndexer.removeIndexingContext( context, true );
 176         }
 177
 178         File indexerDirectory = new File( org.apache.archiva.common.utils.FileUtils.getBasedir(), "/target/repos/" + repository + "/.indexer" );
 179
 180         if ( indexerDirectory.exists() )
 181         {
 182             FileUtils.deleteDirectory( indexerDirectory );
 183         }
 184
 185         assertFalse( indexerDirectory.exists() );
 186
 187         File lockFile = new File( org.apache.archiva.common.utils.FileUtils.getBasedir(), "/target/repos/" + repository + "/.indexer/write.lock" );
 188         if ( lockFile.exists() )
 189         {
 190             lockFile.delete();
 191         }
 192
 193         assertFalse( lockFile.exists() );
 194
 195         File repo = new File( org.apache.archiva.common.utils.FileUtils.getBasedir(), "src/test/" + repository );
 196         assertTrue( repo.exists() );
 197         File indexDirectory =
 198             new File( org.apache.archiva.common.utils.FileUtils.getBasedir(), "target/index/test-" + Long.toString( System.currentTimeMillis() ) );
 199         indexDirectory.deleteOnExit();
 200         FileUtils.deleteDirectory( indexDirectory );
 201
 202         context = nexusIndexer.addIndexingContext( repository, repository, repo, indexDirectory,
 203                                                    repo.toURI().toURL().toExternalForm(),
 204                                                    indexDirectory.toURI().toURL().toString(), indexCreators );
 205
 206         // minimize datas in memory
 207 //        context.getIndexWriter().setMaxBufferedDocs( -1 );
 208 //        context.getIndexWriter().setRAMBufferSizeMB( 1 );
 209         for ( File artifactFile : filesToBeIndexed )
 210         {
 211             assertTrue( "file not exists " + artifactFile.getPath(), artifactFile.exists() );
 212             ArtifactContext ac = artifactContextProducer.getArtifactContext( context, artifactFile );
 213
 214             if ( artifactFile.getPath().endsWith( ".pom" ) )
 215             {
 216                 ac.getArtifactInfo().setFileExtension( "pom" );
 217                 ac.getArtifactInfo().setPackaging( "pom" );
 218                 ac.getArtifactInfo().setClassifier( "pom" );
 219             }
 220             nexusIndexer.addArtifactToIndex( ac, context );
 221             context.updateTimestamp( true );
 222         }
 223
 224         if ( scan )
 225         {
 226             nexusIndexer.scan( context, new ArtifactScanListener(), false );
 227         }
 228         // force flushing
 229         context.getIndexWriter().commit();
 230         context.setSearchable( true );
 231
 232     }
 233
 234     static class ArtifactScanListener
 235         implements ArtifactScanningListener
 236     {
 237         protected Logger log = LoggerFactory.getLogger( getClass() );
 238
 239         @Override
 240         public void scanningStarted( IndexingContext ctx )
 241         {
 242             //
 243         }
 244
 245         @Override
 246         public void scanningFinished( IndexingContext ctx, ScanningResult result )
 247         {
 248             // no op
 249         }
 250
 251         @Override
 252         public void artifactError( ArtifactContext ac, Exception e )
 253         {
 254             log.debug( "artifactError {}", ac.getArtifact().getPath(), e );
 255         }
 256
 257         @Override
 258         public void artifactDiscovered( ArtifactContext ac )
 259         {
 260             log.debug( "artifactDiscovered {}:{}", //
 261                        ac.getArtifact() == null ? "" : ac.getArtifact().getPath(), //
 262                        ac.getArtifact() == null ? "" : ac.getArtifactInfo() );
 263         }
 264     }
 265
 266     public String niceDisplay( SearchResults searchResults )
 267         throws Exception
 268     {
 269         StringBuilder sb = new StringBuilder();
 270         for ( SearchResultHit hit : searchResults.getHits() )
 271         {
 272             sb.append( hit.toString() ).append( SystemUtils.LINE_SEPARATOR );
 273         }
 274         return sb.toString();
 275     }
 276 }