View Javadoc

    1   /*
    2    * $Id: StrutsMockPageContext.java 471756 2006-11-06 15:01:43Z husted $
    3    *
    4    * Licensed to the Apache Software Foundation (ASF) under one
    5    * or more contributor license agreements.  See the NOTICE file
    6    * distributed with this work for additional information
    7    * regarding copyright ownership.  The ASF licenses this file
    8    * to you under the Apache License, Version 2.0 (the
    9    * "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   */
    35  public class StrutsMockPageContext extends MockPageContext {
    36
    37      private Map attributes = new HashMap();
    38      private ServletResponse response;
    39
    40
    41      public void setAttribute(String s, Object o) {
    42          if ((s == null) || (o == null)) {
    43              throw new NullPointerException("PageContext does not accept null attributes");
    44          }
    45
    46          this.attributes.put(s, o);
    47      }
    48
    49      public Object getAttribute(String key) {
    50          return attributes.get(key);
    51      }
    52
    53      public Object getAttributes(String key) {
    54          return this.attributes.get(key);
    55      }
    56
    57      public void setResponse(ServletResponse response) {
    58          this.response = response;
    59      }
    60
    61      public ServletResponse getResponse() {
    62          return response;
    63      }
    64
    65      public HttpSession getSession() {
    66          HttpSession session = super.getSession();
    67
    68          if (session == null) {
    69              session = ((HttpServletRequest) getRequest()).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  }