com.greenpepper.interpreter.flow.dowith.DoWithRowSelector Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of greenpepper-remote-agent Show documentation
The newest version!
package com.greenpepper.interpreter.flow.dowith;

import com.greenpepper.Example;
import static com.greenpepper.GreenPepper.isAnInterpreter;
import com.greenpepper.TypeLoaderChain;
import com.greenpepper.annotation.Annotations;
import com.greenpepper.interpreter.flow.Row;
import com.greenpepper.interpreter.flow.RowSelector;
import com.greenpepper.reflect.Fixture;
import com.greenpepper.reflect.Type;
import com.greenpepper.reflect.TypeLoader;
import com.greenpepper.util.ExampleUtil;

import java.util.Collection;

/**
 * DoWithRowSelector class.
 *
 * @author oaouattara
 * @version $Id: $Id
 */
public class DoWithRowSelector implements RowSelector
{
    protected final Fixture fixture;

    private TypeLoader typeLoader;

    /**
     * Constructor for DoWithRowSelector.
     *
     * @param fixture a {@link com.greenpepper.reflect.Fixture} object.
     */
    public DoWithRowSelector(Fixture fixture)
    {
        this.fixture = fixture;
        this.typeLoader = new TypeLoaderChain(Row.class);
        searchPackage( getClass().getPackage().getName() );
        addSuffix("Row");
    }

    /**
     * searchPackage.
     *
     * @param name a {@link java.lang.String} object.
     */
    public void searchPackage(String name)
    {
        typeLoader.searchPackage(name);
    }

    /**
     * addSuffix.
     *
     * @param suffix a {@link java.lang.String} object.
     */
    public void addSuffix(String suffix)
    {
        typeLoader.addSuffix(suffix);
    }

    /* (non-Javadoc)
	 * @see com.greenpepper.interpreter.dowith.RowSelector#select(com.greenpepper.Example)
	 */
    /** {@inheritDoc} */
    public Row select(Example example)
    {
        if (isARow(identifier(example)))
            return instantiateRow(example);

        if (isAnInterpreter(identifier(example)))
            return new InterpretRow(fixture);

        return new DefaultRow(fixture);
    }

    private boolean isARow(String name)
    {
        Type type = loadRowType(name);
        return type != null && !type.getUnderlyingClass().equals(Row.class);
    }

    private Type loadRowType(String name)
    {
        return typeLoader.loadType(name);
    }

    private String identifier(Example row)
    {
        return ExampleUtil.contentOf(row.firstChild());
    }

    private Row instantiateRow(Example row)
    {
        Type rowClass = loadRowType(identifier(row));
        try
        {
            return rowClass.newInstance(fixture);
        }
        catch (Throwable throwable)
        {
            row.firstChild().annotate(Annotations.exception(throwable));
            return new SkipRow();
        }
    }

    /**
     * addSuffixes.
     *
     * @param suffixes a {@link java.util.Collection} object.
     */
    public void addSuffixes(Collection suffixes)
    {
        for (String suffix : suffixes)
        {
            addSuffix(suffix);
        }
    }

    /**
     * addPackages.
     *
     * @param packages a {@link java.util.Collection} object.
     */
    public void addPackages(Collection packages)
    {
        for (String aPackage : packages)
        {
            searchPackage(aPackage);
        }
    }
}