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

Question

我在 Spring 集成测试中有一个特定的类（比如说

MyTest

），它在 Spring 组件上使用 PowerMock

@PrepareForTest

注释：

@PrepareForTest(MyComponent.class)

。这意味着 PowerMock 将加载此类并进行一些修改。问题是，我的

@ContextConfiguration

是在由

MyTest

扩展的超类上定义的，并且

ApplicationContext

缓存在不同的测试类之间。现在，如果首先运行

MyTest

，它将具有正确的 PowerMock 版本

MyComponent

，但如果没有 - 测试将失败，因为将加载上下文以进行另一个测试（没有 @PrepareForTest）。

所以我想做的是在

MyTest

之前重新加载我的上下文。我可以通过

做到这一点

@DirtiesContext(classMode = DirtiesContext.ClassMode.BEFORE_CLASS)

但是如果我还想在测试完成后重新加载上下文怎么办？所以我将再次拥有干净的

MyComponent

，而无需PowerMock修改。有没有办法同时做到

BEFORE_CLASS

和

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

Answer 1

有没有办法同时执行 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 的替代方案

如果您想让上下文在测试类的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 { /* ... */ }

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

不，这也是不可能的。

问候，

Sam（Spring TestContext 框架的作者）

Answer 2

我最近遇到了这种需求，我发现我正在破解所有测试用例。我得出了与您相同的结论，将 @DirtiesContext(classMode = DirtiesContext.ClassMode.AFTER_CLASS) 放在测试类和 @DirtiesContext(methodMode = DirtiesContext.MethodMode.BEFORE_METHOD) 和每个测试方法上。但后来我查看了测试类的常见执行顺序，并将 @DirtiesContext(classMode = DirtiesContext.ClassMode.AFTER_CLASS) 放在我正在敲击的测试类之前的测试类上，这解决了我的问题。

我认为这是有道理的，它是一个测试类或方法，它会弄脏上下文，因此它应该在该类或方法之后被弄脏。所以我得出的结论是，特定的测试类以一种不明显的方式污染了我的上下文。