2009/05/20 - Apache Shale has been retired.

   /*
    * Licensed to the Apache Software Foundation (ASF) under one or more
    * contributor license agreements.  See the NOTICE file distributed with
    * this work for additional information regarding copyright ownership.
    * The ASF licenses this file to you 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.apache.shale.test.mock;
  
  import java.util.ArrayList;
  import java.util.Collection;
  import java.util.Enumeration;
  import java.util.HashSet;
  import java.util.Iterator;
  import java.util.List;
  import java.util.Map;
  import java.util.Set;
  
  import javax.servlet.ServletContext;
  
  /***
   * <p>Mock impementation of <code>Map</code> for the application scope
   * attributes managed by {@link MockExternalContext}.</p>
   *
   * $Id$
   */
  
  class MockApplicationMap implements Map {
  
  
      // ------------------------------------------------------------ Constructors
  
  
      /***
       * <p>Construct a default instance.</p>
       *
       * @param context ServletContext whose attributes we are wrapping
       */
      public MockApplicationMap(ServletContext context) {
  
          this.context = context;
  
      }
  
  
      // ----------------------------------------------------- Mock Object Methods
  
  
      // ------------------------------------------------------ Instance Variables
  
  
      /***
       * <p>The <code>ServletContext</code> instance we are wrapping.</p>
       */
      private ServletContext context = null;
  
  
      // ------------------------------------------------------------- Map Methods
  
  
      /*** {@inheritDoc} */
      public void clear() {
  
          Iterator keys = keySet().iterator();
          while (keys.hasNext()) {
              context.removeAttribute((String) keys.next());
          }
  
      }
  
  
      /*** {@inheritDoc} */
      public boolean containsKey(Object key) {
  
          return context.getAttribute(key(key)) != null;
  
      }
  
  
      /*** {@inheritDoc} */
      public boolean containsValue(Object value) {
  
          if (value == null) {
              return false;
          }
          Enumeration keys = context.getAttributeNames();
          while (keys.hasMoreElements()) {
              Object next = context.getAttribute((String) keys.nextElement());
              if (next == value) {
                 return true;
             }
         }
         return false;
 
     }
 
 
     /*** {@inheritDoc} */
     public Set entrySet() {
 
         Set set = new HashSet();
         Enumeration keys = context.getAttributeNames();
         while (keys.hasMoreElements()) {
             set.add(context.getAttribute((String) keys.nextElement()));
         }
         return set;
 
     }
 
 
     /*** {@inheritDoc} */
     public boolean equals(Object o) {
 
         return context.equals(o);
 
     }
 
 
     /*** {@inheritDoc} */
     public Object get(Object key) {
 
         return context.getAttribute(key(key));
 
     }
 
 
     /*** {@inheritDoc} */
     public int hashCode() {
 
         return context.hashCode();
 
     }
 
 
     /*** {@inheritDoc} */
     public boolean isEmpty() {
 
         return size() < 1;
 
     }
 
 
     /*** {@inheritDoc} */
     public Set keySet() {
 
         Set set = new HashSet();
         Enumeration keys = context.getAttributeNames();
         while (keys.hasMoreElements()) {
             set.add(keys.nextElement());
         }
         return set;
 
     }
 
 
     /*** {@inheritDoc} */
     public Object put(Object key, Object value) {
 
         if (value == null) {
             return remove(key);
         }
         String skey = key(key);
         Object previous = context.getAttribute(skey);
         context.setAttribute(skey, value);
         return previous;
 
     }
 
 
     /*** {@inheritDoc} */
     public void putAll(Map map) {
 
         Iterator keys = map.keySet().iterator();
         while (keys.hasNext()) {
             String key = (String) keys.next();
             context.setAttribute(key, map.get(key));
         }
 
     }
 
 
     /*** {@inheritDoc} */
     public Object remove(Object key) {
 
         String skey = key(key);
         Object previous = context.getAttribute(skey);
         context.removeAttribute(skey);
         return previous;
 
     }
 
 
     /*** {@inheritDoc} */
     public int size() {
 
         int n = 0;
         Enumeration keys = context.getAttributeNames();
         while (keys.hasMoreElements()) {
             keys.nextElement();
             n++;
         }
         return n;
 
     }
 
 
     /*** {@inheritDoc} */
     public Collection values() {
 
         List list = new ArrayList();
         Enumeration keys = context.getAttributeNames();
         while (keys.hasMoreElements()) {
             list.add(context.getAttribute((String) keys.nextElement()));
         }
         return list;
 
     }
 
 
     /***
      * <p>Return the specified key, converted to a String.</p>
      *
      * @param key The key to convert
      */
     private String key(Object key) {
 
         if (key == null) {
             throw new IllegalArgumentException();
         } else if (key instanceof String) {
             return (String) key;
         } else {
             return key.toString();
         }
 
     }
 
 
 }