1 // Copyright (C) 2005 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.AssertUtilities;
25 import net.grinder.util.html.HTMLElement;
26
27 import junit.framework.TestCase;
28
29
30 /**
31 * Unit test case for {@link HTTPResponse}.
32 *
33 * @author Philip Aston
34 */
35 public class TestHTTPResponse extends TestCase {
36
37 public void testHTTPResponse() throws Exception {
38 final HTTPResponse response = new HTTPResponse();
39
40 final String result0 = response.toString();
41 AssertUtilities.assertStartsWith(result0,
42 "HTTP/1.0 500 Internal Server Error\r\n");
43 AssertUtilities.assertContainsHeader(result0,
44 "Proxy-agent", "The Grinder .*");
45 AssertUtilities.assertEndsWith(result0, "\r\n\r\n");
46
47 response.setStatus("200 OK");
48 final String result1 = response.toString();
49 AssertUtilities.assertStartsWith(result1, "HTTP/1.0 200 OK\r\n");
50
51 final HTMLElement message = new HTMLElement();
52 message.addElement("p").addText("testing");
53 response.setMessage("Test", message);
54
55 final String result2 = response.toString();
56 AssertUtilities.assertContainsHeader(result2, "Content-Type", "text/html");
57 AssertUtilities.assertContainsHeader(result2, "Connection", "close");
58
59 AssertUtilities.assertContainsPattern(result2,
60 "<title>.*Test.*</title>.*<h1>.*Test.*</h1>.*<p>testing</p>");
61 AssertUtilities.assertEndsWith(result0, "\r\n\r\n");
62 }
63
64 }