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

   1 package org.apache.archiva.consumers.core.repository;
   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 org.apache.archiva.admin.model.beans.ManagedRepository;
  23 import org.apache.archiva.metadata.model.ArtifactMetadata;
  24 import org.apache.archiva.metadata.repository.MetadataRepository;
  25 import org.apache.archiva.metadata.repository.RepositorySession;
  26 import org.apache.archiva.metadata.repository.storage.maven2.Maven2RepositoryPathTranslator;
  27 import org.apache.archiva.repository.ManagedRepositoryContent;
  28 import org.apache.archiva.repository.events.RepositoryListener;
  29 import org.apache.archiva.test.utils.ArchivaSpringJUnit4ClassRunner;
  30 import org.apache.commons.io.FileUtils;
  31 import org.apache.commons.lang.StringUtils;
  32 import org.apache.maven.index.NexusIndexer;
  33 import org.apache.maven.index.context.IndexingContext;
  34 import org.easymock.EasyMock;
  35 import org.easymock.IMocksControl;
  36 import org.junit.After;
  37 import org.junit.Before;
  38 import org.junit.runner.RunWith;
  39 import org.slf4j.LoggerFactory;
  40 import org.springframework.context.ApplicationContext;
  41 import org.springframework.test.context.ContextConfiguration;
  42
  43 import javax.inject.Inject;
  44 import java.io.File;
  45 import java.io.IOException;
  46 import java.nio.file.Files;
  47 import java.nio.file.Path;
  48 import java.nio.file.Paths;
  49 import java.util.ArrayList;
  50 import java.util.List;
  51 import java.util.stream.Collectors;
  52
  53 import static org.junit.Assert.assertFalse;
  54 import static org.junit.Assert.assertTrue;
  55 import static org.mockito.Mockito.mock;
  56 import static org.mockito.Mockito.when;
  57
  58 /**
  59  */
  60 @RunWith(ArchivaSpringJUnit4ClassRunner.class)
  61 @ContextConfiguration(locations = { "classpath*:/META-INF/spring-context.xml", "classpath:/spring-context.xml" })
  62 public abstract class AbstractRepositoryPurgeTest
  63 {
  64     public static final String TEST_REPO_ID = "test-repo";
  65
  66     public static final String TEST_REPO_NAME = "Test Repository";
  67
  68     public static final int TEST_RETENTION_COUNT = 2;
  69
  70     public static final int TEST_DAYS_OLDER = 30;
  71
  72     public static final String PATH_TO_BY_DAYS_OLD_ARTIFACT =
  73         "org/apache/maven/plugins/maven-install-plugin/2.2-SNAPSHOT/maven-install-plugin-2.2-20061118.060401-2.jar";
  74
  75     public static final String PATH_TO_BY_DAYS_OLD_METADATA_DRIVEN_ARTIFACT =
  76         "org/codehaus/plexus/plexus-utils/1.4.3-SNAPSHOT/plexus-utils-1.4.3-20070113.163208-4.jar";
  77
  78     public static final String PATH_TO_BY_RETENTION_COUNT_ARTIFACT =
  79         "org/jruby/plugins/jruby-rake-plugin/1.0RC1-SNAPSHOT/jruby-rake-plugin-1.0RC1-20070504.153317-1.jar";
  80
  81     public static final String PATH_TO_BY_RETENTION_COUNT_POM =
  82         "org/codehaus/castor/castor-anttasks/1.1.2-SNAPSHOT/castor-anttasks-1.1.2-20070506.163513-2.pom";
  83
  84     public static final String PATH_TO_TEST_ORDER_OF_DELETION =
  85         "org/apache/maven/plugins/maven-assembly-plugin/1.1.2-SNAPSHOT/maven-assembly-plugin-1.1.2-20070615.105019-3.jar";
  86
  87     protected static final String RELEASES_TEST_REPO_ID = "releases-test-repo-one";
  88
  89     protected static final String RELEASES_TEST_REPO_NAME = "Releases Test Repo One";
  90
  91     private ManagedRepository config;
  92
  93     private ManagedRepositoryContent repo;
  94
  95     protected RepositoryPurge repoPurge;
  96
  97     protected IMocksControl listenerControl;
  98
  99     protected RepositoryListener listener;
 100
 101     protected RepositorySession repositorySession;
 102
 103     protected MetadataRepository metadataRepository;
 104
 105     @Inject
 106     protected ApplicationContext applicationContext;
 107
 108     @Inject
 109     protected PlexusSisuBridge plexusSisuBridge;
 110
 111
 112     @Before
 113     public void setUp()
 114         throws Exception
 115     {
 116
 117         removeMavenIndexes();
 118
 119         listenerControl = EasyMock.createControl();
 120
 121         listener = listenerControl.createMock( RepositoryListener.class );
 122
 123         repositorySession = mock( RepositorySession.class );
 124         metadataRepository = mock( MetadataRepository.class );
 125         when( repositorySession.getRepository() ).thenReturn( metadataRepository );
 126
 127
 128     }
 129
 130     @After
 131     public void tearDown()
 132         throws Exception
 133     {
 134         removeMavenIndexes();
 135         config = null;
 136         repo = null;
 137
 138     }
 139
 140     protected void removeMavenIndexes()
 141         throws Exception
 142     {
 143         NexusIndexer nexusIndexer = plexusSisuBridge.lookup( NexusIndexer.class );
 144         for ( IndexingContext indexingContext : nexusIndexer.getIndexingContexts().values() )
 145         {
 146             nexusIndexer.removeIndexingContext( indexingContext, false );
 147         }
 148     }
 149
 150     protected static String fixPath( String path )
 151     {
 152         if ( path.contains( " " ) )
 153         {
 154             LoggerFactory.getLogger( AbstractRepositoryPurgeTest.class.getName() ).error(
 155                 "You are building and testing with a path: \n " + path + " containing space. Consider relocating." );
 156             return path.replaceAll( " ", "&amp;20" );
 157         }
 158         return path;
 159     }
 160
 161     public ManagedRepository getRepoConfiguration( String repoId, String repoName )
 162     {
 163         config = new ManagedRepository();
 164         config.setId( repoId );
 165         config.setName( repoName );
 166         config.setDaysOlder( TEST_DAYS_OLDER );
 167         String path = AbstractRepositoryPurgeTest.fixPath(
 168             new File( "target/test-" + getName() + "/" + repoId ).getAbsolutePath() );
 169         config.setLocation( path );
 170         config.setReleases( true );
 171         config.setSnapshots( true );
 172         config.setDeleteReleasedSnapshots( true );
 173         config.setRetentionCount( TEST_RETENTION_COUNT );
 174
 175         return config;
 176     }
 177
 178     public ManagedRepositoryContent getRepository()
 179         throws Exception
 180     {
 181         if ( repo == null )
 182         {
 183             repo = applicationContext.getBean( "managedRepositoryContent#default", ManagedRepositoryContent.class );
 184             repo.setRepository( getRepoConfiguration( TEST_REPO_ID, TEST_REPO_NAME ) );
 185         }
 186
 187         return repo;
 188     }
 189
 190     protected void assertDeleted( String path )
 191     {
 192         assertFalse( "File should have been deleted: " + path, new File( path ).exists() );
 193     }
 194
 195     protected void assertExists( String path )
 196     {
 197         assertTrue( "File should exist: " + path, new File( path ).exists() );
 198     }
 199
 200     protected File getTestRepoRoot()
 201     {
 202         return new File( "target/test-" + getName() + "/" + TEST_REPO_ID );
 203     }
 204
 205     protected Path getTestRepoRootPath() {
 206         return Paths.get("target/test-"+getName()+"/"+TEST_REPO_ID);
 207     }
 208
 209     protected String prepareTestRepos()
 210         throws Exception
 211     {
 212         removeMavenIndexes();
 213         File testDir = new File( AbstractRepositoryPurgeTest.fixPath( getTestRepoRoot().getAbsolutePath() ) );
 214         FileUtils.deleteDirectory( testDir );
 215         File sourceDir = new File( new File( "target/test-classes/" + TEST_REPO_ID ).getAbsolutePath() );
 216         FileUtils.copyDirectory( sourceDir, testDir );
 217
 218         File releasesTestDir = new File( AbstractRepositoryPurgeTest.fixPath(
 219             new File( "target/test-" + getName() + "/" + RELEASES_TEST_REPO_ID ).getAbsolutePath() ) );
 220
 221         FileUtils.deleteDirectory( releasesTestDir );
 222         File sourceReleasesDir =
 223             new File( new File( "target/test-classes/" + RELEASES_TEST_REPO_ID ).getAbsolutePath() );
 224         FileUtils.copyDirectory( sourceReleasesDir, releasesTestDir );
 225
 226         return AbstractRepositoryPurgeTest.fixPath( testDir.getAbsolutePath() );
 227     }
 228
 229     public String getName()
 230     {
 231         return StringUtils.substringAfterLast( getClass().getName(), "." );
 232     }
 233
 234     protected List<ArtifactMetadata> getArtifactMetadataFromDir( String repoId, String projectName, Path repoDir, Path vDir ) throws IOException
 235     {
 236         Maven2RepositoryPathTranslator translator = new Maven2RepositoryPathTranslator( new ArrayList<>(  ) );
 237         return Files.find(vDir, 1,
 238                     (path, basicFileAttributes) -> basicFileAttributes.isRegularFile() && path.getFileName().toString().startsWith(projectName))
 239             .map( path ->
 240                                 translator.getArtifactForPath( repoId, repoDir.relativize( path ).toString() )
 241             ).collect( Collectors.toList());
 242     }
 243 }