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