Spring JUnit 和 Mockito - SimpleJdbcTemplate

Question

给定一个扩展 SimpleJdbcDaoSupport 的类，如何模拟 SimpleJdbcTemplate？

public class SimpleJdbcDaoSupportExtension extends SimpleJdbcDaoSupport {  
     public SimpleJdbcDaoSupportExtension (JdbcTemplate jdbcTemplate){  
             super.setJdbcTemplate(jdbcTemplate);  
     }

     public MyDomainObj getResult(){
         SimpleJdbcTemplate sjdbc = getSimpleJdbcTemplate();  
         MyDomainObj result = sjdbc.query(*whatever necessary args*.);
         return result;
     }
}

然后，使用Mockito：

public class Test {  
    @Mock private JdbcTemplate mockedJdbcTemplateDedendency;  
    private SimpleJdbcDaoSupportExtension testObj;  

    @Before
    public void doBeforeEachTestCase() {
        MockitoAnnotations.initMocks(this);
        SimpleJdbcDaoSupportExtension sje = new SimpleJdbcDaoSupportExtension (mockedJdbcTemplateDedendency);
    }  
    @Test
    public final void test(){           
        when(mockedJdbcTemplateDedendency.query("what to query").thenReturn(new MyDomainObj());
    }
}

注入模拟的JdbcTemplate，但由于dao类依赖SimpleJdbcTemplate进行查询（用于映射到对象），并且它是由SimpleJdbcDaoSupport内部构造的 - 模拟JdcbTemplate对SimpleJdbcTemplate没有影响。那么，当没有公共设置器时，并且信任 SimpleJdbcTemplate 的唯一方法是依赖该方法 getSimpleJdbcObject() 时，该如何做到这一点？

Answer 1

您应该模拟一个接口（它具有您需要的方法），而不是模拟具体的类。

例如：

public class SimpleJdbcDaoSupportExtension extends SimpleJdbcDaoSupport implements MyDomainDao{  
     public SimpleJdbcDaoSupportExtension (JdbcTemplate jdbcTemplate){  
             super.setJdbcTemplate(jdbcTemplate);  
     }

     public MyDomainObj getResult(){
         SimpleJdbcTemplate sjdbc = getSimpleJdbcTemplate();  
         MyDomainObj result = sjdbc.query(*whatever necessary args*.);
         return result;
     }
}

public class Test {  
    @Mock private MyDomainDao myDomainDao ;
    private YourController yourController;  

    @Before
    public void doBeforeEachTestCase() {
        MockitoAnnotations.initMocks(this);
        yourController = new YourController(myDomainDao);
    }  
    @Test
    public final void testSomething(){           
        when(myDomainDao.getResult().thenReturn(new MyDomainObj());
        //on to testing the usages of myDomainDao
        yourController.doSomething();
        //verify
        verify(myDomainDao, times(2)).getResult();
    }
}

Answer 2

为什么要嘲笑 JdbcTemplate？使用 HSQL 等内存数据库来进行实际操作。

Answer 3

还有一件事。我通过将 SimpleJdbcTemplate 替换为 JdbcTemplate 解决了这个问题。我对 Java 有点陌生，从 .Net 世界跨界而来，最初使用 SimpleJdbcTemplate 只是因为我遇到了一些描述其用法的文档，这些文档完美地满足了我的需求。但 Skaffman 的评论让我更深入地研究了JdbcTemplate，然后我意识到我并不真正需要SimpleJdbcTemplate。
尽管如此，问题的哲学部分仍然存在。如何模拟只能通过询问容器本身来创建的东西？在我看来，Spring 对容器的紧密依赖违反了 DI 原则。

Answer 4

另一种允许您测试更多 dao 的方法是模拟连接、准备好的语句和结果集。这比仅仅模拟 jdbctemplate 需要做更多的工作，但它允许您验证准备好的语句是否设置了正确的值，以及您的自定义行映射器是否正常工作。

public class Test {  
  private MyDao dao;  
  private JdbcTemplate jdbcTemplate;
  private Connection conn;
  private PreparedStatement ps;
  private ResultSet rs;

@Before
public void setUp() throws Exception {
    dao = new MyDao();
    conn = mock(Connection.class);      
    ps = mock(PreparedStatement.class);
    rs = mock(ResultSet.class);
    jdbcTemplate = new JdbcTemplate(new SingleConnectionDataSource(conn, false));
    doa.setJdbcTemplate(jdbcTemplate);
}

@Test
public void test() throws Exception {           
    when(conn.prepareStatement(any(String.class))).thenReturn(ps);
    when(ps.executeQuery()).thenReturn(rs);
    // return one row
    when(rs.next()).thenReturn(true).thenReturn(false);

    when(rs.getInt("id")).thenReturn(1234);
    when(rs.getString("name")).thenReturn("Bob");

    MyDto myDto = dao.someDaoMethod(...)

    // verify ParameterSource
    verify(ps, times(1)).setInt(1, 1234);

    // these verify if you are mapping the columns to the right object attribute.
    assertEquals(1234, myDto.getId().intValue());
    assertEquals("Bob", myDto.getName());
}
}