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