1 // Copyright (C) 2005 - 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.tools.tcpproxy;
23
24 import net.grinder.testutility.RandomStubFactory;
25
26
27 public class MyFilterStubFactory extends RandomStubFactory<TCPProxyFilter> {
28 private boolean m_resultSet;
29 private byte[] m_result;
30
31 public MyFilterStubFactory() {
32 super(TCPProxyFilter.class);
33 }
34
35 public byte[] override_handle(Object proxy,
36 ConnectionDetails connectionDetails,
37 byte[] originalBuffer,
38 int bytesRead) {
39 return m_resultSet ? m_result : originalBuffer;
40 }
41
42 public void setResult(byte[] result) {
43 m_resultSet = true;
44 m_result = result;
45 }
46
47 public void assertIsWrappedBy(TCPProxyFilter filter)
48 throws Exception {
49
50 assertNoMoreCalls();
51
52 final ConnectionDetails connectionDetails =
53 new ConnectionDetails(new EndPoint("A", 55),
54 new EndPoint("B", 80), false);
55
56 filter.connectionOpened(connectionDetails);
57 assertSuccess("connectionOpened", connectionDetails);
58
59 filter.connectionClosed(connectionDetails);
60 assertSuccess("connectionClosed", connectionDetails);
61
62 filter.connectionOpened(connectionDetails);
63 assertSuccess("connectionOpened", connectionDetails);
64
65 filter.connectionClosed(connectionDetails);
66 assertSuccess("connectionClosed", connectionDetails);
67
68 // We don't test stop() here because our wrapper prevents it from being
69 // called more than once, so we would break future tests.
70
71 assertNoMoreCalls();
72 }
73 }