/*
* Licensed to the Apache Software Foundation (ASF) under one
* or more contributor license agreements. See the NOTICE file
* distributed with this work for additional information
* regarding copyright ownership. The ASF licenses this file
* to you 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.sling.servlets.post.impl;
import java.io.IOException;
import java.io.UnsupportedEncodingException;
import java.net.URLEncoder;
import java.util.StringTokenizer;
import junit.framework.TestCase;
import org.apache.sling.api.SlingHttpServletRequest;
import org.apache.sling.commons.testing.sling.MockSlingHttpServletRequest;
import org.apache.sling.commons.testing.sling.MockSlingHttpServletResponse;
import org.apache.sling.servlets.post.HtmlResponse;
import org.apache.sling.servlets.post.JSONResponse;
import org.apache.sling.servlets.post.PostResponse;
import org.apache.sling.servlets.post.SlingPostConstants;
import org.apache.sling.servlets.post.impl.helper.MediaRangeList;
public class SlingPostServletTest extends TestCase {
private SlingPostServlet servlet;
@Override
protected void setUp() throws Exception {
super.setUp();
servlet = new SlingPostServlet();
}
public void testIsSetStatus() {
StatusParamSlingHttpServletRequest req = new StatusParamSlingHttpServletRequest();
// 1. null parameter, expect true
req.setStatusParam(null);
assertTrue("Standard status expected for null param",
servlet.isSetStatus(req));
// 2. "standard" parameter, expect true
req.setStatusParam(SlingPostConstants.STATUS_VALUE_STANDARD);
assertTrue("Standard status expected for '"
+ SlingPostConstants.STATUS_VALUE_STANDARD + "' param",
servlet.isSetStatus(req));
// 3. "browser" parameter, expect false
req.setStatusParam(SlingPostConstants.STATUS_VALUE_BROWSER);
assertFalse("Browser status expected for '"
+ SlingPostConstants.STATUS_VALUE_BROWSER + "' param",
servlet.isSetStatus(req));
// 4. any parameter, expect true
String param = "knocking on heaven's door";
req.setStatusParam(param);
assertTrue("Standard status expected for '" + param + "' param",
servlet.isSetStatus(req));
}
public void testGetJsonResponse() {
MockSlingHttpServletRequest req = new MockSlingHttpServletRequest(null, null, null, null, null) {
@Override
public String getHeader(String name) {
return name.equals(MediaRangeList.HEADER_ACCEPT) ? "application/json" : super.getHeader(name);
}
public <AdapterType> AdapterType adaptTo(Class<AdapterType> type) {
return null;
}
};
PostResponse result = servlet.createPostResponse(req);
assertTrue(result instanceof JSONResponse);
}
public void testRedirection() throws Exception {
String utf8Path = "\u0414\u0440\u0443\u0433\u0430";
String encodedUtf8 = "%D0%94%D1%80%D1%83%D0%B3%D0%B0";
testRedirection("/", "/fred", "*.html", "/fred.html");
testRedirection("/xyz/", "/xyz/"+utf8Path, "*", "/xyz/"+encodedUtf8);
testRedirection("/", "/fred/abc", "http://forced", "http://forced");
testRedirection("/", "/fred/"+utf8Path, "http://forced/xyz/*", "http://forced/xyz/"+encodedUtf8);
testRedirection("/", "/fred/"+utf8Path, null, null);
}
private void testRedirection(String requestPath, String resourcePath, String redirect, String expected)
throws Exception {
RedirectServletResponse resp = new RedirectServletResponse();
SlingHttpServletRequest request = new RedirectServletRequest(redirect, requestPath);
PostResponse htmlResponse = new HtmlResponse();
htmlResponse.setPath(resourcePath);
assertEquals(expected != null, servlet.redirectIfNeeded(request, htmlResponse, resp));
assertEquals(expected, resp.redirectLocation);
}
/**
*
*/
private final class RedirectServletRequest extends MockSlingHttpServletRequest {
private String requestPath;
private String redirect;
private RedirectServletRequest(String redirect, String requestPath) {
super(null, null, null, null, null);
this.requestPath = requestPath;
this.redirect = redirect;
}
public String getPathInfo() {
return requestPath;
}
@Override
public String getParameter(String name) {
return SlingPostConstants.RP_REDIRECT_TO.equals(name) ? redirect : null;
}
}
private final class RedirectServletResponse extends MockSlingHttpServletResponse {
private String redirectLocation;
@Override
public String encodeRedirectURL(String s) {
StringTokenizer st = new StringTokenizer(s, "/", true);
StringBuilder sb = new StringBuilder();
try {
while (st.hasMoreTokens()) {
String token = st.nextToken();
if ("/".equals(token)) {
sb.append(token);
} else {
sb.append(URLEncoder.encode(token, "UTF-8"));
}
}
} catch (UnsupportedEncodingException e) {
fail("Should have UTF-8?? " + e);
return null;
}
return sb.toString();
}
@Override
public void sendRedirect(String s) throws IOException {
redirectLocation = s;
}
}
private static class StatusParamSlingHttpServletRequest extends
MockSlingHttpServletRequest {
private String statusParam;
public StatusParamSlingHttpServletRequest() {
// nothing to setup, we don't care
super(null, null, null, null, null);
}
@Override
public String getParameter(String name) {
if (SlingPostConstants.RP_STATUS.equals(name)) {
return statusParam;
}
return super.getParameter(name);
}
void setStatusParam(String statusParam) {
this.statusParam = statusParam;
}
public <AdapterType> AdapterType adaptTo(Class<AdapterType> type) {
return null;
}
}
}