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

   1 package org.apache.maven.repository.indexing;
   2
   3 /*
   4  * Copyright 2005-2006 The Apache Software Foundation.
   5  *
   6  * Licensed under the Apache License, Version 2.0 (the "License");
   7  * you may not use this file except in compliance with the License.
   8  * You may obtain a copy of the License at
   9  *
  10  *      http://www.apache.org/licenses/LICENSE-2.0
  11  *
  12  * Unless required by applicable law or agreed to in writing, software
  13  * distributed under the License is distributed on an "AS IS" BASIS,
  14  * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
  15  * See the License for the specific language governing permissions and
  16  * limitations under the License.
  17  */
  18
  19 import org.apache.maven.artifact.Artifact;
  20 import org.apache.maven.artifact.factory.ArtifactFactory;
  21 import org.apache.maven.artifact.repository.ArtifactRepository;
  22 import org.apache.maven.artifact.repository.ArtifactRepositoryFactory;
  23 import org.apache.maven.artifact.repository.layout.ArtifactRepositoryLayout;
  24 import org.codehaus.plexus.PlexusTestCase;
  25 import org.codehaus.plexus.util.FileUtils;
  26
  27 import java.io.File;
  28 import java.util.List;
  29
  30 /**
  31  *
  32  */
  33 public class ArtifactRepositoryIndexingTest
  34     extends PlexusTestCase
  35 {
  36     private static final String GROUPID = "groupId";
  37
  38     private static final String ARTIFACTID = "artifactId";
  39
  40     private static final String VERSION = "version";
  41
  42     private static final String CLASSES = "classes";
  43
  44     private static final String PACKAGES = "packages";
  45
  46     private static final String FILES = "files";
  47
  48     private ArtifactFactory artifactFactory;
  49
  50     private ArtifactRepository repository;
  51
  52     private String indexPath;
  53
  54     protected void setUp()
  55         throws Exception
  56     {
  57         super.setUp();
  58
  59         File repositoryDirectory = getTestFile( "src/test/repository" );
  60         String repoDir = repositoryDirectory.toURL().toString();
  61         ArtifactRepositoryLayout layout = (ArtifactRepositoryLayout) lookup( ArtifactRepositoryLayout.ROLE, "default" );
  62         ArtifactRepositoryFactory repoFactory = (ArtifactRepositoryFactory) lookup( ArtifactRepositoryFactory.ROLE );
  63         repository = repoFactory.createArtifactRepository( "test", repoDir, layout, null, null );
  64
  65         indexPath = "target/index";
  66
  67         FileUtils.deleteDirectory( indexPath );
  68     }
  69
  70     public void testIndexerExceptions()
  71         throws Exception
  72     {
  73         ArtifactRepositoryIndex indexer;
  74         try
  75         {
  76             String notIndexDir = new File( "pom.xml" ).getAbsolutePath();
  77             indexer = (ArtifactRepositoryIndex) lookup( RepositoryIndex.ROLE, "artifact" );
  78             indexer.open( notIndexDir );
  79             fail( "Must throw exception on non-directory index directory" );
  80         }
  81         catch ( RepositoryIndexException e )
  82         {
  83             //expected
  84         }
  85
  86         try
  87         {
  88             String notIndexDir = new File( "" ).getAbsolutePath();
  89             indexer = (ArtifactRepositoryIndex) lookup( RepositoryIndex.ROLE, "artifact" );
  90             indexer.open( notIndexDir );
  91             fail( "Must throw an exception on a non-index directory" );
  92         }
  93         catch ( RepositoryIndexException e )
  94         {
  95             //expected
  96         }
  97
  98         //indexer = (ArtifactRepositoryIndex) factory.getArtifactRepositoryIndexer( indexPath, repository );
  99         //indexer.close();
 100         indexer = (ArtifactRepositoryIndex) lookup( RepositoryIndex.ROLE, "artifact" );
 101         Artifact artifact = getArtifact( "test", "test-artifactId", "1.0" );
 102         artifact.setFile( new File( repository.getBasedir(), repository.pathOf( artifact ) ) );
 103
 104         try
 105         {
 106             indexer.indexArtifact( artifact );
 107             fail( "Must throw exception on add index with closed index." );
 108         }
 109         catch ( RepositoryIndexException e )
 110         {
 111             //expected
 112         }
 113
 114         try
 115         {
 116             indexer.optimize();
 117             fail( "Must throw exception on optimize index with closed index." );
 118         }
 119         catch ( RepositoryIndexException e )
 120         {
 121             //expected
 122         }
 123
 124         indexer.open( indexPath );
 125
 126         try
 127         {
 128             indexer.index( "should fail" );
 129             fail( "Must throw exception on add non-Artifact object." );
 130         }
 131         catch ( RepositoryIndexException e )
 132         {
 133             //expected
 134         }
 135
 136         indexer.close();
 137     }
 138
 139     public void testIndex()
 140         throws Exception
 141     {
 142         //indexer = (ArtifactRepositoryIndex) factory.getArtifactRepositoryIndexer( indexPath, repository );
 143         ArtifactRepositoryIndex indexer = (ArtifactRepositoryIndex) lookup( RepositoryIndex.ROLE, "artifact" );
 144         indexer.open( indexPath );
 145
 146         Artifact artifact = getArtifact( "org.apache.maven", "maven-artifact", "2.0.1" );
 147         artifact.setFile( new File( repository.getBasedir(), repository.pathOf( artifact ) ) );
 148         indexer.indexArtifact( artifact );
 149
 150         artifact = getArtifact( "org.apache.maven", "maven-model", "2.0" );
 151         artifact.setFile( new File( repository.getBasedir(), repository.pathOf( artifact ) ) );
 152         indexer.indexArtifact( artifact );
 153
 154         indexer.optimize();
 155         indexer.close();
 156
 157         indexer.open( indexPath );
 158         artifact = getArtifact( "test", "test-artifactId", "1.0" );
 159         artifact.setFile( new File( repository.getBasedir(), repository.pathOf( artifact ) ) );
 160         indexer.index( artifact );
 161         indexer.close();
 162
 163         // TODO: assert something!
 164     }
 165
 166     public void testSearch()
 167         throws Exception
 168     {
 169         ArtifactRepositoryIndex indexer = (ArtifactRepositoryIndex) lookup( RepositoryIndex.ROLE, "artifact" );
 170         indexer.open( getTestPath( "src/test/index" ) );
 171
 172         RepositoryIndexSearcher repoSearcher =
 173             (RepositoryIndexSearcher) lookup( RepositoryIndexSearcher.ROLE, "artifact" );
 174
 175         List artifacts = repoSearcher.search( indexer, "test", GROUPID );
 176         assertEquals( 1, artifacts.size() );
 177
 178         artifacts = repoSearcher.search( indexer, "test", ARTIFACTID );
 179         assertEquals( 1, artifacts.size() );
 180
 181         artifacts = repoSearcher.search( indexer, "1.0", VERSION );
 182         assertEquals( 1, artifacts.size() );
 183
 184         artifacts = repoSearcher.search( indexer, "App", CLASSES );
 185         assertEquals( 1, artifacts.size() );
 186
 187         artifacts = repoSearcher.search( indexer, "groupId", PACKAGES );
 188         assertEquals( 1, artifacts.size() );
 189
 190         artifacts = repoSearcher.search( indexer, "pom.xml", FILES );
 191         assertEquals( 3, artifacts.size() );
 192
 193         artifacts = repoSearcher.search( indexer, "org.apache.maven", GROUPID );
 194         assertEquals( 2, artifacts.size() );
 195
 196         artifacts = repoSearcher.search( indexer, "maven-artifact", ARTIFACTID );
 197         assertEquals( 1, artifacts.size() );
 198
 199         artifacts = repoSearcher.search( indexer, "2", VERSION );
 200         assertEquals( 2, artifacts.size() );
 201     }
 202
 203     private Artifact getArtifact( String groupId, String artifactId, String version )
 204         throws Exception
 205     {
 206         if ( artifactFactory == null )
 207         {
 208             artifactFactory = (ArtifactFactory) lookup( ArtifactFactory.ROLE );
 209         }
 210
 211         return artifactFactory.createBuildArtifact( groupId, artifactId, version, "jar" );
 212     }
 213 }