VaryAggregatorTest.java

package org.esigate.test.cases;

import java.io.IOException;
import java.net.MalformedURLException;

import org.apache.http.HttpResponse;
import org.apache.http.client.HttpClient;
import org.apache.http.client.config.CookieSpecs;
import org.apache.http.client.config.RequestConfig;
import org.apache.http.client.methods.HttpGet;
import org.apache.http.client.protocol.HttpClientContext;
import org.apache.http.impl.client.BasicCookieStore;
import org.apache.http.impl.client.HttpClientBuilder;
import org.apache.http.impl.cookie.BasicClientCookie;
import org.esigate.http.HttpResponseUtils;
import org.xml.sax.SAXException;

import junit.framework.TestCase;

/**
 * "Vary" header testing
 * 
 * @author Nicolas Richeton
 */
public class VaryAggregatorTest extends TestCase {
    private final static String APPLICATION_PATH = "http://localhost:8080/esigate-app-aggregator/";

    /**
     * Send a request with a Cookie "test-cookie" to vary.jsp (which will get content from provider) and ensure the
     * result is valid.
     * 
     * @param addCookie
     * @param value
     * @param refresh
     * @return
     * @throws MalformedURLException
     * @throws IOException
     * @throws SAXException
     */
    private String doRequestWithHeader(String headerValue, boolean forceRefresh) throws Exception {
        RequestConfig requestConfig = RequestConfig.custom().setCookieSpec(CookieSpecs.DEFAULT).build();
        HttpClient client = HttpClientBuilder.create().setDefaultRequestConfig(requestConfig).build();
        HttpClientContext context = new HttpClientContext();
        context.setCookieStore(new BasicCookieStore());

        HttpGet request = new HttpGet(APPLICATION_PATH + "vary.jsp");
        if (headerValue != null) {
            BasicClientCookie cookie = new BasicClientCookie("test-cookie", headerValue);
            cookie.setDomain("localhost");
            cookie.setPath("/");
            context.getCookieStore().addCookie(cookie);
            request.setHeader("foo", headerValue);
        }
        if (forceRefresh) {
            request.addHeader("Cache-Control", "no-cache");
        }
        HttpResponse response = client.execute(request, context);
        // Ensure content is valid.
        String text = HttpResponseUtils.toString(response, null);
        assertNotNull(text);
        if (headerValue != null) {
            assertTrue("no value '" + headerValue + "' found", text.contains(headerValue));
        } else {
            assertTrue("no cookie found", text.contains("no cookie"));
        }

        // Ensure vary and Cache-Control header were forwarded
        assertEquals("foo", response.getFirstHeader("Vary").getValue());
        assertEquals("public, max-age=3600", response.getFirstHeader("Cache-Control").getValue());

        // Return page timestamp. Can be used to detect cache hits.
        return text.substring(text.indexOf("stime") + 5, text.indexOf("etime"));
    }

    /**
     * Call "vary.jsp" with different cookie values and ensure cache use and content are valid.
     * 
     * @throws Exception
     */
    public void testBlockVary() throws Exception {

        // Cache test
        String value1 = doRequestWithHeader("value1", false);
        String value2 = doRequestWithHeader("value2", false);
        assertEquals(value1, doRequestWithHeader("value1", false));
        assertEquals(value2, doRequestWithHeader("value2", false));

    }

}