CollectionTypeManufacturerImpl.java

package uk.co.jemos.podam.typeManufacturers;

import uk.co.jemos.podam.api.AttributeMetadata;
import uk.co.jemos.podam.api.DataProviderStrategy;
import uk.co.jemos.podam.common.ManufacturingContext;

import java.lang.reflect.Type;
import java.util.ArrayList;
import java.util.Collection;
import java.util.HashSet;
import java.util.LinkedList;
import java.util.Map;

/**
 * Default collection type manufacturer.
 *
 * @since 7.0.0.RELEASE
 */
public class CollectionTypeManufacturerImpl extends AbstractTypeManufacturer<Collection<Object>> {

    @Override
    public Collection<Object> getType(DataProviderStrategy strategy,
            AttributeMetadata attributeMetadata,
            ManufacturingContext manufacturingCtx) {

        Class<?> collectionType = attributeMetadata.getAttributeType();
        Collection<Object> retValue = null;

        // Default list and set are ArrayList and HashSet. If users
        // wants a particular collection flavour they have to initialise
        // the collection
        if (collectionType.isAssignableFrom(ArrayList.class)) {
            // List
            retValue = new ArrayList<Object>();
        } else if (collectionType.isAssignableFrom(HashSet.class)) {
            // Set
            retValue = new HashSet<Object>();
        } else if (collectionType.isAssignableFrom(LinkedList.class)) {
            // Queue
            retValue = new LinkedList<Object>();
        }
        return retValue;
    }
}