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

   1 package org.apache.maven.archiva.webdav;
   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.maven.archiva.webdav.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 void assertResponseInternalServerError( WebResponse response )
 101     {
 102         assertNotNull( "Should have recieved a response", response );
 103         Assert.assertEquals( "Should have been an 500/Internal Server Error response code.", HttpServletResponse.SC_INTERNAL_SERVER_ERROR, response
 104             .getResponseCode() );
 105     }
 106
 107     protected ManagedRepositoryConfiguration createManagedRepository( String id, String name, File location )
 108     {
 109         ManagedRepositoryConfiguration repo = new ManagedRepositoryConfiguration();
 110         repo.setId( id );
 111         repo.setName( name );
 112         repo.setLocation( location.getAbsolutePath() );
 113         return repo;
 114     }
 115
 116     protected RemoteRepositoryConfiguration createRemoteRepository( String id, String name, String url )
 117     {
 118         RemoteRepositoryConfiguration repo = new RemoteRepositoryConfiguration();
 119         repo.setId( id );
 120         repo.setName( name );
 121         repo.setUrl( url );
 122         return repo;
 123     }
 124
 125     protected void dumpResponse( WebResponse response )
 126     {
 127         System.out.println( "---(response)---" );
 128         System.out.println( "" + response.getResponseCode() + " " + response.getResponseMessage() );
 129
 130         String headerNames[] = response.getHeaderFieldNames();
 131         for ( String headerName : headerNames )
 132         {
 133             System.out.println( "[header] " + headerName + ": " + response.getHeaderField( headerName ) );
 134         }
 135
 136         System.out.println( "---(text)---" );
 137         try
 138         {
 139             System.out.println( response.getText() );
 140         }
 141         catch ( IOException e )
 142         {
 143             System.err.print( "[Exception] : " );
 144             e.printStackTrace( System.err );
 145         }
 146     }
 147
 148     protected void saveConfiguration( ArchivaConfiguration archivaConfiguration )
 149         throws Exception
 150     {
 151         archivaConfiguration.save( archivaConfiguration.getConfiguration() );
 152     }
 153
 154     protected void setUp()
 155         throws Exception
 156     {
 157         super.setUp();
 158
 159         String appserverBase = getTestFile( "target/appserver-base" ).getAbsolutePath();
 160         System.setProperty( "appserver.base", appserverBase );
 161
 162         File testConf = getTestFile( "src/test/resources/repository-archiva.xml" );
 163         File testConfDest = new File( appserverBase, "conf/archiva.xml" );
 164         FileUtils.copyFile( testConf, testConfDest );
 165
 166         archivaConfiguration = (ArchivaConfiguration) lookup( ArchivaConfiguration.class );
 167         repoRootInternal = new File( appserverBase, "data/repositories/internal" );
 168         Configuration config = archivaConfiguration.getConfiguration();
 169
 170         config.addManagedRepository( createManagedRepository( REPOID_INTERNAL, "Internal Test Repo", repoRootInternal ) );
 171         saveConfiguration( archivaConfiguration );
 172
 173         CacheManager.getInstance().removeCache( "url-failures-cache" );
 174
 175         HttpUnitOptions.setExceptionsThrownOnErrorStatus( false );
 176
 177         sr = new ServletRunner( getTestFile( "src/test/resources/WEB-INF/web.xml" ) );
 178         sr.registerServlet( "/repository/*", UnauthenticatedRepositoryServlet.class.getName() );
 179         sc = sr.newClient();
 180     }
 181
 182     @Override
 183     protected String getPlexusConfigLocation()
 184     {
 185         return "org/apache/maven/archiva/webdav/RepositoryServletTest.xml";
 186     }
 187
 188     @Override
 189     protected void tearDown()
 190         throws Exception
 191     {
 192         if ( sc != null )
 193         {
 194             sc.clearContents();
 195         }
 196
 197         if ( sr != null )
 198         {
 199             sr.shutDown();
 200         }
 201
 202         if (repoRootInternal.exists())
 203         {
 204             FileUtils.deleteDirectory(repoRootInternal);
 205         }
 206
 207         super.tearDown();
 208     }
 209
 210     protected void setupCleanRepo( File repoRootDir )
 211         throws IOException
 212     {
 213         FileUtils.deleteDirectory( repoRootDir );
 214         if ( !repoRootDir.exists() )
 215         {
 216             repoRootDir.mkdirs();
 217         }
 218     }
 219
 220     protected void assertManagedFileNotExists( File repoRootInternal, String resourcePath )
 221     {
 222         File repoFile = new File( repoRootInternal, resourcePath );
 223         assertFalse( "Managed Repository File <" + repoFile.getAbsolutePath() + "> should not exist.", repoFile
 224             .exists() );
 225     }
 226
 227     protected void setupCleanInternalRepo()
 228         throws Exception
 229     {
 230         setupCleanRepo( repoRootInternal );
 231     }
 232
 233     protected File populateRepo( File repoRootManaged, String path, String contents )
 234         throws Exception
 235     {
 236         File destFile = new File( repoRootManaged, path );
 237         destFile.getParentFile().mkdirs();
 238         FileUtils.writeStringToFile( destFile, contents, null );
 239         return destFile;
 240     }
 241 }