package org.litesoft.bo;

import java.util.HashMap;
import java.util.NoSuchElementException;

import org.litesoft.bo.BoAttribute.*;

public class MockBoMetaData implements BoMetaData
{
    private HashMap<String, BoAttribute> mAttributes = new HashMap<String, BoAttribute>();

    /**
     * @see #addAttribute(BoAttribute)
     * @see #addAttribute(String, AttributeType)
     */
    public MockBoMetaData()
    {
    }

    public MockBoMetaData(BoAttribute... pAttributes)
    {
        for (BoAttribute attribute : pAttributes)
        {
            addAttribute(attribute);
        }
    }

    public MockBoMetaData(String[] pAttributeNames, AttributeType[] pAttributeTypes)
    {
        // TODO: preserve order?
        assert pAttributeNames.length == pAttributeTypes.length;
        for (int i = 0; i < pAttributeNames.length; i++)
        {
            addAttribute(pAttributeNames[i], pAttributeTypes[i]);
        }
    }
    
    public void addAttribute(String pAttributeName, AttributeType pAttributeType)
    {
        addAttribute(new MockBoAttribute(pAttributeName, pAttributeType));
    }

    public void addAttribute(BoAttribute pAttribute)
    {
        mAttributes.put(pAttribute.getName(), pAttribute);
    }

    public String[] getAttributeNames()
    {
        return mAttributes.keySet().toArray(new String[mAttributes.size()]);
    }

    public BoAttribute[] getBoAttributes()
    {
        return mAttributes.values().toArray(new BoAttribute[mAttributes.size()]);
    }

    public BoAttribute getBoAttribute(String pAttributeName)
    {
        BoAttribute result = mAttributes.get(pAttributeName);

        if (result == null)
            throw new NoSuchElementException(pAttributeName);

        return result;
    }
}