org.devtcg.five.util.streaming.LocalHttpServerTest.java Source code

Introduction

Here is the source code for org.devtcg.five.util.streaming.LocalHttpServerTest.java

Source

/*
 * Copyright (C) 2008 Josh Guilfoyle <jasta@devtcg.org>
 *
 * This program is free software; you can redistribute it and/or modify it
 * under the terms of the GNU General Public License as published by the
 * Free Software Foundation; either version 2, or (at your option) any
 * later version.
 *
 * This program is distributed in the hope that it will be useful, but
 * WITHOUT ANY WARRANTY; without even the implied warranty of
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
 * General Public License for more details.
 */

package org.devtcg.five.util.streaming;

import java.io.IOException;
import java.io.InputStream;
import java.io.OutputStream;
import java.io.OutputStreamWriter;
import java.util.Locale;
import java.util.Random;

import junit.framework.TestCase;

import org.apache.http.HttpEntity;
import org.apache.http.HttpException;
import org.apache.http.HttpRequest;
import org.apache.http.HttpResponse;
import org.apache.http.HttpStatus;
import org.apache.http.MethodNotSupportedException;
import org.apache.http.RequestLine;
import org.apache.http.StatusLine;
import org.apache.http.client.HttpClient;
import org.apache.http.client.methods.HttpGet;
import org.apache.http.entity.ContentProducer;
import org.apache.http.entity.EntityTemplate;
import org.apache.http.impl.client.DefaultHttpClient;
import org.apache.http.protocol.HttpContext;
import org.apache.http.protocol.HttpRequestHandler;

import android.util.Log;

public class LocalHttpServerTest extends TestCase {
    public static final String TAG = "LocalHttpServerTest";

    public MyHttpServer startMyHttpServer() throws IOException {
        int port = 1024 + (new Random()).nextInt(1000);

        Log.v(TAG, "Starting server on port " + port);

        MyHttpServer server = new MyHttpServer(port);
        server.start();

        return server;
    }

    public void testServerQuick() throws IOException {
        LocalHttpServer server = startMyHttpServer();
        server.shutdown();
    }

    public void testServerResponse() throws IOException {
        LocalHttpServer server = startMyHttpServer();

        try {
            String url = "http://127.0.0.1:" + server.getPort() + "/testing/1/2/3";
            HttpClient cli = new DefaultHttpClient();
            HttpGet method = new HttpGet(url);

            HttpResponse resp = cli.execute(method);

            StatusLine status = resp.getStatusLine();
            assertEquals(status.getStatusCode(), HttpStatus.SC_OK);

            HttpEntity ent = resp.getEntity();
            assertNotNull(ent);

            InputStream in = ent.getContent();

            byte[] b = new byte[2048];
            int n;
            long recvd = 0;

            while ((n = in.read(b)) >= 0)
                recvd += n;

            in.close();

            assertTrue(recvd > 0);
        } finally {
            server.shutdown();
        }
    }

    public static class MyHttpServer extends LocalHttpServer {
        public MyHttpServer(int port) throws IOException {
            super(port);
            setRequestHandler(mHttpHandler);
        }

        private final HttpRequestHandler mHttpHandler = new HttpRequestHandler() {
            public void handle(HttpRequest request, HttpResponse response, HttpContext context)
                    throws HttpException, IOException {
                RequestLine reqLine = request.getRequestLine();
                Log.v(TAG, "reqLine=" + reqLine);

                String method = reqLine.getMethod().toUpperCase(Locale.ENGLISH);
                if (method.equals("GET") == false) {
                    throw new MethodNotSupportedException(method + " method not supported");
                }

                response.setEntity(new MyBoilerPlateEntity());
                response.setStatusCode(HttpStatus.SC_OK);
            }
        };

        public static class MyBoilerPlateEntity extends EntityTemplate {
            public MyBoilerPlateEntity() {
                super(new ContentProducer() {
                    public void writeTo(OutputStream out) throws IOException {
                        OutputStreamWriter w = new OutputStreamWriter(out, "UTF-8");

                        w.write("<html><body>");
                        w.write("<h1>Success!</h1>");
                        w.write("</body></html>");
                        w.flush();
                    }
                });

                setContentType("text/html; charset=UTF-8");
            }
        }
    }
}