有没有一种方法可以针对多个接口实现(DataSources)运行TestNG测试

Question

我正在寻找一种针对多个DataSource s运行一些TestNG测试的方法。我已经看到了TestNG的DataProvider选项，但它们并不能完全满足我的要求。我想对我的MyBatis映射器运行一些非常简单的测试，主要是为了确保我没有任何简单的语法错误、拼写错误等。现在，我有一个像这样的基本测试类：

public class MapperTestBaseH2 {
    protected SqlSessionManager sessionManager;
    @BeforeClass
    public void beforeClass() throws SQLException, LiquibaseUpdateException {
        JdbcDataSource h2DataSource = new JdbcDataSource();
        h2DataSource.setURL("jdbc:h2:mem:test-db;DB_CLOSE_DELAY=-1");
        sessionManager = MyBatisUtils.createSqlSessionManager(h2DataSource);
        LiquibaseUpdater.update(h2DataSource.getConnection(), LiquibaseUpdater.DEFAULT_CONTEXTS);
    }
    @BeforeMethod
    public void beforeMethod() {
        sessionManager.startManagedSession();
    }
    @AfterMethod
    public void afterMethod() {
        sessionManager.rollback();
    }
}

然后我为每个映射器创建一个测试：

public class SequenceMapperTest extends MapperTestBaseH2 {
    private SequenceMapper sequenceMapper;
    @BeforeClass
    @Override
    public void beforeClass() throws SQLException, LiquibaseUpdateException {
        super.beforeClass();
        sequenceMapper = super.sessionManager.getMapper(SequenceMapper.class);
    }
    @Test
    public void selectBidSequenceTest() {
        sequenceMapper.selectSequence(Sequences.BID_SEQ);
    }
    @Test
    public void updateBidSequenceTest() {
        sequenceMapper.updateSequence(sequenceMapper.selectSequence(Sequences.BID_SEQ));
    }
}

我想做的是运行每个测试类两次（或两次以上），每次使用不同的SqlSessionManager。我可以使用TestNG的DataProvider来运行具有多个SqlSessionManager的每个测试，但之后我必须启动&在每个测试中而不是在@BeforeMethod中回滚会话。

有没有一种简单的方法可以实现我想要的？

Answer 1

我想我在偶然发现这篇文章后就明白了。诀窍是使用@Factory和@DataProvider的组合。首先，我创建了一个抽象基类，用于管理提供者和事务。

public abstract class MyBatisTestBase {
    public abstract SqlSessionManager getSessionManager();
    @BeforeMethod
    public void beforeMethod() {
        getSessionManager().startManagedSession();
    }
    @AfterMethod
    public void afterMethod() {
        getSessionManager().rollback();
    }
    @DataProvider(name = "sql-session-manager-provider")
    public static Object[][] sqlSessionManagerProvider() throws SQLException, LiquibaseUpdateException {
        return new Object[][] {
                {createDerbySessionManager()},
                {createH2SessionManager()}
        };
    }
    private static SqlSessionManager createDerbySessionManager() throws SQLException, LiquibaseUpdateException {
        EmbeddedDataSource40 dataSource = new EmbeddedDataSource40();
        dataSource.setDatabaseName("memory:test-db");
        dataSource.setCreateDatabase("create");
        LiquibaseUtils.update(dataSource.getConnection(), LiquibaseUtils.DEFAULT_CONTEXTS);
        return MyBatisUtils.createSqlSessionManager(dataSource);
    }
    private static SqlSessionManager createH2SessionManager() throws SQLException, LiquibaseUpdateException {
        JdbcDataSource dataSource = new JdbcDataSource();
        dataSource.setURL("jdbc:h2:mem:test-db;DB_CLOSE_DELAY=-1");
        LiquibaseUtils.update(dataSource.getConnection(), LiquibaseUtils.DEFAULT_CONTEXTS);
        return MyBatisUtils.createSqlSessionManager(dataSource);
    }
}

然后在扩展类的构造函数上使用@Factory注释。这具有从我的@DataProvider为每个SqlSessionManager创建一个测试类实例的效果。

public class SequenceMapperTest extends MyBatisTestBase {
    private final SqlSessionManager sessionManager;
    private final SequenceMapper sequenceMapper;
    @Factory(dataProvider = "sql-session-manager-provider")
    public SequenceMapperTest(SqlSessionManager sessionManager) {
        this.sessionManager = sessionManager;
        this.sequenceMapper = sessionManager.getMapper(SequenceMapper.class);
    }
    @Test
    public void selectBidSequenceTest() {
        sequenceMapper.selectSequence(Sequences.BID_SEQ);
    }
    @Test
    public void updateBidSequenceTest() {
        sequenceMapper.updateSequence(sequenceMapper.selectSequence(Sequences.BID_SEQ));
    }
    @Override
    public SqlSessionManager getSessionManager() {
        return this.sessionManager;
    }
}

当我运行测试时，会创建两个SequenceMapperTest实例，其中一个用于我在@DataProvider中设置的每个SqlSessionManager。到目前为止，它似乎可以工作，但我的IDE（IntelliJ IDEA）在分割每个单独测试的日志输出时遇到了一些问题。