我有一个类,它接收数据库URL和用户名,连接到数据库,并执行查询,将结果分配给哈希映射。
import java.sql.*;
import java.util.Map;
public class MyClass {
private final String dbUrl;
private final String username;
MyClass(final String dbUrl, final String username) {
this.dbUrl = dbUrl;
this.username = username;
}
public void queryDatabase(final Map<Integer, String> userIdToName) {
final String query = "select user_id, name from users";
try (final Connection connection = DriverManager.getConnection(dbUrl, username, null);
final PreparedStatement ps = connection.prepareStatement(query);
final ResultSet rs = ps.executeQuery()) {
while (rs.next()) {
final int userId = rs.getInt(1);
final String name = rs.getString(2);
userIdToName.put(userId, name);
}
} catch (final SQLException e) {
System.out.println(e.toString());
}
}
}
我正在尝试使用Mockito模拟queryDatabase()
。我将PowerMockito添加到模拟DriverManager
。
我的测试班:
import org.junit.Before;
import org.junit.jupiter.api.Test;
import org.junit.runner.RunWith;
import org.mockito.InjectMocks;
import org.mockito.Mock;
import org.mockito.Mockito;
import org.powermock.core.classloader.annotations.PrepareForTest;
import org.powermock.modules.junit4.PowerMockRunner;
import java.sql.*;
import java.util.HashMap;
import java.util.Map;
import static org.mockito.ArgumentMatchers.*;
import static org.mockito.Mockito.mock;
import static org.powermock.api.mockito.PowerMockito.mockStatic;
import static org.powermock.api.mockito.PowerMockito.when;
@RunWith(PowerMockRunner.class)
@PrepareForTest(DriverManager.class)
class MyClassTest {
@Mock private Connection mockConnection = mock(Connection.class);
@Mock private PreparedStatement mockStatement = mock(PreparedStatement.class);
@Mock private ResultSet resultSet = mock(ResultSet.class);
@InjectMocks private static MyClass myClass = new MyClass("test_url", "test_username");
@Before
public void setUp() throws Exception {
mockStatic(DriverManager.class);
when(DriverManager.getConnection(anyString(), anyString(), null)).thenReturn(mockConnection);
}
@Test
void testQueryDatabase() throws Exception {
when(mockConnection.prepareStatement(anyString())).thenReturn(mockStatement);
when(mockStatement.executeQuery(anyString())).thenReturn(resultSet);
when(resultSet.getInt(Mockito.anyInt())).thenReturn(1);
when(resultSet.getString(anyInt())).thenReturn("john smith");
final Map<Integer, String> useridToUser = new HashMap<>();
myClass.queryDatabase(useridToUser);
System.out.println(useridToUser);
}
}
运行测试的输出是:
java.sql.SQLException: No suitable driver found for test_url
{}
我希望:
{1=john smith}
我该如何正确嘲笑?
除非绝对必要,我已经读过避免PowerMock
,所以我也很愿意提出重构建议,以尽可能避免。
您可以尝试执行以下操作:
class MyClassTest extends PowerMockTestCase {
添加依赖项:
<dependency>
<groupId>org.powermock.tests</groupId>
<artifactId>powermock-tests-utils</artifactId>
<version>1.6.6</version>
<scope>test</scope>
</dependency>
<dependency>
<groupId>org.powermock</groupId>
<artifactId>powermock-module-testng</artifactId>
<version>1.4.9</version>
<scope>test</scope>
</dependency>
可能还需要在安装中使用initMock:
@Before
public void setUp() throws Exception {
mockStatic(DriverManager.class);
initMocks(this);