java - 将 junit 监听器添加到 SpringJUnit4ClassRunner

Question

我有一个使用 SpringJUnit4ClassRunner 运行的单元测试，如下所示:

@RunWith(SpringJUnit4ClassRunner.class)
@ContextConfiguration(locations = "classpath:aConfig.xml")
public class TestService
{
     @Resource
     EmbeddedMysqlDatabase mysqlDB;

     ...
}

我有一个在单元测试中使用的嵌入式数据库，我想在运行完所有测试后将其关闭。我知道在单元测试中嵌入数据库不是通常/好的做法，但在这种特殊情况下，这非常有用。

@AfterClass 不是一个选项，因为它必须是静态的，而且我的数据库实例是由 spring 注入(inject)的。不能注入(inject)静态成员。

我如何通过监听器或任何其他方式做到这一点？

谢谢。

Answer 1

您可以使用@TestExecutionListeners。像这样:

public class ShutdownExecutionListener extends AbstractTestExecutionListener {
    @Override 
    public void beforeTestClass(TestContext testContext) throwsException {
    }      
    @Override 
    public void afterTestClass(TestContext testContext) throws Exception{
        EmbeddedMysqlDatabase mysqlDB= 
            (EmbeddedMysqlDatabase)testContext.getApplicationContext().getBean(mysqlDB);
        mysqlDB.shutdown();     
    } 
}

在你的测试中:

@RunWith(SpringJUnit4ClassRunner.class)
@ContextConfiguration(locations = "classpath:aConfig.xml")
@TestExecutionListeners(listeners = ShutdownExecutionListener.class)
public class TestService
{
     @Resource
     EmbeddedMysqlDatabase mysqlDB;

     ...
}