com.nominanuda.web.http.ServletHelperTest.java Source code

Introduction

Here is the source code for com.nominanuda.web.http.ServletHelperTest.java

Source

/*
 * Copyright 2008-2011 the original author or authors.
 * 
 * Licensed under the Apache License, Version 2.0 (the "License");
 * you may not use this file except in compliance with the License.
 * You may obtain a copy of the License at
 * 
 *      http://www.apache.org/licenses/LICENSE-2.0
 * 
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 * See the License for the specific language governing permissions and
 * limitations under the License.
 */
package com.nominanuda.web.http;

import java.io.*;
import static org.junit.Assert.*;
import javax.servlet.*;
import javax.servlet.http.*;

import org.apache.http.*;
import org.apache.http.client.*;
import org.apache.http.client.methods.*;
import org.apache.http.entity.*;
import org.apache.http.message.*;
import org.apache.http.util.*;
import org.eclipse.jetty.server.*;
import org.eclipse.jetty.server.handler.*;
import org.junit.*;
import org.springframework.mock.web.*;

public class ServletHelperTest extends BaseHttpTest {
    ServletHelper servletHelper = new ServletHelper();

    @Test
    public void testGetRequestLineURI() {
        MockHttpServletRequest req = new MockHttpServletRequest("GET", "/foo%20bar?z=+1");
        String reqLineURI = servletHelper.getRequestLineURI(req, false);
        System.err.println(reqLineURI);
    }

    @Test
    public void testCopyRequest() throws Exception {
        final String msg = "mio";
        final String mediaType = CT_APPLICATION_OCTET_STREAM;
        Server server = startServer(10000, new AbstractHandler() {
            public void handle(String arg0, Request jettyReq, HttpServletRequest servletReq,
                    HttpServletResponse arg3) throws IOException, ServletException {
                HttpRequest r = servletHelper.copyRequest(servletReq, false);
                asyncAssertEquals("bar", r.getFirstHeader("X-foo").getValue());
                asyncAssertEquals("PUT", r.getRequestLine().getMethod());
                HttpEntity e = ((HttpEntityEnclosingRequest) r).getEntity();
                asyncAssert(msg.getBytes("UTF-8").length == e.getContentLength(), "length");
                asyncAssert(e.getContentType().getValue().startsWith(mediaType));
                asyncAssertEquals(mediaType, ContentType.get(e).getMimeType());
                asyncAssertEquals(msg, EntityUtils.toString(e));
            }
        });
        HttpClient c = buildClient(1);
        HttpPut req = new HttpPut("http://localhost:10000/foo/bar?a=b&a=");
        req.setEntity(new StringEntity(msg, ContentType.create(mediaType, CS_UTF_8)));
        req.addHeader("X-foo", "bar");
        c.execute(req);
        server.stop();
        dumpFailures(System.err);
        Assert.assertFalse(isFailed());
    }

    @Test
    public void shouldAddCookiesOnServletHttpResponseWhenHeaderSetCookie() throws IOException {
        MockHttpServletResponse servletResponse = new MockHttpServletResponse();

        HttpResponse response = new BasicHttpResponse(new HttpCoreHelper().statusLine(200));
        Header[] headers = new Header[2];
        headers[0] = new BasicHeader("Set-Cookie",
                "etalia=dGltZXN0YW1wPTEzMzE4ODQxxNzE4ODc.dXNlcm5hbWU9bHVjYQ.Y3JlYXRlZF9hdD0xxMzMxxODg0MTcxxODg3.cGFzc3dvcmQ9b0Z2WWtNU0dqcUdBZjRWa0JWMGZ1bmZHdW9F.Y29uZmlybV9hY2s9MDg0ZGM3ZTItM2VlMC00NWNkLWI2NmQtZGVmZTlmM2E5NTRm;Domain=localhost;Path=/;Expires=Mon, 02-Apr-2012 10:30:00 GMT");
        headers[1] = new BasicHeader("Set-Cookie",
                "etalia_hash=4bYEnD7APP19SKPlf1x1IDrPzNY0;Domain=localhost;Path=/;Expires=Mon, 02-Apr-2012 10:30:00 GMT");
        response.setHeaders(headers);

        servletHelper.copyResponse(response, servletResponse);

        Cookie[] cookies = servletResponse.getCookies();

        assertEquals(2, cookies.length);

        assertEquals("/", cookies[0].getPath());
        assertEquals("localhost", cookies[0].getDomain());
        assertNotSame(0, cookies[0].getMaxAge());

        assertEquals("/", cookies[1].getPath());
        assertEquals("localhost", cookies[1].getDomain());
        assertNotSame(0, cookies[1].getMaxAge());
    }

}