/*
* Copyright (c) 2002-2012 Alibaba Group Holding Limited.
* All rights reserved.
*
* 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.alibaba.citrus.util.internal.webpagelite.myprocessor;
import static com.alibaba.citrus.util.CollectionUtil.*;
import static com.alibaba.citrus.util.StringUtil.*;
import static org.junit.Assert.*;
import java.io.File;
import java.io.IOException;
import java.io.PrintWriter;
import java.text.SimpleDateFormat;
import java.util.Date;
import java.util.Locale;
import java.util.Map;
import java.util.TimeZone;
import javax.servlet.ServletContext;
import javax.servlet.http.HttpServlet;
import javax.servlet.http.HttpServletRequest;
import javax.servlet.http.HttpServletResponse;
import javax.servlet.http.HttpServletResponseWrapper;
import com.alibaba.citrus.util.internal.webpagelite.RequestProcessor;
import com.alibaba.citrus.util.internal.webpagelite.ServletRequestContext;
import com.meterware.httpunit.WebResponse;
import com.meterware.servletunit.InvocationContext;
import com.meterware.servletunit.ServletRunner;
import com.meterware.servletunit.ServletUnitClient;
import org.junit.Before;
import org.junit.Test;
public class LastModifiedTests {
private SimpleDateFormat fmt;
private ServletUnitClient client;
private InvocationContext invocationContext;
private HttpServletRequest request;
private HttpServletResponse response;
private ServletContext servletContext;
private Map<String, String> textResources;
private MyProcessor page;
@Before
public void init() throws Exception {
fmt = new SimpleDateFormat("EEE, dd MMM yyyy HH:mm:ss z", Locale.US);
fmt.setTimeZone(TimeZone.getTimeZone("GMT"));
ServletRunner servletRunner = new ServletRunner();
servletRunner.registerServlet("/myservlet/*", MyServlet.class.getName());
client = servletRunner.newClient();
client.setExceptionsThrownOnErrorStatus(false);
client.getClientProperties().setAutoRedirect(false);
invocationContext = client.newInvocation("http://localhost/myservlet");
request = invocationContext.getRequest();
// http unit在这里有个可耻的错误，HH:mm:ss写成了hh:mm:ss，导致日期错误。
response = new HttpServletResponseWrapper(invocationContext.getResponse()) {
@Override
public void setDateHeader(String name, long date) {
super.setHeader(name, fmt.format(new Date(date)));
}
};
servletContext = invocationContext.getServlet().getServletConfig().getServletContext();
textResources = createHashMap();
textResources.put("file1", "file1 content");
textResources.put("path/to/", "list");
textResources.put("path/to/file2", "file2 content");
page = new MyProcessor();
}
@Test
public void lastModifiedResource() throws Exception {
page.processRequest(getRequestContext("/dummy.txt"));
assertEquals(new File(getClass().getResource("dummy.txt").toURI()).lastModified() / 1000 * 1000,
getResponseLastModified());
page.processRequest(getRequestContext("/prototype.js"));
assertEquals(new File(RequestProcessor.class.getResource("prototype.js").toURI()).lastModified() / 1000 * 1000,
getResponseLastModified());
}
@Test
public void lastModifiedResourceTemplate() throws Exception {
page.processRequest(getRequestContext("/style.txt"));
assertEquals(-1, getResponseLastModified());
}
@Test
public void lastModifiedListRedirect() throws Exception {
page.processRequest(getRequestContext("/path/to"));
assertEquals(-1, getResponseLastModified());
}
@Test
public void lastModifiedRenderList() throws Exception {
page.processRequest(getRequestContext("/path/to/"));
assertEquals(-1, getResponseLastModified());
}
@Test
public void lastModifiedNotFound() throws Exception {
page.processRequest(getRequestContext("/path/to/notFound"));
assertEquals(-1, getResponseLastModified());
}
private ServletRequestContext getRequestContext(String name) {
return new ServletRequestContext(request, response, servletContext, "http://localhost/myservlet", name) {
};
}
private long getResponseLastModified() throws Exception {
WebResponse webResponse = client.getResponse(invocationContext);
String lastModifiedStr = trimToNull(webResponse.getHeaderField("last-modified"));
if (lastModifiedStr == null) {
return -1;
} else {
return fmt.parse(lastModifiedStr).getTime();
}
}
private class MyProcessor extends RequestProcessor<ServletRequestContext> {
@Override
protected void renderPage(ServletRequestContext request, String resourceName) throws IOException {
PrintWriter out = request.getWriter("text/plain");
out.println(textResources.get(resourceName));
}
@Override
protected boolean resourceExists(String resourceName) {
return textResources.containsKey(resourceName);
}
}
public static class MyServlet extends HttpServlet {
private static final long serialVersionUID = 4555315413735967231L;
}
}