01: package com.mockrunner.base;
02:
03: import com.mockrunner.mock.web.WebMockObjectFactory;
04:
05: /**
06: * This class provides some convenience methods for
07: * request and session attribute handling
08: */
09: public abstract class WebTestModule {
10: private WebMockObjectFactory mockFactory;
11:
12: public WebTestModule(WebMockObjectFactory mockFactory) {
13: this .mockFactory = mockFactory;
14: }
15:
16: /**
17: * Adds an empty request parameter. Same as
18: * <code>addRequestParameter(key, "")</code>.
19: * @param key the request key
20: */
21: public void addRequestParameter(String key) {
22: addRequestParameter(key, "");
23: }
24:
25: /**
26: * Adds a request parameter.
27: * @param key the request key
28: * @param value the request value
29: */
30: public void addRequestParameter(String key, String value) {
31: mockFactory.getMockRequest().setupAddParameter(key, value);
32: }
33:
34: /**
35: * Adds several request parameters.
36: * @param key the
37: * @param values the request values
38: */
39: public void addRequestParameter(String key, String[] values) {
40: mockFactory.getMockRequest().setupAddParameter(key, values);
41: }
42:
43: /**
44: * Returns the request parameter for the specified key
45: * @param key the request key
46: * @return the parameter
47: */
48: public String getRequestParameter(String key) {
49: return mockFactory.getWrappedRequest().getParameter(key);
50: }
51:
52: /**
53: * Returns the request attribute for the specified key
54: * @param key the request key
55: * @return the attribute
56: */
57: public Object getRequestAttribute(String key) {
58: return mockFactory.getWrappedRequest().getAttribute(key);
59: }
60:
61: /**
62: * Sets the request attribute for the specified key
63: * @param key the request key
64: * @param value the value
65: */
66: public void setRequestAttribute(String key, Object value) {
67: mockFactory.getWrappedRequest().setAttribute(key, value);
68: }
69:
70: /**
71: * Returns the session attribute for the specified key
72: * @param key the session key
73: * @return the attribute
74: */
75: public Object getSessionAttribute(String key) {
76: return mockFactory.getMockSession().getAttribute(key);
77: }
78:
79: /**
80: * Sets the session attribute for the specified key
81: * @param key the session key
82: * @param value the value
83: */
84: public void setSessionAttribute(String key, Object value) {
85: mockFactory.getMockSession().setAttribute(key, value);
86: }
87: }
|