modules/tomcat-lite/test/org/apache/tomcat/lite/http/HttpChannelInMemoryTest.java

/*
 */
package org.apache.tomcat.lite.http;

import java.io.BufferedReader;
import java.io.IOException;

import junit.framework.TestCase;

import org.apache.tomcat.lite.http.HttpChannel.HttpService;
import org.apache.tomcat.lite.http.HttpChannel.RequestCompleted;
import org.apache.tomcat.lite.io.BBuffer;
import org.apache.tomcat.lite.io.CBuffer;
import org.apache.tomcat.lite.io.IOBuffer;
import org.apache.tomcat.lite.io.IOChannel;
import org.apache.tomcat.lite.io.IOConnector;
import org.apache.tomcat.lite.io.MemoryIOConnector;
import org.apache.tomcat.lite.io.MemoryIOConnector.MemoryIOChannel;

// TODO: rename to Http11ConnectionTest
public class HttpChannelInMemoryTest extends TestCase {
    /**
     *  Connection under test
     */
    Http11Connection conn;

    /**
     * Last http channel created by the connection
     */
    volatile HttpChannel http;

    // Input/output for the connection
    MemoryIOConnector.MemoryIOChannel net = new MemoryIOChannel();

    HttpConnector serverConnector = new HttpConnector(null);

    // Callback results for callback tests
    boolean hasBody = false;
    boolean bodyDone = false;
    boolean bodySentDone = false;
    boolean headersDone = false;
    boolean allDone = false;

    public void setUp() throws IOException {
        // Requests will not be serviced - you must manually generate
        // the response.
        serverConnector.setHttpService(null);

        conn = new Http11Connection(serverConnector) {
            protected HttpChannel checkHttpChannel() throws IOException {
                return http = super.checkHttpChannel();
            }
        }.serverMode();
        conn.setSink(net);
    }


    public void test2Req() throws IOException {
        String req = "GET /index.html?q=b&c=d HTTP/1.1\r\n" +
        "Host:  Foo.com \n" +
        "H2:Bar\r\n" +
        "H3: Foo \r\n" +
        " Bar\r\n" +
        "H4: Foo\n" +
        "    Bar\n" +
        "\r\n" +
        "HEAD /r2? HTTP/1.1\n" +
        "Host: Foo.com\r\n" +
        "H3: Foo \r\n" +
        "       Bar\r\n" +
        "H4: Foo\n" +
        " Bar\n" +
        "\r\n";
        net.getIn().append(req);

        //http = lastServer.get(0);
        assertTrue(http.getRequest().method().equals("GET"));
        assertTrue(http.getRequest().protocol().equals("HTTP/1.1"));
        assertEquals(http.getRequest().getMimeHeaders().size(), 4);
        assertEquals(http.getRequest().getMimeHeaders().getHeader("Host").toString(),
                "Foo.com");
        assertEquals(http.getRequest().getMimeHeaders().getHeader("H2").toString(),
                "Bar");

        http.getOut().append("Response1");
        http.getOut().close();
        http.startSending();
        http.release();

        // now second response must be in.
        // the connector will create a new http channel

        //http = lastServer.get(1);

        assertTrue(http.getRequest().method().equals("HEAD"));
        assertTrue(http.getRequest().protocol().equals("HTTP/1.1"));
        assertTrue(http.getRequest().getMimeHeaders().size() == 3);
        assertTrue(http.getRequest().getMimeHeaders().getHeader("Host")
                .equals("Foo.com"));
    }

    public void testHttp11Close() throws IOException {
        String req = "GET /index.html?q=b&c=d HTTP/1.1\r\n" +
        "Host:  Foo.com\n" +
        "Connection: close\n" +
        "\n";
        net.getIn().append(req);

        assertTrue(http.getRequest().method().equals("GET"));
        assertTrue(http.getRequest().protocol().equals("HTTP/1.1"));

        http.getOut().append("Response1");
        http.getOut().close();
        http.startSending();
        http.release();

        assertTrue(net.out.indexOf("connection:close") > 0);
        assertFalse(net.isOpen());
    }

    public void testHttp10Close() throws IOException {
        String req = "GET /index.html?q=b&c=d HTTP/1.0\r\n" +
        "Host:  Foo.com \n" +
        "\r\n";
        net.getIn().append(req);

        assertTrue(http.getRequest().method().equals("GET"));
        assertTrue(http.getRequest().protocol().equals("HTTP/1.0"));

        http.getOut().append("Response1");
        http.getOut().close();
        http.startSending();

        assertTrue(net.out.indexOf("connection:close") > 0);
        assertFalse(net.isOpen());
    }

    public void testHttp10KA() throws IOException {
        String req = "GET /index.html?q=b&c=d HTTP/1.0\r\n" +
        "Connection: Keep-Alive\n" +
        "Host:  Foo.com \n" +
        "\r\n";
        net.getIn().append(req);

        assertTrue(http.getRequest().method().equals("GET"));
        assertTrue(http.getRequest().protocol().equals("HTTP/1.0"));

        http.getOut().append("Hi");
        http.getOut().close();
        http.startSending();

        // after request
        assertEquals(conn.activeHttp, null);

        assertTrue(net.out.indexOf("connection:keep-alive") > 0);
        assertTrue(net.isOpen());
        // inserted since we can calculate the response
        assertEquals(http.getResponse().getHeader("Content-Length"),
                   "2");
    }

    public void testHttp10KANoCL() throws IOException {
        String req = "GET /index.html?q=b&c=d HTTP/1.0\r\n" +
        "Connection: Keep-Alive\n" +
        "Host:  Foo.com \n" +
        "\r\n";
        net.getIn().append(req);

        assertTrue(http.getRequest().method().equals("GET"));
        assertTrue(http.getRequest().protocol().equals("HTTP/1.0"));

        http.getOut().append("Hi");
        http.startSending();

        http.getOut().append("After");
        http.getOut().close();
        http.startSending();

        // after request
        assertEquals(conn.activeHttp, null);

        assertFalse(net.out.indexOf("connection:keep-alive") > 0);
        assertFalse(net.isOpen());
        // inserted since we can calculate the response
        assertEquals(http.getResponse().getHeader("Content-Length"),
                null);
        assertEquals(http.getResponse().getHeader("Transfer-Encoding"),
                null);
    }

    public void testMultiLineHead() throws IOException {
        net.getIn().append("GET / HTTP/1.0\n" +
                "Cookie: 1234\n" +
                "  456 \n" +
                "Connection:   Close\n\n");
        net.getIn().close();

        MultiMap headers = http.getRequest().getMimeHeaders();
        CBuffer cookie = headers.getHeader("Cookie");
        CBuffer conn = headers.getHeader("Connection");
        assertEquals(conn.toString(), "close");
        assertEquals(cookie.toString(), "1234 456");

        assertEquals(http.conn.headRecvBuf.toString(),
                "GET / HTTP/1.0\n" +
                "Cookie: 1234 456   \n" + // \n -> trailing space
                "Connection:   Close\n\n");
    }

    public void testCloseSocket() throws IOException {
        net.getIn().append("GET / HTTP/1.1\n"
                + "Host: localhost\n"
                + "\n");
        assertTrue(((Http11Connection)http.conn).keepAlive());

        net.getIn().close();
        assertFalse(((Http11Connection)http.conn).keepAlive());
    }

    public void test2ReqByte2Byte() throws IOException {
        String req = "GET /index.html?q=b&c=d HTTP/1.1\r\n" +
        "Host:  Foo.com \n" +
        "H2:Bar\r\n" +
        "H3: Foo \r\n" +
        " Bar\r\n" +
        "H4: Foo\n" +
        "    Bar\n" +
        "\r\n" +
        "HEAD /r2? HTTP/1.1\n" +
        "Host: Foo1.com\n" +
        "H3: Foo \r\n" +
        "       Bar\r\n" +
        "\r\n";
        for (int i = 0; i < req.length(); i++) {
            net.getIn().append(req.charAt(i));
        }

        assertTrue(http.getRequest().method().equals("GET"));
        assertTrue(http.getRequest().protocol().equals("HTTP/1.1"));
        assertTrue(http.getRequest().getMimeHeaders().size() == 4);
        assertTrue(http.getRequest().getMimeHeaders().getHeader("Host")
                .equals("Foo.com"));

        // send a response
        http.sendBody.append("Response1");
        http.getOut().close();

        http.startSending(); // This will trigger a pipelined request

        http.release(); // now second response must be in

        assertTrue(http.getRequest().method().equals("HEAD"));
        assertTrue(http.getRequest().protocol().equals("HTTP/1.1"));
        assertTrue(http.getRequest().getMimeHeaders().size() == 2);
        assertTrue(http.getRequest().getMimeHeaders().getHeader("Host")
                .equals("Foo1.com"));

        // send a response - service method will be called
        http.sendBody.append("Response2");
        http.getOut().close();
        http.release(); // now second response must be in


    }

    public void testEndWithoutFlushCallbacks() throws IOException {

        net.getIn().append(POST);

        net.getIn().close();
        http.setCompletedCallback(new RequestCompleted() {
            public void handle(HttpChannel data, Object extra)
            throws IOException {
                allDone = true;
            }
        });

        http.sendBody.queue("Hi");
        http.getOut().close();
        http.startSending(); // will call handleEndSend

        assertTrue(allDone);

    }

    public void testCallbacks() throws IOException {
        // already accepted - will change
        serverConnector.setHttpService(new HttpService() {
            public void service(HttpRequest httpReq, HttpResponse httpRes)
                    throws IOException {

                headersDone = true;
                HttpChannel http = httpReq.getHttpChannel();

                http.setCompletedCallback(new RequestCompleted() {
                    public void handle(HttpChannel data, Object extra)
                    throws IOException {
                        allDone = true;
                    }
                });
                http.setDataReceivedCallback(new IOConnector.DataReceivedCallback() {
                    @Override
                    public void handleReceived(IOChannel ch) throws IOException {
                        if (ch.getIn().isAppendClosed()) {
                            bodyDone = true;
                        }
                    }
                });
                http.setDataFlushedCallback(new IOConnector.DataFlushedCallback() {
                    @Override
                    public void handleFlushed(IOChannel ch) throws IOException {
                        if (ch.getOut().isAppendClosed()) {
                            bodySentDone = true;
                        }
                    }
                });
            }
        });

        // Inject the request
        net.getIn().append("POST / HTTP/1.0\n" +
                "Connection: Close\n" +
                "Content-Length: 4\n\n" +
                "1");
        assertTrue(headersDone);
        net.getIn().append("234");

        net.getIn().close();
        assertTrue(bodyDone);


        http.sendBody.queue("Hi");
        http.getOut().close();
        http.startSending();
        assertTrue(bodySentDone);

        assertTrue(allDone);

    }

    public static String POST = "POST / HTTP/1.0\n" +
        "Connection: Close\n" +
        "Content-Length: 4\n\n" +
        "1234";

    public void testClose() throws IOException {
        net.getIn().append(POST);
        net.getIn().close();

        IOBuffer receiveBody = http.receiveBody;
        IOBuffer appData = receiveBody;
        BBuffer res = BBuffer.allocate(1000);
        appData.readAll(res);

        assertEquals(res.toString(), "1234");
        assertFalse(((Http11Connection)http.conn).keepAlive());

        http.sendBody.queue(res);
        http.getOut().close();
        http.startSending();

        assertTrue(net.getOut().isAppendClosed());
        assertTrue(net.out.toString().indexOf("\n1234") > 0);

    }

    public void testReadLine() throws Exception {
        net.getIn().append("POST / HTTP/1.0\n" +
        		"Content-Length: 28\n\n" +
                "Line 1\n" +
                "Line 2\r\n" +
                "Line 3\r" +
                "Line 4");
        net.getIn().close();

        BufferedReader r = http.getRequest().getReader();
        assertEquals("Line 1", r.readLine());
        assertEquals("Line 2", r.readLine());
        assertEquals("Line 3", r.readLine());
        assertEquals("Line 4", r.readLine());
        assertEquals(null, r.readLine());

    }
}