StepClassFilterTest.java

/*
 *  Copyright Technophobia Ltd 2012
 *
 *   This file is part of Substeps.
 *
 *    Substeps is free software: you can redistribute it and/or modify
 *    it under the terms of the GNU Lesser General Public License as published by
 *    the Free Software Foundation, either version 3 of the License, or
 *    (at your option) any later version.
 *
 *    Substeps is distributed in the hope that it will be useful,
 *    but WITHOUT ANY WARRANTY; without even the implied warranty of
 *    MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
 *    GNU Lesser General Public License for more details.
 *
 *    You should have received a copy of the GNU Lesser General Public License
 *    along with Substeps.  If not, see <http://www.gnu.org/licenses/>.
 */
package com.technophobia.substeps.runner.runtime;

import static org.junit.Assert.assertFalse;
import static org.junit.Assert.assertTrue;

import org.junit.Before;
import org.junit.Test;

import com.google.common.base.Predicate;
import com.technophobia.substeps.runner.runtime.fake.AnnotatedStepClassFake;
import com.technophobia.substeps.runner.runtime.fake.NonAnnotatedStepClassFake;

public class StepClassFilterTest {

    private Predicate<Class<?>> stepClassFilter;


    @Before
    public void initialise() {
        stepClassFilter = new StepClassFilter();
    }


    @Test
    public void findsClassesWithAnnotatedStepsCorrectly() {
        assertTrue(stepClassFilter.apply(AnnotatedStepClassFake.class));
    }


    @Test
    public void filtersOutNonStepClasses() {
        assertFalse(stepClassFilter.apply(NonAnnotatedStepClassFake.class));
    }
}