在测试类之前和之后使ApplicationContext变脏

时间:2016-09-01 16:55:33

标签: java spring spring-test

我在Spring集成测试中有一个特定的类(比如MyTest),它在Spring组件上使用了PowerMock @PrepareForTest注释:@PrepareForTest(MyComponent.class)。这意味着PowerMock将通过一些修改加载此类。问题是,我的@ContextConfiguration是在超级类上定义的,它由MyTest扩展,ApplicationContext在不同的测试类之间缓存。现在,如果首先运行MyTest,它将具有正确的PowerMock版本MyComponent,但如果没有 - 测试将失败,因为上下文将被加载以进行另一次测试(没有@PrepareForTest)。

所以我想做的是在MyTest之前重新加载我的上下文。我可以通过

来做到这一点
@DirtiesContext(classMode = DirtiesContext.ClassMode.BEFORE_CLASS)

但是如果我还想在完成此测试后重新加载上下文怎么办?因此,如果没有PowerMock修改,我将再次清除MyComponent。有没有办法同时执行BEFORE_CLASSAFTER_CLASS

现在我用以下黑客做到了:

@FixMethodOrder(MethodSorters.NAME_ASCENDING)

在MyTest上然后

/**
 * Stub test to reload ApplicationContext before execution of real test methods of this class.
 */
@DirtiesContext(methodMode = DirtiesContext.MethodMode.BEFORE_METHOD)
@Test
public void aa() {
}

/**
 * Stub test to reload ApplicationContext after execution of real test methods of this class.
 */
@DirtiesContext(methodMode = DirtiesContext.MethodMode.AFTER_METHOD)
@Test
public void zz() {
}

我想知道是否有更漂亮的方法呢?

作为一个附带问题,是否可以仅重新加载某些bean而不是完整的上下文?

1 个答案:

答案 0 :(得分:10)

  

有没有办法同时做BEFORE_CLASS和AFTER_CLASS?

不,遗憾的是,@DirtiesContext不支持。

但是,您真正要说的是,您希望ApplicationContext的新MyTest与父测试类的上下文相同,但只有MyTest才有效。并且......您不希望影响为父测试类缓存的上下文。

因此,考虑到这一点,以下技巧应该可以胜任。

@RunWith(SpringJUnit4ClassRunner.class)
// Inherit config from parent and combine with local 
// static Config class to create a new context
@ContextConfiguration
@DirtiesContext
public class MyTest extends BaseTests {

    @Configuration
    static class Config {
        // No need to define any actual @Bean methods.
        // We only need to add an additional @Configuration
        // class so that we get a new ApplicationContext.
    }
}

替代@DirtiesContext

如果您希望在之前之后测试类弄脏上下文,则可以实现完全相同的自定义TestExecutionListener。例如,以下内容将起到作用。

import org.springframework.core.Ordered;
import org.springframework.test.annotation.DirtiesContext.HierarchyMode;
import org.springframework.test.context.TestContext;
import org.springframework.test.context.support.AbstractTestExecutionListener;

public class DirtyContextBeforeAndAfterClassTestExecutionListener
        extends AbstractTestExecutionListener {

    @Override
    public int getOrder() {
        return Ordered.HIGHEST_PRECEDENCE;
    }

    @Override
    public void beforeTestClass(TestContext testContext) throws Exception {
        testContext.markApplicationContextDirty(HierarchyMode.EXHAUSTIVE);
    }

    @Override
    public void afterTestClass(TestContext testContext) throws Exception {
        testContext.markApplicationContextDirty(HierarchyMode.EXHAUSTIVE);
    }

}

然后,您可以使用MyTest中的自定义侦听器,如下所示。

import org.springframework.test.context.TestExecutionListeners;
import org.springframework.test.context.TestExecutionListeners.MergeMode;

@TestExecutionListeners(
    listeners = DirtyContextBeforeAndAfterClassTestExecutionListener.class, 
    mergeMode = MergeMode.MERGE_WITH_DEFAULTS
)
public class MyTest extends BaseTest { /* ... */ }
  

作为一个附带问题,是否可以仅重新加载某些bean而不是完整的上下文?

不,这也是不可能的。

此致

Sam( Spring TestContext Framework的作者