Question

我正在寻找一种方法来针对多个DataSource运行一些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的每个测试，但是我必须开始＆amp;在每个测试中回滚会话而不是@BeforeMethod。

有没有一种简单的方法可以实现我的目标？

Answer 1

我认为我在遇到this post.后遇到了这个问题。诀窍是使用@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注释。这具有从SqlSessionManager每@DataProvider创建一个测试类实例的效果。

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的两个实例，我在SqlSessionManager中设置的每个@DataProvider一个实例。到目前为止，它似乎正在工作，但我的IDE（IntelliJ IDEA）在分割每个单独测试的日志输出时遇到了一些麻烦。

有没有办法可以针对多个接口实现（DataSources）运行TestNG测试？

1 个答案: