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

   1 package org.apache.maven.archiva.web.rss;
   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 javax.servlet.http.HttpServletResponse;
  23
  24 import org.apache.commons.codec.Encoder;
  25 import org.apache.commons.codec.binary.Base64;
  26 import org.codehaus.plexus.spring.PlexusInSpringTestCase;
  27
  28 import sun.misc.BASE64Encoder;
  29
  30 import com.meterware.httpunit.GetMethodWebRequest;
  31 import com.meterware.httpunit.HttpException;
  32 import com.meterware.httpunit.WebRequest;
  33 import com.meterware.httpunit.WebResponse;
  34 import com.meterware.servletunit.ServletRunner;
  35 import com.meterware.servletunit.ServletUnitClient;
  36
  37 /**
  38  *
  39  * @version
  40  */
  41 public class RssFeedServletTest
  42     extends PlexusInSpringTestCase
  43 {
  44     private ServletRunner sr;
  45
  46     private ServletUnitClient client;
  47
  48     public void setUp()
  49         throws Exception
  50     {
  51         sr = new ServletRunner( getTestFile( "src/test/webapp/WEB-INF/feedServletTest-web.xml" ) );
  52         client = sr.newClient();
  53     }
  54
  55     public void testRetrieveServlet()
  56         throws Exception
  57     {
  58         RssFeedServlet servlet =
  59             (RssFeedServlet) client.newInvocation( "http://localhost/feeds/test-repo" ).getServlet();
  60         assertNotNull( servlet );
  61     }
  62
  63     public void testRequestNewArtifactsInRepo()
  64         throws Exception
  65     {
  66         RssFeedServlet servlet =
  67             (RssFeedServlet) client.newInvocation( "http://localhost/feeds/test-repo" ).getServlet();
  68         assertNotNull( servlet );
  69
  70         WebRequest request = new GetMethodWebRequest( "http://localhost/feeds/test-repo" );
  71
  72         BASE64Encoder encoder = new BASE64Encoder();
  73         String userPass = "user1:password1";
  74         String encodedUserPass = encoder.encode( userPass.getBytes() );
  75         request.setHeaderField( "Authorization", "BASIC " + encodedUserPass );
  76
  77         WebResponse response = client.getResponse( request );
  78         assertEquals( RssFeedServlet.MIME_TYPE, response.getHeaderField( "CONTENT-TYPE" ) );
  79         assertNotNull( "Should have recieved a response", response );
  80         assertEquals( "Should have been an OK response code.", HttpServletResponse.SC_OK, response.getResponseCode() );
  81     }
  82
  83     public void testRequestNewVersionsOfArtifact()
  84         throws Exception
  85     {
  86         RssFeedServlet servlet =
  87             (RssFeedServlet) client.newInvocation(
  88                                                    "http://localhost/feeds/org/apache/archiva/artifact-two" ).getServlet();
  89         assertNotNull( servlet );
  90
  91         WebRequest request = new GetMethodWebRequest( "http://localhost/feeds/org/apache/archiva/artifact-two" );
  92
  93         BASE64Encoder encoder = new BASE64Encoder();
  94         String userPass = "user1:password1";
  95         String encodedUserPass = encoder.encode( userPass.getBytes() );
  96         request.setHeaderField( "Authorization", "BASIC " + encodedUserPass );
  97
  98         WebResponse response = client.getResponse( request );
  99         assertEquals( RssFeedServlet.MIME_TYPE, response.getHeaderField( "CONTENT-TYPE" ) );
 100         assertNotNull( "Should have recieved a response", response );
 101         assertEquals( "Should have been an OK response code.", HttpServletResponse.SC_OK, response.getResponseCode() );
 102     }
 103
 104     public void testInvalidRequest()
 105         throws Exception
 106     {
 107         RssFeedServlet servlet =
 108             (RssFeedServlet) client.newInvocation(
 109                                                    "http://localhost/feeds?invalid_param=xxx" ).getServlet();
 110         assertNotNull( servlet );
 111
 112         try
 113         {
 114             client.getResponse( "http://localhost/feeds?invalid_param=xxx" );
 115             fail( "Expected exception" );
 116         }
 117         catch ( HttpException he )
 118         {
 119             assertEquals( "Should have been a bad request response code.", HttpServletResponse.SC_BAD_REQUEST, he.getResponseCode() );
 120         }
 121     }
 122
 123     public void testInvalidAuthenticationRequest()
 124         throws Exception
 125     {
 126         RssFeedServlet servlet =
 127             (RssFeedServlet) client.newInvocation(
 128                                                    "http://localhost/feeds/unauthorized-repo" ).getServlet();
 129         assertNotNull( servlet );
 130
 131
 132         WebRequest request = new GetMethodWebRequest( "http://localhost/feeds/unauthorized-repo" );
 133
 134         Encoder encoder = new Base64();
 135         String userPass = "unauthUser:unauthPass";
 136         String encodedUserPass = new String( ( byte[] ) encoder.encode( userPass.getBytes() ) );
 137         request.setHeaderField( "Authorization", "BASIC " + encodedUserPass );
 138
 139         try
 140         {
 141             client.getResponse( request );
 142             fail( "Expected exception" );
 143         }
 144         catch ( HttpException he )
 145         {
 146             assertEquals( "Should have been a unauthorized response.", HttpServletResponse.SC_UNAUTHORIZED, he.getResponseCode() );
 147         }
 148     }
 149
 150     public void testUnauthorizedRequest()
 151         throws Exception
 152     {
 153         RssFeedServlet servlet =
 154             (RssFeedServlet) client.newInvocation(
 155                                                    "http://localhost/feeds/unauthorized-repo" ).getServlet();
 156         assertNotNull( servlet );
 157
 158
 159         WebRequest request = new GetMethodWebRequest( "http://localhost/feeds/unauthorized-repo" );
 160
 161         BASE64Encoder encoder = new BASE64Encoder();
 162         String userPass = "user1:password1";
 163         String encodedUserPass = encoder.encode( userPass.getBytes() );
 164         request.setHeaderField( "Authorization", "BASIC " + encodedUserPass );
 165
 166         try
 167         {
 168             client.getResponse( request );
 169             fail( "Expected exception" );
 170         }
 171         catch ( HttpException he )
 172         {
 173             assertEquals( "Should have been a unauthorized response.", HttpServletResponse.SC_UNAUTHORIZED, he.getResponseCode() );
 174         }
 175     }
 176
 177     @Override
 178     protected String getPlexusConfigLocation()
 179     {
 180         return "org/apache/maven/archiva/web/rss/RssFeedServletTest.xml";
 181     }
 182
 183     @Override
 184     protected void tearDown()
 185         throws Exception
 186     {
 187         if ( client != null )
 188         {
 189             client.clearContents();
 190         }
 191
 192         if ( sr != null )
 193         {
 194             sr.shutDown();
 195         }
 196
 197         super.tearDown();
 198     }
 199
 200 }