如何使用Mockito在java中为PreparedStatement编写测试用例

Mockito 是一个流行的 Java 测试框架，用于创建和管理测试中的模拟对象。在编写涉及数据库操作的单元测试时，使用 Mockito 可以有效地隔离代码与数据库的依赖，从而提高测试的速度和可靠性。以下是如何使用 Mockito 为 PreparedStatement 编写测试用例的步骤和相关概念。

基础概念

Mockito: 一个 Java 测试框架，用于创建和管理模拟对象。
PreparedStatement: Java 中用于执行预编译 SQL 语句的接口。
单元测试: 针对代码中最小可测试单元（通常是方法）的测试。

优势

隔离性: 通过模拟 PreparedStatement，可以避免实际访问数据库，从而提高测试速度。
可重复性: 模拟对象的行为可以精确控制，确保每次测试结果一致。
专注性: 可以专注于测试业务逻辑，而不是数据库连接和数据状态。

类型

模拟对象: Mockito 创建的对象，其行为可以被预设。
存根: 提供固定返回值的模拟对象。
间谍: 部分模拟对象，允许调用真实方法。

应用场景

数据库操作: 测试涉及 SQL 查询和更新的方法。
外部服务调用: 模拟 HTTP 请求、文件读写等外部依赖。

示例代码

假设我们有一个简单的 DAO 类，用于执行数据库查询：

public class UserDAO {
    private Connection connection;

    public UserDAO(Connection connection) {
        this.connection = connection;
    }

    public User getUserById(int id) throws SQLException {
        String sql = "SELECT * FROM users WHERE id = ?";
        try (PreparedStatement ps = connection.prepareStatement(sql)) {
            ps.setInt(1, id);
            ResultSet rs = ps.executeQuery();
            if (rs.next()) {
                return new User(rs.getInt("id"), rs.getString("name"));
            }
        }
        return null;
    }
}

我们可以使用 Mockito 编写一个测试用例：

import static org.mockito.Mockito.*;
import static org.junit.jupiter.api.Assertions.*;

import java.sql.*;
import org.junit.jupiter.api.BeforeEach;
import org.junit.jupiter.api.Test;
import org.mockito.Mock;
import org.mockito.MockitoAnnotations;

public class UserDAOTest {

    @Mock
    private Connection connection;

    @Mock
    private PreparedStatement preparedStatement;

    @Mock
    private ResultSet resultSet;

    private UserDAO userDAO;

    @BeforeEach
    public void setUp() throws SQLException {
        MockitoAnnotations.openMocks(this);
        when(connection.prepareStatement(anyString())).thenReturn(preparedStatement);
        when(preparedStatement.executeQuery()).thenReturn(resultSet);
        userDAO = new UserDAO(connection);
    }

    @Test
    public void testGetUserById() throws SQLException {
        int userId = 1;
        when(resultSet.next()).thenReturn(true);
        when(resultSet.getInt("id")).thenReturn(userId);
        when(resultSet.getString("name")).thenReturn("John Doe");

        User user = userDAO.getUserById(userId);

        assertNotNull(user);
        assertEquals(userId, user.getId());
        assertEquals("John Doe", user.getName());

        verify(preparedStatement).setInt(1, userId);
        verify(preparedStatement).executeQuery();
    }
}