Mockito thenReturn 在来自泛型函数时返回 null



我想模拟类中的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

相关内容

  • 没有找到相关文章

最新更新