com.mockrunner.mock.web.MockFunctionMapper Maven / Gradle / Ivy

Go to download
package com.mockrunner.mock.web;

import java.lang.reflect.Method;
import java.util.HashMap;
import java.util.Map;

import javax.servlet.jsp.el.FunctionMapper;

/**
 * Mock implementation of FunctionMapper.
 */
public class MockFunctionMapper implements FunctionMapper
{
    private Map functions = new HashMap();
    
    /**
     * Adds a function for the specified prefix and name.
     * @param prefix the prefix of the function
     * @param localName the name of the function
     * @param function the function as Method
     */
    public void addFunction(String prefix, String localName, Method function)
    {
        FunctionMappingEntry entry = new FunctionMappingEntry(prefix, localName);
        functions.put(entry, function);
    }
    
    /**
     * Clears all functions.
     */
    public void clearFunctions()
    {
        functions.clear();
    }
    
    public Method resolveFunction(String prefix, String localName)
    {
        FunctionMappingEntry entry = new FunctionMappingEntry(prefix, localName);
        return (Method)functions.get(entry);
    }

    private class FunctionMappingEntry
    {
        private String prefix;
        private String localName;
        
        public FunctionMappingEntry(String prefix, String localName)
        {
            this.prefix = prefix;
            this.localName = localName;
        }
        
        public String getLocalName()
        {
            return localName;
        }

        public String getPrefix()
        {
            return prefix;
        }
        
        public boolean equals(Object obj)
        {
            if(!(obj instanceof FunctionMappingEntry)) return false;
            if(obj == this) return true;
            FunctionMappingEntry otherEntry = (FunctionMappingEntry)obj;
            boolean prefixOk = false;
            boolean nameOk = false;
            if(null == prefix)
            {
                prefixOk = (otherEntry.getPrefix() == null);
            }
            else
            {
                prefixOk = prefix.equals(otherEntry.getPrefix());
            }
            if(null == localName)
            {
                nameOk = (otherEntry.getLocalName() == null);
            }
            else
            {
                nameOk = localName.equals(otherEntry.getLocalName());
            }
            return (prefixOk && nameOk);
        }

        public int hashCode()
        {
            int hashCode = 17;
            if(null != prefix) hashCode = (31 * hashCode) + prefix.hashCode();
            if(null != localName) hashCode = (31 * hashCode) + localName.hashCode();
            return hashCode;
        }
    }
}