1package fi.iki.elonen;
2
3/*
4 * #%L
5 * NanoHttpd-Core
6 * %%
7 * Copyright (C) 2012 - 2015 nanohttpd
8 * %%
9 * Redistribution and use in source and binary forms, with or without modification,
10 * are permitted provided that the following conditions are met:
11 *
12 * 1. Redistributions of source code must retain the above copyright notice, this
13 *    list of conditions and the following disclaimer.
14 *
15 * 2. Redistributions in binary form must reproduce the above copyright notice,
16 *    this list of conditions and the following disclaimer in the documentation
17 *    and/or other materials provided with the distribution.
18 *
19 * 3. Neither the name of the nanohttpd nor the names of its contributors
20 *    may be used to endorse or promote products derived from this software without
21 *    specific prior written permission.
22 *
23 * THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS "AS IS" AND
24 * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED
25 * WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE DISCLAIMED.
26 * IN NO EVENT SHALL THE COPYRIGHT HOLDER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT,
27 * INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING,
28 * BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE,
29 * DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF
30 * LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE
31 * OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED
32 * OF THE POSSIBILITY OF SUCH DAMAGE.
33 * #L%
34 */
35
36import static junit.framework.Assert.fail;
37
38import java.io.ByteArrayOutputStream;
39import java.io.PipedInputStream;
40import java.io.PipedOutputStream;
41
42import org.junit.Test;
43
44public class HttpKeepAliveTest extends HttpServerTest {
45
46    private Throwable error = null;
47
48    @Test
49    public void testManyGetRequests() throws Exception {
50        String request = "GET " + HttpServerTest.URI + " HTTP/1.1\r\n\r\n";
51        String[] expected = {
52            "HTTP/1.1 200 OK",
53            "Content-Type: text/html",
54            "Date: .*",
55            "Connection: keep-alive",
56            "Content-Length: 0",
57            ""
58        };
59        testManyRequests(request, expected);
60    }
61
62    @Test
63    public void testManyPutRequests() throws Exception {
64        String data = "BodyData 1\nLine 2";
65        String request = "PUT " + HttpServerTest.URI + " HTTP/1.1\r\nContent-Length: " + data.length() + "\r\n\r\n" + data;
66        String[] expected = {
67            "HTTP/1.1 200 OK",
68            "Content-Type: text/html",
69            "Date: .*",
70            "Connection: keep-alive",
71            "Content-Length: 0",
72            ""
73        };
74        testManyRequests(request, expected);
75    }
76
77    /**
78     * Issue the given request many times to check whether an error occurs. For
79     * this test, a small stack size is used, since a stack overflow is among
80     * the possible errors.
81     *
82     * @param request
83     *            The request to issue
84     * @param expected
85     *            The expected response
86     */
87    public void testManyRequests(final String request, final String[] expected) throws Exception {
88        Runnable r = new Runnable() {
89
90            @Override
91            public void run() {
92                try {
93                    PipedOutputStream requestStream = new PipedOutputStream();
94                    PipedInputStream inputStream = new PipedInputStream(requestStream);
95                    ByteArrayOutputStream outputStream = new ByteArrayOutputStream();
96                    NanoHTTPD.DefaultTempFileManager tempFileManager = new NanoHTTPD.DefaultTempFileManager();
97                    try {
98                        NanoHTTPD.HTTPSession session = HttpKeepAliveTest.this.testServer.createSession(tempFileManager, inputStream, outputStream);
99                        for (int i = 0; i < 2048; i++) {
100                            requestStream.write(request.getBytes());
101                            requestStream.flush();
102                            outputStream.reset();
103                            session.execute();
104                            assertResponse(outputStream, expected);
105                        }
106
107                        // Finally, try "Connection: Close"
108                        String closeReq = request.replaceAll("HTTP/1.1", "HTTP/1.1\r\nConnection: Close");
109                        expected[3] = "Connection: close";
110                        requestStream.write(closeReq.getBytes());
111                        outputStream.reset();
112                        requestStream.flush();
113                        // Server should now close the socket by throwing a
114                        // SocketException:
115                        try {
116                            session.execute();
117                        } catch (java.net.SocketException se) {
118                            junit.framework.Assert.assertEquals(se.getMessage(), "NanoHttpd Shutdown");
119                        }
120                        assertResponse(outputStream, expected);
121
122                    } finally {
123                        tempFileManager.clear();
124                    }
125                } catch (Throwable t) {
126                    HttpKeepAliveTest.this.error = t;
127                }
128            }
129        };
130        Thread t = new Thread(null, r, "Request Thread", 1 << 17);
131        t.start();
132        t.join();
133        if (this.error != null) {
134            fail("" + this.error);
135            this.error.printStackTrace();
136        }
137    }
138}
139