1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22 package org.apache.struts2.dojo.views.jsp.ui;
23
24 import java.util.HashMap;
25 import java.util.Map;
26
27 import javax.servlet.ServletResponse;
28 import javax.servlet.http.HttpServletRequest;
29 import javax.servlet.http.HttpSession;
30
31 import com.mockobjects.servlet.MockPageContext;
32
33
34 /***
35 */
36 public class StrutsMockPageContext extends MockPageContext {
37
38 private Map attributes = new HashMap();
39 private ServletResponse response;
40
41
42 public void setAttribute(String s, Object o) {
43 if ((s == null) || (o == null)) {
44 throw new NullPointerException("PageContext does not accept null attributes");
45 }
46
47 this.attributes.put(s, o);
48 }
49
50 public Object getAttribute(String key) {
51 return attributes.get(key);
52 }
53
54 public Object getAttributes(String key) {
55 return this.attributes.get(key);
56 }
57
58 public void setResponse(ServletResponse response) {
59 this.response = response;
60 }
61
62 public ServletResponse getResponse() {
63 return response;
64 }
65
66 public HttpSession getSession() {
67 HttpSession session = super.getSession();
68
69 if (session == null) {
70 session = ((HttpServletRequest) getRequest()).getSession(true);
71 }
72
73 return session;
74 }
75
76 public Object findAttribute(String s) {
77 return attributes.get(s);
78 }
79
80 public void removeAttribute(String key) {
81 this.attributes.remove(key);
82 }
83 }