如何编写JUnit测试用例验证数据库连接功能

别担心，这个方法的JUnit测试其实可以拆成几个核心场景来覆盖，我给你梳理下具体思路和可落地的代码示例：

你的connectToDB()方法主要做了三件事：初始化QBasicConnection实例、调用open()方法、捕获特定异常并转为IllegalStateException。我们需要针对这几个行为分别测试：

1. 正常连接成功的场景 #

验证当QBasicConnection.open()能正常执行时，连接实例被正确初始化，且open()方法确实被调用。这里要用到Mock框架（比如Mockito）来模拟QBasicConnection，避免真的连接数据库导致测试依赖外部环境。

2. 捕获IOException并转异常的场景 #

模拟open()抛出IOException，验证方法会正确抛出IllegalStateException，且异常的根因是原IOException。

3. 捕获QException并转异常的场景 #

和上面类似，模拟open()抛出QException，验证异常转换逻辑正确。

假设你的类叫DBConnector，先看测试类的实现：

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

public class DBConnectorTest {
    private DBConnector dbConnector;
    private AutoCloseable mockStatic; // 用来管理静态mock的资源

    @BeforeEach
    void setUp() {
        // 初始化测试用的DBConnector实例，传入测试参数
        dbConnector = new DBConnector("test-host", 1234, "test-user", "test-pass");
        // 拦截QBasicConnection的构造方法，后续返回我们的mock实例
        mockStatic = Mockito.mockStatic(QBasicConnection.class);
    }

    @AfterEach
    void tearDown() throws Exception {
        // 关闭静态mock，避免污染其他测试用例
        mockStatic.close();
    }

    @Test
    void connectToDB_ShouldInitializeConnectionAndCallOpen() throws IOException, QException {
        // 创建QBasicConnection的mock实例
        QBasicConnection mockConn = mock(QBasicConnection.class);
        // 指定当调用构造方法时返回这个mock实例
        mockStatic.when(() -> new QBasicConnection("test-host", 1234, "test-user", "test-pass"))
                .thenReturn(mockConn);

        // 执行待测试方法
        dbConnector.connectToDB();

        // 验证open()方法被调用了一次
        verify(mockConn, times(1)).open();
    }

    @Test
    void connectToDB_WhenOpenThrowsIOException_ShouldThrowIllegalStateException() throws IOException, QException {
        // 创建mock连接，并设置open()抛出IOException
        QBasicConnection mockConn = mock(QBasicConnection.class);
        IOException testIoException = new IOException("Failed to connect");
        doThrow(testIoException).when(mockConn).open();

        // 指定构造方法返回mock实例
        mockStatic.when(() -> new QBasicConnection("test-host", 1234, "test-user", "test-pass"))
                .thenReturn(mockConn);

        // 断言调用方法会抛出预期的异常
        IllegalStateException thrownException = assertThrows(IllegalStateException.class, 
                                                             () -> dbConnector.connectToDB());
        // 验证异常的根因是原来的IOException
        assertSame(testIoException, thrownException.getCause());
    }

    @Test
    void connectToDB_WhenOpenThrowsQException_ShouldThrowIllegalStateException() throws IOException, QException {
        // 逻辑和上面的IOException测试一致，只是异常类型换为QException
        QBasicConnection mockConn = mock(QBasicConnection.class);
        QException testQException = new QException("Q-specific connection error");
        doThrow(testQException).when(mockConn).open();

        mockStatic.when(() -> new QBasicConnection("test-host", 1234, "test-user", "test-pass"))
                .thenReturn(mockConn);

        IllegalStateException thrownException = assertThrows(IllegalStateException.class, 
                                                             () -> dbConnector.connectToDB());
        assertSame(testQException, thrownException.getCause());
    }
}

• 确保你用的是Mockito 3.4及以上版本，这个版本开始支持静态方法和构造方法的Mock，不需要额外依赖PowerMock。
• 测试前一定要重置mock资源（比如@AfterEach里关闭静态mock），避免测试用例之间的状态污染。
• 如果你需要验证conn成员变量是否被正确赋值，可以通过反射来访问（因为它是private的），不过一般来说验证open()被调用已经足够覆盖核心逻辑。

内容的提问来源于stack exchange，提问作者breddie forden