模拟 getJdbcTemplate 方法

Question

我的场景如下：

我在SampleDao界面中有no setter methods，在SampleDaoImpl类中有no direct field

参考：How to mock getJdbcTemplate().queryForObject()?

public interface SampleDao {
    // some methods
}

public class SampleDaoImpl extends JdbcDaoSupport implements SampleDao {
    // implementation of some methods
    public String someMethod(String param1, String param2){
        // .....//
        List<String> data = getJdbcTemplate().query(.....);  // -> getJdbcTemplate() is the method of JdbcDaoSupport to get JdbcTemplate
    }
}

我想模拟getJdbcTemplate().query(.....) 的结果，其中getJdbcTemplate() 属于JdbcDaoSupport 类，由SampleDaoImpl 扩展，SampleDao 没有。

我的测试用例如下：

创建SampleDaoImpl的对象并分配给SampleDao

@RunWith(Parameterized.class)
public class MockSampleDao {

    String param1 = "", param2 = "";
    @Mock
    SampleDao sampleDao = new SampleDaoImpl();


    public MockSampleDao(String param1, String param2) {
        super();
        this.param1 = param1;
        this.param2 = param2;   
    }

    @Parameterized.Parameters
    public static Collection primeNumbers() {
        return Arrays.asList(new Object[][] { 
            { "test1", "test1" },
            { "test2", "test2" }
        });
    }

    @Test
    public void testSomeMethod(){
        try {
        // HOW TO MOCK THE RESULT FROM getJdbcTemplate().query() HERE

            sampleDao.someMethod(param1, param2);
        } catch (Exception e) {
            e.printStackTrace();
        }
    }

}

Answer 1

你在嘲笑错误的对象。您正在尝试测试 SampleDaoImpl.someMethod()，因此您必须创建一个 具体 SampleDaoImpl 实例，而不是模拟实例。

一旦你有了一个具体的实例，你就可以调用setJdbcTemplate() 并传入一个mocked JdbcTemplate 对象。最后，您可以控制该模拟的行为，以便当someMethod() 调用query() 时，返回您的首选数据。

例如：

public class MockSampleDao {

    // parameterized stuff omitted for brevity

    @Test
    public void testSomeMethod() throws Exception {

      SampleDaoImpl impl = new SampleDaoImpl();
      JdbcTemplate template = mock(JdbcTemplate.class);
      List<String> someList = // populate this list
      when(template.query(....)).thenReturn(someList);

      impl.setJdbcTemplate(template);
      impl.someMethod(param1, param2);

      // further testing etc.
    }

}

Answer 2

应该是这个样子（根据你要覆盖的JdbcTemplate的查询API来调整mock）：

@RunWith(Parameterized.class)
public class MockSampleDao extends SampleDaoImpl{
      JdbcTemplate jdbcTemplate = mock(JdbcTemplate.class);

     @Override
     protected JdbcTemplate getJdbcTemplate(){
        return jdbcTemplate;
     }

    @Test
    public void testSomeMethod(){
        try {
             when(jdbcTemplate.query(/* put here the api that you want to overload*/)).
             thenReturn(/* the mock output that you want */);

            sampleDao.someMethod(param1, param2);
        } catch (Exception e) {
            e.printStackTrace();
        }
}