gistfile1.java

package lv.craftsmans;

import org.junit.Rule;
import org.springframework.beans.factory.config.AutowireCapableBeanFactory;
import org.springframework.test.context.TestContext;
import org.springframework.test.context.support.DependencyInjectionTestExecutionListener;
import org.springframework.util.ReflectionUtils;

import java.lang.reflect.Field;

public class RuleSupportingDependencyInjectionTestExecutionListener extends DependencyInjectionTestExecutionListener {

    protected void injectDependencies(final TestContext testContext) throws Exception {

        AutowireCapableBeanFactory beanFactory = testContext.getApplicationContext().getAutowireCapableBeanFactory();

        Object bean = testContext.getTestInstance();
        autowireTestInstance(testContext, beanFactory, bean);
        autowireRuleFields(testContext, beanFactory, bean);

        testContext.removeAttribute(REINJECT_DEPENDENCIES_ATTRIBUTE);
    }

    private void autowireTestInstance(TestContext testContext, AutowireCapableBeanFactory beanFactory, Object bean) {
        beanFactory.autowireBeanProperties(bean, AutowireCapableBeanFactory.AUTOWIRE_NO, false);
        beanFactory.initializeBean(bean, testContext.getTestClass().getName());
    }

    private void autowireRuleFields(TestContext testContext, AutowireCapableBeanFactory beanFactory, Object bean) throws IllegalAccessException {
        ReflectionUtils.doWithFields(bean.getClass(), field -> {
            if (field.isAnnotationPresent(Rule.class)) {
                Object fieldValue = field.get(bean);
                autowireRuleField(testContext, beanFactory, field, fieldValue);
            }
        });
    }

    private void autowireRuleField(TestContext testContext, AutowireCapableBeanFactory beanFactory, Field field, Object fieldValue) {
        beanFactory.autowireBeanProperties(fieldValue, AutowireCapableBeanFactory.AUTOWIRE_NO, false);
        beanFactory.initializeBean(fieldValue, testContext.getTestClass().getName() + "#" + field.getName());
    }
}