1   /*
2    * $HeadURL: https://svn.apache.org/repos/asf/httpcomponents/oac.hc3x/trunk/src/test/org/apache/commons/httpclient/ProxyTestDecorator.java $
3    * $Revision: 1425331 $
4    * $Date: 2012-12-22 18:29:41 +0000 (Sat, 22 Dec 2012) $
5    *
6    * ====================================================================
7    *
8    *  Licensed to the Apache Software Foundation (ASF) under one or more
9    *  contributor license agreements.  See the NOTICE file distributed with
10   *  this work for additional information regarding copyright ownership.
11   *  The ASF licenses this file to You under the Apache License, Version 2.0
12   *  (the "License"); you may not use this file except in compliance with
13   *  the License.  You may obtain a copy of the License at
14   *
15   *      http://www.apache.org/licenses/LICENSE-2.0
16   *
17   *  Unless required by applicable law or agreed to in writing, software
18   *  distributed under the License is distributed on an "AS IS" BASIS,
19   *  WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
20   *  See the License for the specific language governing permissions and
21   *  limitations under the License.
22   * ====================================================================
23   *
24   * This software consists of voluntary contributions made by many
25   * individuals on behalf of the Apache Software Foundation.  For more
26   * information on the Apache Software Foundation, please see
27   * <http://www.apache.org/>.
28   *
29   */
30  package org.apache.commons.httpclient;
31  
32  import java.util.Enumeration;
33  
34  import junit.extensions.TestSetup;
35  import junit.framework.Test;
36  import junit.framework.TestSuite;
37  
38  
39  /***
40   * A TestDecorator that configures instances of HttpClientTestBase to use
41   * a proxy server.
42   */
43  public class ProxyTestDecorator extends TestSetup {
44  
45      /***
46       * Iterates through all test cases included in the suite and adds
47       * copies of them modified to use a proxy server.
48       * @param suite
49       */
50      public static void addTests(TestSuite suite) {
51          TestSuite ts2 = new TestSuite();
52          addTest(ts2, suite);
53          suite.addTest(ts2);        
54      }
55      
56      private static void addTest(TestSuite suite, Test t) {
57          if (t instanceof HttpClientTestBase) {
58              suite.addTest(new ProxyTestDecorator((HttpClientTestBase) t));
59          } else if (t instanceof TestSuite) {
60              Enumeration en = ((TestSuite) t).tests();
61              while (en.hasMoreElements()) {
62                  addTest(suite, (Test) en.nextElement());
63              }
64          }
65      }
66      
67      public ProxyTestDecorator(HttpClientTestBase test) {
68          super(test);
69      }
70              
71      protected void setUp() throws Exception {
72          HttpClientTestBase base = (HttpClientTestBase) fTest;
73          base.setUseProxy(true);
74      }        
75  }