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

   1 package org.apache.archiva.web.servlet;
   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 com.meterware.httpunit.WebResponse;
  23 import com.meterware.httpunit.HttpUnitOptions;
  24 import com.meterware.servletunit.ServletRunner;
  25 import com.meterware.servletunit.ServletUnitClient;
  26 import net.sf.ehcache.CacheManager;
  27 import org.apache.commons.io.FileUtils;
  28 import org.apache.maven.archiva.configuration.ArchivaConfiguration;
  29 import org.apache.maven.archiva.configuration.Configuration;
  30 import org.apache.maven.archiva.configuration.ManagedRepositoryConfiguration;
  31 import org.apache.maven.archiva.configuration.RemoteRepositoryConfiguration;
  32 import org.apache.archiva.web.servlet.RepositoryServlet;
  33 import org.codehaus.plexus.spring.PlexusInSpringTestCase;
  34
  35 import javax.servlet.http.HttpServletResponse;
  36 import java.io.File;
  37 import java.io.IOException;
  38
  39 import junit.framework.Assert;
  40
  41 /**
  42  * AbstractRepositoryServletTestCase
  43  *
  44  * @version $Id$
  45  */
  46 public abstract class AbstractRepositoryServletTestCase
  47     extends PlexusInSpringTestCase
  48 {
  49     protected static final String REPOID_INTERNAL = "internal";
  50
  51     protected ServletUnitClient sc;
  52
  53     protected File repoRootInternal;
  54
  55     private ServletRunner sr;
  56
  57     protected ArchivaConfiguration archivaConfiguration;
  58
  59     protected void saveConfiguration()
  60         throws Exception
  61     {
  62         saveConfiguration( archivaConfiguration );
  63     }
  64
  65     protected void assertFileContents( String expectedContents, File repoRoot, String path )
  66         throws IOException
  67     {
  68         File actualFile = new File( repoRoot, path );
  69         assertTrue( "File <" + actualFile.getAbsolutePath() + "> should exist.", actualFile.exists() );
  70         assertTrue( "File <" + actualFile.getAbsolutePath() + "> should be a file (not a dir/link/device/etc).",
  71                     actualFile.isFile() );
  72
  73         String actualContents = FileUtils.readFileToString( actualFile, null );
  74         assertEquals( "File Contents of <" + actualFile.getAbsolutePath() + ">", expectedContents, actualContents );
  75     }
  76
  77     protected void assertRepositoryValid( RepositoryServlet servlet, String repoId )
  78     {
  79         ManagedRepositoryConfiguration repository = servlet.getRepository( repoId );
  80         assertNotNull( "Archiva Managed Repository id:<" + repoId + "> should exist.", repository );
  81         File repoRoot = new File( repository.getLocation() );
  82         assertTrue( "Archiva Managed Repository id:<" + repoId + "> should have a valid location on disk.", repoRoot
  83             .exists()
  84             && repoRoot.isDirectory() );
  85     }
  86
  87     protected void assertResponseOK( WebResponse response )
  88     {
  89         assertNotNull( "Should have recieved a response", response );
  90         Assert.assertEquals( "Should have been an OK response code.", HttpServletResponse.SC_OK, response.getResponseCode() );
  91     }
  92
  93     protected void assertResponseNotFound( WebResponse response )
  94     {
  95         assertNotNull( "Should have recieved a response", response );
  96         Assert.assertEquals( "Should have been an 404/Not Found response code.", HttpServletResponse.SC_NOT_FOUND, response
  97             .getResponseCode() );
  98     }
  99
 100     protected ManagedRepositoryConfiguration createManagedRepository( String id, String name, File location )
 101     {
 102         ManagedRepositoryConfiguration repo = new ManagedRepositoryConfiguration();
 103         repo.setId( id );
 104         repo.setName( name );
 105         repo.setLocation( location.getAbsolutePath() );
 106         return repo;
 107     }
 108
 109     protected RemoteRepositoryConfiguration createRemoteRepository( String id, String name, String url )
 110     {
 111         RemoteRepositoryConfiguration repo = new RemoteRepositoryConfiguration();
 112         repo.setId( id );
 113         repo.setName( name );
 114         repo.setUrl( url );
 115         return repo;
 116     }
 117
 118     protected void dumpResponse( WebResponse response )
 119     {
 120         System.out.println( "---(response)---" );
 121         System.out.println( "" + response.getResponseCode() + " " + response.getResponseMessage() );
 122
 123         String headerNames[] = response.getHeaderFieldNames();
 124         for ( String headerName : headerNames )
 125         {
 126             System.out.println( "[header] " + headerName + ": " + response.getHeaderField( headerName ) );
 127         }
 128
 129         System.out.println( "---(text)---" );
 130         try
 131         {
 132             System.out.println( response.getText() );
 133         }
 134         catch ( IOException e )
 135         {
 136             System.err.print( "[Exception] : " );
 137             e.printStackTrace( System.err );
 138         }
 139     }
 140
 141     protected void saveConfiguration( ArchivaConfiguration archivaConfiguration )
 142         throws Exception
 143     {
 144         archivaConfiguration.save( archivaConfiguration.getConfiguration() );
 145     }
 146
 147     protected void setUp()
 148         throws Exception
 149     {
 150         super.setUp();
 151
 152         String appserverBase = getTestFile( "target/appserver-base" ).getAbsolutePath();
 153         System.setProperty( "appserver.base", appserverBase );
 154
 155         File testConf = getTestFile( "src/test/resources/repository-archiva.xml" );
 156         File testConfDest = new File( appserverBase, "conf/archiva.xml" );
 157         FileUtils.copyFile( testConf, testConfDest );
 158
 159         archivaConfiguration = (ArchivaConfiguration) lookup( ArchivaConfiguration.class );
 160         repoRootInternal = new File( appserverBase, "data/repositories/internal" );
 161         Configuration config = archivaConfiguration.getConfiguration();
 162
 163         config.addManagedRepository( createManagedRepository( REPOID_INTERNAL, "Internal Test Repo", repoRootInternal ) );
 164         saveConfiguration( archivaConfiguration );
 165
 166         CacheManager.getInstance().removeCache( "url-failures-cache" );
 167
 168         HttpUnitOptions.setExceptionsThrownOnErrorStatus( false );
 169
 170         sr = new ServletRunner( getTestFile( "src/test/resources/WEB-INF/web.xml" ) );
 171         sr.registerServlet( "/repository/*", RepositoryServlet.class.getName() );
 172         sc = sr.newClient();
 173     }
 174
 175     @Override
 176     protected String getPlexusConfigLocation()
 177     {
 178         return "org/apache/maven/archiva/webdav/RepositoryServletTest.xml";
 179     }
 180
 181     @Override
 182     protected void tearDown()
 183         throws Exception
 184     {
 185         if ( sc != null )
 186         {
 187             sc.clearContents();
 188         }
 189
 190         if ( sr != null )
 191         {
 192             sr.shutDown();
 193         }
 194
 195         if (repoRootInternal.exists())
 196         {
 197             FileUtils.deleteDirectory(repoRootInternal);
 198         }
 199
 200         super.tearDown();
 201     }
 202
 203     protected void setupCleanRepo( File repoRootDir )
 204         throws IOException
 205     {
 206         FileUtils.deleteDirectory( repoRootDir );
 207         if ( !repoRootDir.exists() )
 208         {
 209             repoRootDir.mkdirs();
 210         }
 211     }
 212
 213     protected void assertManagedFileNotExists( File repoRootInternal, String resourcePath )
 214     {
 215         File repoFile = new File( repoRootInternal, resourcePath );
 216         assertFalse( "Managed Repository File <" + repoFile.getAbsolutePath() + "> should not exist.", repoFile
 217             .exists() );
 218     }
 219
 220     protected void setupCleanInternalRepo()
 221         throws Exception
 222     {
 223         setupCleanRepo( repoRootInternal );
 224     }
 225
 226     protected File populateRepo( File repoRootManaged, String path, String contents )
 227         throws Exception
 228     {
 229         File destFile = new File( repoRootManaged, path );
 230         destFile.getParentFile().mkdirs();
 231         FileUtils.writeStringToFile( destFile, contents, null );
 232         return destFile;
 233     }
 234 }