/*
* Copyright (c) 2012, Inversoft Inc., 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 org.primeframework.mvc.action.result;
import javax.servlet.ServletException;
import javax.servlet.http.HttpServletRequest;
import javax.servlet.http.HttpServletResponse;
import java.io.IOException;
import java.lang.annotation.Annotation;
import java.util.ArrayList;
import java.util.List;
import org.primeframework.mvc.action.ActionInvocation;
import org.primeframework.mvc.action.ActionInvocationStore;
import org.primeframework.mvc.action.result.annotation.Status;
import org.primeframework.mvc.action.result.annotation.Status.Header;
import org.primeframework.mvc.message.Message;
import org.primeframework.mvc.message.MessageStore;
import org.primeframework.mvc.message.scope.MessageScope;
import org.primeframework.mvc.parameter.el.ExpressionEvaluator;
import org.testng.annotations.Test;
import static org.easymock.EasyMock.*;
/**
* This class tests the status result.
*
* @author Brian Pontarelli
*/
public class StatusResultTest {
@Test
public void noHeaders() throws IOException, ServletException {
ExpressionEvaluator ee = createStrictMock(ExpressionEvaluator.class);
replay(ee);
HttpServletRequest request = createStrictMock(HttpServletRequest.class);
expect(request.getContextPath()).andReturn("");
replay(request);
HttpServletResponse response = createStrictMock(HttpServletResponse.class);
response.setStatus(200);
replay(response);
ActionInvocationStore store = createStrictMock(ActionInvocationStore.class);
expect(store.getCurrent()).andReturn(new ActionInvocation(null, null, "/foo", "", null));
replay(store);
List<Message> messages = new ArrayList<Message>();
MessageStore messageStore = createStrictMock(MessageStore.class);
expect(messageStore.get(MessageScope.REQUEST)).andReturn(messages);
messageStore.clear(MessageScope.REQUEST);
messageStore.addAll(MessageScope.FLASH, messages);
replay(messageStore);
Status status = new StatusImpl("success", 200, "");
StatusResult result = new StatusResult(ee, response, store);
result.execute(status);
verify(response);
}
@Test
public void expansion() throws IOException, ServletException {
Object action = new Object();
ExpressionEvaluator ee = createStrictMock(ExpressionEvaluator.class);
expect(ee.expand("someFieldName", action, false)).andReturn("200");
replay(ee);
HttpServletRequest request = createStrictMock(HttpServletRequest.class);
expect(request.getContextPath()).andReturn("");
replay(request);
HttpServletResponse response = createStrictMock(HttpServletResponse.class);
response.setStatus(200);
replay(response);
ActionInvocationStore store = createStrictMock(ActionInvocationStore.class);
expect(store.getCurrent()).andReturn(new ActionInvocation(action, null, "/foo", "", null));
replay(store);
List<Message> messages = new ArrayList<Message>();
MessageStore messageStore = createStrictMock(MessageStore.class);
expect(messageStore.get(MessageScope.REQUEST)).andReturn(messages);
messageStore.clear(MessageScope.REQUEST);
messageStore.addAll(MessageScope.FLASH, messages);
replay(messageStore);
Status status = new StatusImpl("success", 200, "someFieldName");
StatusResult result = new StatusResult(ee, response, store);
result.execute(status);
verify(response);
}
@Test
public void headers() throws IOException, ServletException {
ExpressionEvaluator ee = createStrictMock(ExpressionEvaluator.class);
replay(ee);
HttpServletRequest request = createStrictMock(HttpServletRequest.class);
expect(request.getContextPath()).andReturn("");
replay(request);
HttpServletResponse response = createStrictMock(HttpServletResponse.class);
response.setStatus(200);
response.setHeader("foo", "bar");
response.setHeader("baz", "fred");
replay(response);
ActionInvocationStore store = createStrictMock(ActionInvocationStore.class);
expect(store.getCurrent()).andReturn(new ActionInvocation(null, null, "/foo", "", null));
replay(store);
List<Message> messages = new ArrayList<Message>();
MessageStore messageStore = createStrictMock(MessageStore.class);
expect(messageStore.get(MessageScope.REQUEST)).andReturn(messages);
messageStore.clear(MessageScope.REQUEST);
messageStore.addAll(MessageScope.FLASH, messages);
replay(messageStore);
Status status = new StatusImpl("success", 200, "", new HeaderImpl("foo", "bar"), new HeaderImpl("baz", "fred"));
StatusResult result = new StatusResult(ee, response, store);
result.execute(status);
verify(response);
}
public class StatusImpl implements Status {
private final String code;
private final int status;
private final String statusStr;
private final Header[] headers;
public StatusImpl(String code, int status, String statusStr, Header... headers) {
this.code = code;
this.status = status;
this.statusStr = statusStr;
this.headers = headers;
}
@Override
public Header[] headers() {
return headers;
}
@Override
public int status() {
return status;
}
@Override
public String statusStr() {
return statusStr;
}
public String code() {
return code;
}
@Override
public Class<? extends Annotation> annotationType() {
return Status.class;
}
}
public class HeaderImpl implements Header {
private final String name;
private final String value;
public HeaderImpl(String name, String value) {
this.name = name;
this.value = value;
}
@Override
public String name() {
return name;
}
@Override
public String value() {
return value;
}
@Override
public Class<? extends Annotation> annotationType() {
return Header.class;
}
}
}