01: /*
02: * $Id: StrutsMockPageContext.java 471756 2006-11-06 15:01:43Z husted $
03: *
04: * Licensed to the Apache Software Foundation (ASF) under one
05: * or more contributor license agreements. See the NOTICE file
06: * distributed with this work for additional information
07: * regarding copyright ownership. The ASF licenses this file
08: * to you under the Apache License, Version 2.0 (the
09: * "License"); you may not use this file except in compliance
10: * with the License. You may obtain a copy of the License at
11: *
12: * http://www.apache.org/licenses/LICENSE-2.0
13: *
14: * Unless required by applicable law or agreed to in writing,
15: * software distributed under the License is distributed on an
16: * "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY
17: * KIND, either express or implied. See the License for the
18: * specific language governing permissions and limitations
19: * under the License.
20: */
21: package org.apache.struts2.views.jsp;
22:
23: import java.util.HashMap;
24: import java.util.Map;
25:
26: import javax.servlet.ServletResponse;
27: import javax.servlet.http.HttpServletRequest;
28: import javax.servlet.http.HttpSession;
29:
30: import com.mockobjects.servlet.MockPageContext;
31:
32: /**
33: */
34: public class StrutsMockPageContext extends MockPageContext {
35:
36: private Map attributes = new HashMap();
37: private ServletResponse response;
38:
39: public void setAttribute(String s, Object o) {
40: if ((s == null) || (o == null)) {
41: throw new NullPointerException(
42: "PageContext does not accept null attributes");
43: }
44:
45: this .attributes.put(s, o);
46: }
47:
48: public Object getAttribute(String key) {
49: return attributes.get(key);
50: }
51:
52: public Object getAttributes(String key) {
53: return this .attributes.get(key);
54: }
55:
56: public void setResponse(ServletResponse response) {
57: this .response = response;
58: }
59:
60: public ServletResponse getResponse() {
61: return response;
62: }
63:
64: public HttpSession getSession() {
65: HttpSession session = super .getSession();
66:
67: if (session == null) {
68: session = ((HttpServletRequest) getRequest())
69: .getSession(true);
70: }
71:
72: return session;
73: }
74:
75: public Object findAttribute(String s) {
76: return attributes.get(s);
77: }
78:
79: public void removeAttribute(String key) {
80: this.attributes.remove(key);
81: }
82: }
|