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.metadata.model.ArtifactMetadata;
  23 import org.apache.archiva.metadata.repository.MetadataRepository;
  24 import org.apache.archiva.metadata.repository.RepositorySession;
  25 import org.apache.archiva.metadata.repository.RepositorySessionFactory;
  26 import org.apache.archiva.repository.ManagedRepositoryContent;
  27 import org.apache.archiva.repository.RepositoryRegistry;
  28 import org.apache.archiva.repository.base.ArchivaRepositoryRegistry;
  29 import org.apache.archiva.repository.maven.metadata.storage.Maven2RepositoryPathTranslator;
  30 import org.apache.archiva.repository.base.BasicManagedRepository;
  31 import org.apache.archiva.repository.ReleaseScheme;
  32 import org.apache.archiva.repository.RepositoryContentProvider;
  33 import org.apache.archiva.metadata.audit.RepositoryListener;
  34 import org.apache.archiva.repository.features.ArtifactCleanupFeature;
  35 import org.apache.archiva.test.utils.ArchivaSpringJUnit4ClassRunner;
  36 import org.apache.commons.io.FileUtils;
  37 import org.apache.commons.lang3.StringUtils;
  38 import org.easymock.EasyMock;
  39 import org.easymock.IMocksControl;
  40 import org.junit.After;
  41 import org.junit.Before;
  42 import org.junit.runner.RunWith;
  43 import org.slf4j.LoggerFactory;
  44 import org.springframework.context.ApplicationContext;
  45 import org.springframework.test.context.ContextConfiguration;
  46
  47 import javax.inject.Inject;
  48 import java.io.IOException;
  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.stream.Collectors;
  57
  58 import static org.junit.Assert.*;
  59 import static org.mockito.Mockito.mock;
  60
  61 /**
  62  */
  63 @RunWith(ArchivaSpringJUnit4ClassRunner.class)
  64 @ContextConfiguration(locations = { "classpath*:/META-INF/spring-context.xml", "classpath:/spring-context.xml" })
  65 public abstract class AbstractRepositoryPurgeTest
  66 {
  67     public static final String TEST_REPO_ID = "test-repo";
  68
  69     public static final String TEST_REPO_NAME = "Test Repository";
  70
  71     public static final int TEST_RETENTION_COUNT = 2;
  72
  73     public static final int TEST_DAYS_OLDER = 30;
  74
  75     public static final String PATH_TO_BY_DAYS_OLD_ARTIFACT =
  76         "org/apache/maven/plugins/maven-install-plugin/2.2-SNAPSHOT/maven-install-plugin-2.2-20061118.060401-2.jar";
  77
  78     public static final String PATH_TO_BY_DAYS_OLD_METADATA_DRIVEN_ARTIFACT =
  79         "org/codehaus/plexus/plexus-utils/1.4.3-SNAPSHOT/plexus-utils-1.4.3-20070113.163208-4.jar";
  80
  81     public static final String PATH_TO_BY_RETENTION_COUNT_ARTIFACT =
  82         "org/jruby/plugins/jruby-rake-plugin/1.0RC1-SNAPSHOT/jruby-rake-plugin-1.0RC1-20070504.153317-1.jar";
  83
  84     public static final String PATH_TO_BY_RETENTION_COUNT_POM =
  85         "org/codehaus/castor/castor-anttasks/1.1.2-SNAPSHOT/castor-anttasks-1.1.2-20070506.163513-2.pom";
  86
  87     public static final String PATH_TO_TEST_ORDER_OF_DELETION =
  88         "org/apache/maven/plugins/maven-assembly-plugin/1.1.2-SNAPSHOT/maven-assembly-plugin-1.1.2-20070615.105019-3.jar";
  89
  90     protected static final String RELEASES_TEST_REPO_ID = "releases-test-repo-one";
  91
  92     protected static final String RELEASES_TEST_REPO_NAME = "Releases Test Repo One";
  93
  94     private BasicManagedRepository config;
  95
  96     private ManagedRepositoryContent repo;
  97
  98     protected RepositoryPurge repoPurge;
  99
 100     protected IMocksControl listenerControl;
 101
 102     protected RepositoryListener listener;
 103
 104     protected IMocksControl sessionControl;
 105     protected RepositorySession repositorySession;
 106
 107     protected IMocksControl sessionFactoryControl;
 108     protected RepositorySessionFactory sessionFactory;
 109
 110     protected MetadataRepository metadataRepository;
 111
 112     protected ArchivaRepositoryRegistry repositoryRegistry;
 113
 114     @Inject
 115     protected ApplicationContext applicationContext;
 116
 117
 118     @Before
 119     public void setUp()
 120         throws Exception
 121     {
 122
 123         listenerControl = EasyMock.createControl();
 124
 125         listener = listenerControl.createMock( RepositoryListener.class );
 126
 127         sessionControl = EasyMock.createControl();
 128         sessionFactoryControl = EasyMock.createControl( );
 129
 130         repositorySession = sessionControl.createMock( RepositorySession.class );
 131         metadataRepository = mock( MetadataRepository.class );
 132         sessionFactory = sessionFactoryControl.createMock( RepositorySessionFactory.class );
 133         repositoryRegistry = applicationContext.getBean( "repositoryRegistry", ArchivaRepositoryRegistry.class );
 134         assertNotNull( repositoryRegistry );
 135         EasyMock.expect( repositorySession.getRepository() ).andStubReturn( metadataRepository );
 136         EasyMock.expect( sessionFactory.createSession( ) ).andStubReturn( repositorySession );
 137
 138     }
 139
 140     @After
 141     public void tearDown()
 142         throws Exception
 143     {
 144         config = null;
 145         repo = null;
 146         repositoryRegistry.destroy();
 147
 148     }
 149
 150
 151     protected static String fixPath( String path )
 152     {
 153         if ( path.contains( " " ) )
 154         {
 155             LoggerFactory.getLogger( AbstractRepositoryPurgeTest.class.getName() ).error(
 156                 "You are building and testing with a path: \n {} containing space. Consider relocating.", path );
 157             return path.replaceAll( " ", "&amp;20" );
 158         }
 159         return path;
 160     }
 161
 162     public org.apache.archiva.repository.ManagedRepository getRepoConfiguration( String repoId, String repoName ) throws URISyntaxException, IOException {
 163         Path basePath = Paths.get("target/test-" + getName()).toAbsolutePath();
 164         config = BasicManagedRepository.newFilesystemInstance( repoId, repoName, basePath.resolve(repoId));
 165         config.addActiveReleaseScheme( ReleaseScheme.RELEASE );
 166         config.addActiveReleaseScheme( ReleaseScheme.SNAPSHOT );
 167         ArtifactCleanupFeature atf = config.getFeature( ArtifactCleanupFeature.class ).get();
 168         atf.setRetentionPeriod( Period.ofDays( TEST_DAYS_OLDER) );
 169         String path = AbstractRepositoryPurgeTest.fixPath(
 170             basePath.resolve( repoId ).toAbsolutePath().toString() );
 171         config.setLocation( Paths.get( path ).toFile().toURI() );
 172         atf.setDeleteReleasedSnapshots( true );
 173         atf.setRetentionCount( TEST_RETENTION_COUNT );
 174
 175         return config;
 176     }
 177
 178     public ManagedRepositoryContent getRepository()
 179         throws Exception
 180     {
 181         if ( repo == null )
 182         {
 183             org.apache.archiva.repository.ManagedRepository repoCfg = getRepoConfiguration( TEST_REPO_ID, TEST_REPO_NAME );
 184             RepositoryContentProvider provider = applicationContext.getBean( "repositoryContentProvider#maven", RepositoryContentProvider.class );
 185             repo = provider.createManagedContent( repoCfg );
 186         }
 187
 188         return repo;
 189     }
 190
 191     protected void assertDeleted( String path )
 192     {
 193         assertFalse( "File should have been deleted: " + path, Files.exists(Paths.get( path )) );
 194     }
 195
 196     protected void assertExists( String path )
 197     {
 198         assertTrue( "File should exist: " + path, Files.exists(Paths.get(path)) );
 199     }
 200
 201     protected Path getTestRepoRoot()
 202     {
 203         return Paths.get( "target/test-" + getName() + "/" + TEST_REPO_ID );
 204     }
 205
 206     protected Path getTestRepoRootPath() {
 207         return Paths.get("target/test-"+getName()+"/"+TEST_REPO_ID);
 208     }
 209
 210     protected String prepareTestRepos()
 211         throws Exception
 212     {
 213         Path testDir = Paths.get( AbstractRepositoryPurgeTest.fixPath( getTestRepoRoot().toAbsolutePath().toString() ) );
 214         org.apache.archiva.common.utils.FileUtils.deleteDirectory( testDir );
 215         Path sourceDir = Paths.get( Paths.get( "target/test-classes/" + TEST_REPO_ID ).toAbsolutePath().toString() );
 216         FileUtils.copyDirectory( sourceDir.toFile(), testDir.toFile() );
 217
 218         Path releasesTestDir = Paths.get( AbstractRepositoryPurgeTest.fixPath(
 219             Paths.get( "target/test-" + getName() + "/" + RELEASES_TEST_REPO_ID ).toAbsolutePath().toString() ) );
 220
 221         org.apache.archiva.common.utils.FileUtils.deleteDirectory( releasesTestDir );
 222         Path sourceReleasesDir =
 223             Paths.get( Paths.get( "target/test-classes/" + RELEASES_TEST_REPO_ID ).toAbsolutePath().toString() );
 224         FileUtils.copyDirectory( sourceReleasesDir.toFile(), releasesTestDir.toFile() );
 225
 226         return AbstractRepositoryPurgeTest.fixPath( testDir.toAbsolutePath().toString() );
 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 }