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

   1 package org.apache.archiva.policies;
   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 junit.framework.TestCase;
  23 import org.apache.archiva.common.filelock.DefaultFileLockManager;
  24 import org.apache.archiva.policies.urlcache.UrlFailureCache;
  25 import org.apache.archiva.repository.storage.FilesystemStorage;
  26 import org.apache.archiva.repository.storage.StorageAsset;
  27 import org.apache.archiva.test.utils.ArchivaSpringJUnit4ClassRunner;
  28 import org.junit.Test;
  29 import org.junit.runner.RunWith;
  30 import org.springframework.test.context.ContextConfiguration;
  31
  32 import javax.inject.Inject;
  33 import javax.inject.Named;
  34 import java.io.IOException;
  35 import java.nio.file.Path;
  36 import java.nio.file.Paths;
  37 import java.util.Properties;
  38
  39 /**
  40  * CachedFailuresPolicyTest
  41  *
  42  *
  43  */
  44 @RunWith( ArchivaSpringJUnit4ClassRunner.class )
  45 @ContextConfiguration( locations = { "classpath*:/META-INF/spring-context.xml", "classpath*:/spring-context.xml" } )
  46 public class CachedFailuresPolicyTest
  47     extends TestCase
  48 {
  49
  50
  51     @Inject
  52     private UrlFailureCache urlFailureCache;
  53
  54     private FilesystemStorage filesystemStorage;
  55
  56     @Inject
  57     @Named( value = "preDownloadPolicy#cache-failures" )
  58     DownloadPolicy downloadPolicy;
  59
  60     private DownloadPolicy lookupPolicy()
  61         throws Exception
  62     {
  63         return downloadPolicy;
  64     }
  65
  66     private StorageAsset getFile() throws IOException {
  67         if (filesystemStorage==null) {
  68             filesystemStorage = new FilesystemStorage(Paths.get("target/cache-failures"), new DefaultFileLockManager());
  69         }
  70         return filesystemStorage.getAsset( getName() + ".txt" );
  71     }
  72
  73     private Properties createRequest()
  74     {
  75         Properties request = new Properties();
  76
  77         return request;
  78     }
  79
  80     @Test
  81     public void testPolicyNo()
  82         throws Exception
  83     {
  84         DownloadPolicy policy = lookupPolicy();
  85         StorageAsset localFile = getFile();
  86         Properties request = createRequest();
  87
  88         request.setProperty( "url", "http://a.bad.hostname.maven.org/path/to/resource.txt" );
  89
  90         policy.applyPolicy( CachedFailuresPolicy.NO, request, localFile );
  91     }
  92
  93     @Test( expected = PolicyViolationException.class )
  94     public void testPolicyYes()
  95         throws Exception
  96     {
  97
  98         DownloadPolicy policy = lookupPolicy();
  99         StorageAsset localFile = getFile();
 100         Properties request = createRequest();
 101         // make unique name
 102         String url = "http://a.bad.hostname.maven.org/path/to/resource"+ System.currentTimeMillis() +".txt";
 103
 104         request.setProperty( "url", url );
 105
 106         // should not fail
 107         policy.applyPolicy( CachedFailuresPolicy.YES, request, localFile );
 108         // status Yes Not In cache
 109
 110         // Yes in Cache
 111
 112         urlFailureCache.cacheFailure( url );
 113
 114         request.setProperty( "url", url );
 115
 116         policy.applyPolicy( CachedFailuresPolicy.YES, request, localFile );
 117     }
 118 }