1

我想使用 Mockito（如果需要，还可以使用 PowerMockito）测试我的 DAO 方法，但我不知道该怎么做。调用静态方法的最大问题（MySQLStationDAO 中的 MySQLDAOFactory.getConnection()）。你能帮助我吗？

我通过这种方式获得连接：

public class MySQLDAOFactory extends DAOFactory {     
        public static Connection getConnection() throws DAOException {
            Connection con = null;
            try {
                con = getDataSource().getConnection();
            } catch (SQLException e) {
                throw new DAOException(Messages.CANNOT_OBTAIN_CONNECTION, e);
            }
            return con;
        }

这是一个DAO方法：

public class MySQLStationDAO implements StationDAO {
    @Override
    public List<Station> getAllStations() throws DAOException {
        List<Station> stations = new ArrayList<>();
        Connection con = null;
        Statement stmt = null;
        ResultSet rs = null;
        try {
            con = MySQLDAOFactory.getConnection();
            stmt = con.createStatement();
            rs = stmt.executeQuery(MySQLQueries.SQL_GET_ALL_STATIONS);
            while (rs.next()) {
                stations.add(extractStation(rs));
            }
        } catch (SQLException e) {
            throw new DAOException(Messages.CANNOT_OBTAIN_ALL_STATIONS, e);
        } finally {
            MySQLDAOFactory.close(con, stmt, rs);
        }
        return stations;
    }
4

2 回答 2

0

正如你所说，你的问题是当你调用 MySQLDAOFactory.getConnection(); 在测试方面，您想要测试您的 MySQLStationDAO 类。这是您的 SUT（被测系统）。这意味着您必须模拟您的 SUT 拥有的所有依赖项。在这种情况下，MySQLDAOFactory。

为此，您可以使用 Mockito 轻松模拟该类并存根 MySQLDAOFactory 提供的方法。一个例子是

    package com.iseji.app.dao;

import junit.framework.Assert;
import org.junit.Before;    
import org.junit.Test;
import org.junit.runner.RunWith;
import org.mockito.runners.MockitoJUnitRunner;

import java.sql.Connection;

import static org.mockito.Matchers.anyString;
import static org.mockito.Mockito.mock;
import static org.mockito.Mockito.when;

@RunWith(MockitoJUnitRunner.class)
public class TestMySqlDaoFactory {

    MySqlDaoFactory mySqlDaoFactory;
    Connection connection;

    @Before
    public void setUp() throws DAOException {
        mySqlDaoFactory = mock(MySqlDaoFactory.class);
        connection = mock(Connection.class);
    }

    @Test(expected = DAOException.class)
    public void testEmptyUrlGetsDaoException() throws DAOException {
        when(mySqlDaoFactory.getConnection(null)).thenThrow(new DAOException());
        mySqlDaoFactory.getConnection(null);
    }

    @Test
    public void testFullUrlGetsConnection() throws DAOException {
        when(mySqlDaoFactory.getConnection(anyString())).thenReturn(connection);
        Assert.assertEquals(mySqlDaoFactory.getConnection(anyString()), connection);
    }
}

如您所见，您可以指定 DaoFactory 的行为。这将隔离您的 Dao 类，这是您要测试的类。

于 2015-08-14T23:55:30.477 回答
0

  1. JUnit：@RunWith(PowerMockRunner.class)在类级别使用。

    TestNG：让你的测试类扩展PowerMockTestCase。

  2. 在班级级别使用@PrepareForTest(MySQLDAOFactory.class)，以指示 PowerMock 准备MySQLDAOFactory班级进行测试。

  3. 用于PowerMockito.mockStatic(MySQLDAOFactory.class)模拟类的所有方法MySQLDAOFactory。

    也可以使用部分模拟：

    PowerMockito.stub(PowerMockito.method(MySQLDAOFactory.class, "getConnection")).toReturn(Mockito.mock(Connection.class));

  4. 使用类似的东西来存根getConnection()：

    Connection mockConnection = Mockito.mock(Connection.class); Mockito.when(MySQLDAOFactory.getConnection()).thenReturn(mockConnection);

  5. getAllStations()在 的真实实例上执行MySQLStationDAO，因为您正在测试MySQLStationDAO类。

  6. 如果要验证该getConnection()方法是否已被调用，请使用类似的方法：

    PowerMockito.verifyStatic(); MySQLDAOFactory.getConnection();

    但是，请阅读Mockito.verify(T) javadoc，了解为什么建议使用存根或验证调用，而不是两者兼而有之。

一般来说，您可能需要查阅Mockito 文档和PowerMockito 文档以获取更多信息。

使用 JUnit 4.11、Mockito 1.9.5 和 PowerMock (PowerMockito) 1.5.6 创建的完整示例（请注意版本，因为存在很多兼容性问题）：

@RunWith(PowerMockRunner.class)
@PrepareForTest(MySQLDAOFactory.class)
public class MySQLDAOFactoryTest {

    private StationDAO stationDAO;

    @Mock
    private Connection mockConnection;

    @Mock
    private Statement mockStatement;

    @Mock
    private ResultSet mockResultSet;

    @Before
    public void setUp() {
        stationDAO = new MySQLStationDAO();
    }

    @Test
    public void testGetAllStations_StatementCreated() throws DAOException, SQLException {
        // given
        PowerMockito.mockStatic(MySQLDAOFactory.class);
        Mockito.when(MySQLDAOFactory.getConnection()).thenReturn(mockConnection);
        Mockito.when(mockConnection.createStatement()).thenReturn(mockStatement);
        Mockito.when(mockStatement.executeQuery(anyString())).thenReturn(mockResultSet);

        // when
        stationDAO.getAllStations();

        // then
        Mockito.verify(mockConnection).createStatement();
    }
}

接下来是什么？检查是否executeQuery()使用预期的参数调用了方法？测试如何SQLException处理？这些都是单元测试的合理场景，但是集成测试呢？为此，我会推荐DBUnit。它将您的测试数据库置于测试运行之间的已知状态，并允许根据预期的 XML 数据集验证返回的结果。

于 2015-08-15T00:07:59.363 回答