我想模拟类中的ResultSet
,其参数来自泛型返回函数(getSQLValue
(:
public class MyClass{
TableClass clsTableClass;
public void getResults(Integer index, Integer value, String str) throws SQLException {
try (ResultSet rsResult = clsTableClass.getRecords(
getSQLValue(index, value, -1),
getSQLValue(index, str, "")))
{
// do stuff
}
}
private <T> T getSQLValue(Integer index, T t, T def) {
return (index> 0) ? def : t;
}
}
具有签名方法的表类:
public class TableClass {
public ResultSet getRecords(Integer value, String str) {
// do stuff
}
这是单元测试类:
public class MyClassTest {
@Mock(name = "clsTableClass")
TableClass clsTableClass;
@InjectMocks
MyClass clsMyClass;
@Before
public void setUp() throws Exception
{
MockitoAnnotations.initMocks(this);
}
@Test
public void testClass() throws Exception
{
ResultSet rsResult = Mockito.mock(ResultSet.class);
when( clsTableClass.getRecords(
any(Integer.class),
any(String.class)).thenReturn(rsResult);
}
}
问题是ResultSet
总是空的,调试时可以确认。
事实证明,Mockito
在"接受"第一个参数作为整数之后,无法将第二个参数作为字符串。解决方案(经过大量调试(是将测试函数中的any()
匹配器强制转换为相应的类类型:
@Test
public void testClass() throws Exception
{
ResultSet rsResult = Mockito.mock(ResultSet.class);
when(clsTableClass.getRecords(
(Integer) any(),
(String) any()).thenReturn(rsResult);
}
现在Mockito
将匹配参数,ResultSet
将不再返回null
。