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