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.MavenIndexerUtils;
  26 import org.apache.archiva.common.plexusbridge.PlexusSisuBridge;
  27 import org.apache.archiva.common.utils.FileUtil;
  28 import org.apache.archiva.configuration.ArchivaConfiguration;
  29 import org.apache.archiva.configuration.Configuration;
  30 import org.apache.archiva.configuration.ManagedRepositoryConfiguration;
  31 import org.apache.commons.io.FileUtils;
  32 import org.apache.commons.lang.SystemUtils;
  33 import org.apache.maven.index.ArtifactContext;
  34 import org.apache.maven.index.ArtifactContextProducer;
  35 import org.apache.maven.index.ArtifactScanningListener;
  36 import org.apache.maven.index.NexusIndexer;
  37 import org.apache.maven.index.ScanningResult;
  38 import org.apache.maven.index.context.IndexingContext;
  39 import org.easymock.MockControl;
  40 import org.junit.After;
  41 import org.junit.Before;
  42 import org.junit.runner.RunWith;
  43 import org.slf4j.Logger;
  44 import org.slf4j.LoggerFactory;
  45 import org.springframework.test.context.ContextConfiguration;
  46 import org.springframework.test.context.junit4.SpringJUnit4ClassRunner;
  47
  48 import javax.inject.Inject;
  49 import java.io.File;
  50 import java.util.List;
  51 import org.apache.archiva.test.ArchivaSpringJUnit4ClassRunner;
  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 AbstractNexusRepositorySearch
  59     extends TestCase
  60 {
  61
  62     protected Logger log = LoggerFactory.getLogger( getClass() );
  63
  64     public static String TEST_REPO_1 = "nexus-search-test-repo";
  65
  66     public static String TEST_REPO_2 = "nexus-search-test-repo-2";
  67
  68     NexusRepositorySearch search;
  69
  70     ArchivaConfiguration archivaConfig;
  71
  72     ArtifactContextProducer artifactContextProducer;
  73
  74     MockControl archivaConfigControl;
  75
  76     Configuration config;
  77
  78     @Inject
  79     PlexusSisuBridge plexusSisuBridge;
  80
  81     @Inject
  82     MavenIndexerUtils mavenIndexerUtils;
  83
  84     NexusIndexer nexusIndexer;
  85
  86     @Before
  87     public void setUp()
  88         throws Exception
  89     {
  90         super.setUp();
  91
  92         FileUtils.deleteDirectory( new File( FileUtil.getBasedir(), "/target/repos/" + TEST_REPO_1 + "/.indexer" ) );
  93         assertFalse( new File( FileUtil.getBasedir(), "/target/repos/" + TEST_REPO_1 + "/.indexer" ).exists() );
  94
  95         FileUtils.deleteDirectory( new File( FileUtil.getBasedir(), "/target/repos/" + TEST_REPO_2 + "/.indexer" ) );
  96         assertFalse( new File( FileUtil.getBasedir(), "/target/repos/" + TEST_REPO_2 + "/.indexer" ).exists() );
  97
  98         archivaConfigControl = MockControl.createControl( ArchivaConfiguration.class );
  99
 100         archivaConfig = (ArchivaConfiguration) archivaConfigControl.getMock();
 101
 102         DefaultManagedRepositoryAdmin defaultManagedRepositoryAdmin = new DefaultManagedRepositoryAdmin();
 103         defaultManagedRepositoryAdmin.setArchivaConfiguration( archivaConfig );
 104
 105         DefaultProxyConnectorAdmin defaultProxyConnectorAdmin = new DefaultProxyConnectorAdmin();
 106         defaultProxyConnectorAdmin.setArchivaConfiguration( archivaConfig );
 107
 108         search = new NexusRepositorySearch( plexusSisuBridge, defaultManagedRepositoryAdmin, mavenIndexerUtils,
 109                                             defaultProxyConnectorAdmin );
 110
 111         nexusIndexer = plexusSisuBridge.lookup( NexusIndexer.class );
 112
 113         artifactContextProducer = plexusSisuBridge.lookup( ArtifactContextProducer.class );
 114
 115         config = new Configuration();
 116         config.addManagedRepository( createRepositoryConfig( TEST_REPO_1 ) );
 117         config.addManagedRepository( createRepositoryConfig( TEST_REPO_2 ) );
 118     }
 119
 120     @After
 121     public void tearDown()
 122         throws Exception
 123     {
 124
 125         for ( IndexingContext indexingContext : nexusIndexer.getIndexingContexts().values() )
 126         {
 127             nexusIndexer.removeIndexingContext( indexingContext, true );
 128         }
 129
 130         FileUtils.deleteDirectory( new File( FileUtil.getBasedir(), "/target/repos/" + TEST_REPO_1 ) );
 131         assertFalse( new File( FileUtil.getBasedir(), "/target/repos/" + TEST_REPO_1 ).exists() );
 132
 133         FileUtils.deleteDirectory( new File( FileUtil.getBasedir(), "/target/repos/" + TEST_REPO_2 ) );
 134         assertFalse( new File( FileUtil.getBasedir(), "/target/repos/" + TEST_REPO_2 ).exists() );
 135
 136         super.tearDown();
 137     }
 138
 139     protected ManagedRepositoryConfiguration createRepositoryConfig( String repository )
 140     {
 141         ManagedRepositoryConfiguration repositoryConfig = new ManagedRepositoryConfiguration();
 142         repositoryConfig.setId( repository );
 143         repositoryConfig.setLocation( FileUtil.getBasedir() + "/target/repos/" + repository );
 144         File f = new File( repositoryConfig.getLocation() );
 145         if ( !f.exists() )
 146         {
 147             f.mkdirs();
 148         }
 149         repositoryConfig.setLayout( "default" );
 150         repositoryConfig.setName( repository );
 151         repositoryConfig.setScanned( true );
 152         repositoryConfig.setSnapshots( false );
 153         repositoryConfig.setReleases( true );
 154
 155         return repositoryConfig;
 156     }
 157
 158     protected void createIndex( String repository, List<File> filesToBeIndexed, boolean scan )
 159         throws Exception
 160     {
 161
 162         IndexingContext context = nexusIndexer.getIndexingContexts().get( repository );
 163
 164         if ( context != null )
 165         {
 166             nexusIndexer.removeIndexingContext( context, true );
 167         }
 168
 169         File indexerDirectory = new File( FileUtil.getBasedir(), "/target/repos/" + repository + "/.indexer" );
 170
 171         if ( indexerDirectory.exists() )
 172         {
 173             FileUtils.deleteDirectory( indexerDirectory );
 174         }
 175
 176         assertFalse( indexerDirectory.exists() );
 177
 178         File lockFile = new File( FileUtil.getBasedir(), "/target/repos/" + repository + "/.indexer/write.lock" );
 179         if ( lockFile.exists() )
 180         {
 181             lockFile.delete();
 182         }
 183
 184         assertFalse( lockFile.exists() );
 185
 186         File repo = new File( FileUtil.getBasedir(), "src/test/" + repository );
 187         assertTrue( repo.exists() );
 188         File indexDirectory =
 189             new File( FileUtil.getBasedir(), "target/index/test-" + Long.toString( System.currentTimeMillis() ) );
 190         indexDirectory.deleteOnExit();
 191         FileUtils.deleteDirectory( indexDirectory );
 192
 193         context = nexusIndexer.addIndexingContext( repository, repository, repo, indexDirectory,
 194                                                    repo.toURI().toURL().toExternalForm(),
 195                                                    indexDirectory.toURI().toURL().toString(),
 196                                                    search.getAllIndexCreators() );
 197
 198         // minimize datas in memory
 199         context.getIndexWriter().setMaxBufferedDocs( -1 );
 200         context.getIndexWriter().setRAMBufferSizeMB( 1 );
 201         for ( File artifactFile : filesToBeIndexed )
 202         {
 203             assertTrue( "file not exists " + artifactFile.getPath(), artifactFile.exists() );
 204             ArtifactContext ac = artifactContextProducer.getArtifactContext( context, artifactFile );
 205             nexusIndexer.addArtifactToIndex( ac, context );
 206             context.updateTimestamp( true );
 207         }
 208
 209         if ( scan )
 210         {
 211             nexusIndexer.scan( context, new ArtifactScanListener(), false );
 212         }
 213         // force flushing
 214         context.getIndexWriter().commit();
 215         context.getIndexWriter().close( true );
 216         // wait for io flush ....
 217         //Thread.sleep( 2000 );
 218         context.setSearchable( true );
 219
 220     }
 221
 222     static class ArtifactScanListener
 223         implements ArtifactScanningListener
 224     {
 225         protected Logger log = LoggerFactory.getLogger( getClass() );
 226
 227         public void scanningStarted( IndexingContext ctx )
 228         {
 229
 230         }
 231
 232         public void scanningFinished( IndexingContext ctx, ScanningResult result )
 233         {
 234
 235         }
 236
 237         public void artifactError( ArtifactContext ac, Exception e )
 238         {
 239             log.debug( "artifactError " + ac.getArtifact().getPath(), e );
 240         }
 241
 242         public void artifactDiscovered( ArtifactContext ac )
 243         {
 244             log.debug( "artifactDiscovered {}:", ac.getArtifact().getPath(), ac.getArtifactInfo() );
 245         }
 246     }
 247
 248     public String niceDisplay( SearchResults searchResults )
 249         throws Exception
 250     {
 251         StringBuilder sb = new StringBuilder();
 252         for ( SearchResultHit hit : searchResults.getHits() )
 253         {
 254             sb.append( hit.toString() ).append( SystemUtils.LINE_SEPARATOR );
 255         }
 256         return sb.toString();
 257     }
 258 }