tapestry/tapestry-framework/src/test/org/apache/tapestry/asset/TestAssetService.java

// Copyright 2004, 2005 The Apache Software Foundation
//
// 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 org.apache.tapestry.asset;

import org.apache.commons.logging.LogFactory;
import org.apache.hivemind.ClassResolver;
import org.apache.hivemind.impl.DefaultClassResolver;
import org.apache.tapestry.IRequestCycle;
import org.apache.tapestry.TestBase;
import org.apache.tapestry.util.ContentType;
import org.apache.tapestry.web.WebContext;
import org.apache.tapestry.web.WebRequest;
import org.apache.tapestry.web.WebResponse;
import static org.easymock.EasyMock.checkOrder;
import static org.easymock.EasyMock.expect;
import org.testng.annotations.Test;

import javax.servlet.http.HttpServletResponse;
import java.io.ByteArrayOutputStream;
import java.net.URLConnection;


/**
 * Tests functionality of {@link AssetService}. 
 *
 * @author jkuhnert
 */
@Test(sequential=true)
public class TestAssetService extends TestBase {
    
    public void test_Cached_Resource_Null_Modified()
    {
        WebRequest request = newMock(WebRequest.class);
        checkOrder(request, false);
        
        AssetService service = new AssetService();
        service.setRequest(request);
        service.setLog(LogFactory.getLog("test"));

        expect(request.getDateHeader("If-Modified-Since")).andReturn(-1l);
        
        replay();

        assertFalse(service.cachedResource(null));
        
        verify();
    }
    
    public void test_Cached_Resource_Stale()
    {
        WebRequest request = newMock(WebRequest.class);
        checkOrder(request, false);
        
        AssetService service = new AssetService();
        service.setRequest(request);
        service.setLog(LogFactory.getLog("test"));
        
        URLConnection url = org.easymock.classextension.EasyMock.createMock(URLConnection.class);

        long modifiedSince = System.currentTimeMillis() - 1000;

        expect(request.getDateHeader("If-Modified-Since")).andReturn(modifiedSince);
        expect(url.getLastModified()).andReturn(System.currentTimeMillis());
        
        replay();
        org.easymock.classextension.EasyMock.replay(url);
        
        assertFalse(service.cachedResource(url));
        
        verify();
        org.easymock.classextension.EasyMock.verify(url);
    }
    
    public void test_Cached_Resource_Good()
    throws Exception
    {
        WebRequest request = newMock(WebRequest.class);
        checkOrder(request, false);
        WebResponse response = newMock(WebResponse.class);
        
        AssetService service = new AssetService();
        service.setRequest(request);
        service.setResponse(response);
        service.setLog(LogFactory.getLog("test"));
        
        URLConnection url = org.easymock.classextension.EasyMock.createMock(URLConnection.class);

        long lastModified = System.currentTimeMillis() - 4000;
        long modifiedSince = System.currentTimeMillis();

        expect(request.getDateHeader("If-Modified-Since")).andReturn(modifiedSince);
        expect(url.getLastModified()).andReturn(lastModified);
        
        response.setStatus(HttpServletResponse.SC_NOT_MODIFIED);
        
        replay();
        org.easymock.classextension.EasyMock.replay(url);
        
        assertTrue(service.cachedResource(url));
        
        verify();
        org.easymock.classextension.EasyMock.verify(url);
    }

    public void test_ETag_Header_Response()
            throws Exception
    {
        WebRequest request = newMock(WebRequest.class);
        checkOrder(request, false);
        WebResponse response = newMock(WebResponse.class);
        WebContext context = newMock(WebContext.class);
        IRequestCycle cycle = newMock(IRequestCycle.class);
        ResourceMatcher matcher = newMock(ResourceMatcher.class);
        
        ClassResolver resolver = new DefaultClassResolver();
        URLConnection url = resolver.getResource("/org/apache/tapestry/asset/tapestry-in-action.png").openConnection();

        AssetService service = new AssetService();
        service.setRequest(request);
        service.setResponse(response);
        service.setLog(LogFactory.getLog("test"));
        service.setUnprotectedMatcher(matcher);
        service.setClassResolver(resolver);
        service.setContext(context);

        expect(cycle.getParameter("path")).andReturn("/org/apache/tapestry/asset/tapestry-in-action.png");
        expect(cycle.getParameter("digest")).andReturn(null);

        expect(matcher.containsResource("/org/apache/tapestry/asset/tapestry-in-action.png")).andReturn(true);

        expect(request.getDateHeader("If-Modified-Since")).andReturn(-1L);
        expect(context.getMimeType("/org/apache/tapestry/asset/tapestry-in-action.png")).andReturn("image/png");

        response.setDateHeader("Last-Modified", url.getLastModified());
        response.setDateHeader("Expires", service._expireTime);
        response.setHeader("Cache-Control", "public, max-age=" + (AssetService.MONTH_SECONDS * 3));

        expect(request.getHeader("User-Agent")).andReturn("Mozilla").anyTimes();

        response.setHeader("ETag", "W/\"" + url.getContentLength() + "-" + url.getLastModified() + "\"");
        response.setContentLength(url.getContentLength());

        expect(response.getOutputStream(new ContentType("image/png"))).andReturn(new ByteArrayOutputStream());

        replay();

        service.service(cycle);

        verify();        
    }
}