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

Question

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

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

@DirtiesContext(classMode = DirtiesContext.ClassMode.BEFORE_CLASS)

但是如果我还想在测试完成后重新加载上下文怎么办？所以我将再次拥有干净的 MyComponent 而无需修改 PowerMock。有没有办法同时做到 BEFORE_CLASS 和 AFTER_CLASS？

现在我用以下 hack 做到了:

@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 而不是完整的上下文？

Answer 1

Is there a way to do both BEFORE_CLASS and 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 的替代方案

如果你想让一个上下文在 before 和 after 一个测试类中都被弄脏，你可以实现一个自定义的 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 { /* ... */ }

As a side question, is it possible to reload only certain bean and not full context?

不，那也不可能。

问候，

Sam(Spring TestContext 框架的作者)