com.tngtech.jgiven.integration.spring.SpringScenarioExecutor Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of jgiven-spring Show documentation

Module for using Spring dependency injection together with JGiven

There is a newer version: 2.0.1

package com.tngtech.jgiven.integration.spring;

import org.slf4j.Logger;
import org.slf4j.LoggerFactory;
import org.springframework.aop.Advisor;
import org.springframework.aop.framework.Advised;
import org.springframework.beans.factory.NoSuchBeanDefinitionException;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.context.ApplicationContext;
import org.springframework.util.ClassUtils;

import com.tngtech.jgiven.impl.ScenarioExecutor;
import com.tngtech.jgiven.impl.StandaloneScenarioExecutor;

/**
 * Main class of JGiven for executing scenarios with spring support.
 * See below on how to configure this bean.
 * 
 * Sample Configuration:
 * 
 *	{@literal @}Bean
 *	{@literal @}Scope("prototype")
 *	public SpringScenarioExecutor springScenarioExecutor() {
 *	    return new SpringScenarioExecutor();
 *	}
 * 
 * 
 * The SpringScenarioExecutor is stateful, and thus should use "prototype" scope
 * @since 0.8.0
 */
public class SpringScenarioExecutor extends StandaloneScenarioExecutor implements ScenarioExecutor {

    private static final Logger log = LoggerFactory.getLogger( SpringScenarioExecutor.class );

    @Autowired
    private ApplicationContext applicationContext;

    @Override
    public  T createStageClass( Class stepsClass ) {
        try {
            T bean = applicationContext.getBean(stepsClass);
            Advised advised = (Advised)bean;
            Advisor[] advisors = advised.getAdvisors();
            for (Advisor advisor : advisors) {
                if (advisor.getAdvice() instanceof SpringStepMethodInterceptor ) {
                    SpringStepMethodInterceptor interceptor = (SpringStepMethodInterceptor)advisor.getAdvice();
                    interceptor.setScenarioMethodHandler(this.methodHandler);
                    interceptor.setStackDepth(this.stackDepth);
                    interceptor.enableMethodHandling(true);
                }
            }
            return bean;
        } catch (NoSuchBeanDefinitionException nbe) {
            return super.createStageClass(stepsClass);
        } catch (ClassCastException cce) {
            log.error("class " + ClassUtils.getShortName(stepsClass) + " is not advised with SpringStepMethodInterceptor. Falling back to cglib based proxy, strange things may happen.");
            return super.createStageClass(stepsClass);
        }
    }

    @Override
    public StageState getStageState( Object stage ) {
        StageState stageState = stages.get( stage.getClass());
        return stageState != null ? stageState : super.getStageState(stage);
    }

}