View Javadoc

1   // Copyright (C) 2008 - 2009 Philip Aston
2   // All rights reserved.
3   //
4   // This file is part of The Grinder software distribution. Refer to
5   // the file LICENSE which is part of The Grinder distribution for
6   // licensing details. The Grinder distribution is available on the
7   // Internet at http://grinder.sourceforge.net/
8   //
9   // THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS
10  // "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT
11  // LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS
12  // FOR A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE
13  // COPYRIGHT HOLDERS OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT,
14  // INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES
15  // (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR
16  // SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
17  // HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT,
18  // STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE)
19  // ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED
20  // OF THE POSSIBILITY OF SUCH DAMAGE.
21  
22  package net.grinder.plugin.http;
23  
24  import net.grinder.testutility.RandomStubFactory;
25  import net.grinder.util.Sleeper;
26  import HTTPClient.HTTPConnection.BandwidthLimiter;
27  import junit.framework.TestCase;
28  
29  
30  /**
31   * Unit tests for {@link SlowClientBandwidthLimiterFactory}.
32   *
33   * @author Philip Aston
34   */
35  public class TestSlowClientBandwidthLimiterFactory extends TestCase {
36  
37    public void testFactory() throws Exception {
38      final SlowClientBandwidthLimiterFactory slowClientBandwidthLimiterFactory =
39        new SlowClientBandwidthLimiterFactory(null, 1000);
40  
41      final BandwidthLimiter slowClientBandwidthLimiter =
42        slowClientBandwidthLimiterFactory.create();
43  
44      assertNotNull(slowClientBandwidthLimiter);
45      assertNotSame(slowClientBandwidthLimiter,
46                    slowClientBandwidthLimiterFactory.create());
47    }
48  
49    public void testSlowClientBandwidthLimiter() throws Exception {
50      final RandomStubFactory<Sleeper> sleeperStubFactory =
51        RandomStubFactory.create(Sleeper.class);
52      final SlowClientBandwidthLimiterFactory slowClientBandwidthLimiterFactory =
53        new SlowClientBandwidthLimiterFactory(sleeperStubFactory.getStub(), 1000);
54  
55      final BandwidthLimiter slowClientBandwidthLimiter =
56        slowClientBandwidthLimiterFactory.create();
57  
58      sleeperStubFactory.setResult("getTimeInMilliseconds", new Long(10000));
59  
60      assertEquals(100, slowClientBandwidthLimiter.maximumBytes(0));
61      sleeperStubFactory.assertSuccess("getTimeInMilliseconds");
62      sleeperStubFactory.assertSuccess("sleepNormal", new Long(0), new Long(0));
63      sleeperStubFactory.assertNoMoreCalls();
64  
65      sleeperStubFactory.setResult("getTimeInMilliseconds", new Long(10010));
66      assertEquals(100, slowClientBandwidthLimiter.maximumBytes(100));
67      sleeperStubFactory.assertSuccess("getTimeInMilliseconds");
68      sleeperStubFactory.assertSuccess(
69        "sleepNormal", new Long(1580), new Long(0));
70      sleeperStubFactory.assertNoMoreCalls();
71  
72      sleeperStubFactory.setResult("getTimeInMilliseconds", new Long(12000));
73      assertEquals(100, slowClientBandwidthLimiter.maximumBytes(200));
74      sleeperStubFactory.assertSuccess("getTimeInMilliseconds");
75      sleeperStubFactory.assertSuccess(
76        "sleepNormal", new Long(1380), new Long(0));
77      sleeperStubFactory.assertNoMoreCalls();
78  
79      sleeperStubFactory.setResult("getTimeInMilliseconds", new Long(20000));
80      assertEquals(100, slowClientBandwidthLimiter.maximumBytes(200));
81      sleeperStubFactory.assertSuccess("getTimeInMilliseconds");
82      sleeperStubFactory.assertSuccess("sleepNormal", new Long(0), new Long(0));
83      sleeperStubFactory.assertNoMoreCalls();
84    }
85  }