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

时间:2018-01-23 14:20:54

标签: java generics mockito

我想在类中模拟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始终为空,我可以在调试时确认。

1 个答案:

答案 0 :(得分:7)

在接受&#34;&#34;接受&#34;之后,Mockito第一个参数为Integer,总是希望第二个参数为Integer。解决方案(经过大量调试后)是在测试函数中强制转换 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