使用EasyMock对java类扩展存储过程进行单元测试
当我尝试对扩展StoredProcedure的以下类进行单元测试时,我在JDBCTemplate
类中的return (Map) execute(csc, new CallableStatementCallback()
行得到一个NullPointerException。我模拟了在execute
方法DataSource
和sql中传递的bean
public class MyStoredProc extends StoredProcedure {
/**
* Constructor - sets SQLParameters for the stored procedure.
*
* @param ds - DataSource
*/
public MyStoredProc(DataSource dataSource, String sql) {
super(dataSource, sql);
declareParameter(new SqlOutParameter("return",Types.NUMERIC));
declareParameter(new SqlParameter("BATCH_ID",Types.NUMERIC));
declareParameter(new SqlParameter("PROCESS_TYPE",Types.VARCHAR));
complie();
}
public BigDecimal execute(MyBean bean){
BigDecimal returnValue = BigDecimal.valueOf(-1);
Map in = new HashMap();
in.put("BATCH_ID", bean.getBatchID());
in.put("PROCESS_TYPE", bean.getProcessType());
Object obj = execute(in);
if (obj != null) {
Object output = ((HashMap) obj).get("return");
if( output instanceof BigDecimal) {
returnValue = (BigDecimal)output;
}
}
return bigDec;
}
}
测试用例:p.S-当我调试这个测试用例时,StoredProcedure模拟根本没有被使用。而是使用实际的实现
public class MyStoredProcTest {
private MyStoredProc mysp;
private DataSource dataSource;
private String sql;
@Before
public void setUp() {
dataSource = EasyMock.createMock(DataSource.class);
sql = "Testing";
mysp = new MyStoredProc(dataSource, sql);
}
@Test
public void testExecute() {
StoredProcedure storedProcedure = EasyMock
.createMock(StoredProcedure.class);
HashMap map = new HashMap();
map.put("return", BigDecimal.ONE);
expect(storedProcedure.execute(EasyMock.anyObject(Map.class))).andReturn(map);
Connection con = EasyMock.createMock(Connection.class);
expect(dataSource.getConnection()).andReturn(con);
MyBean bean = EasyMock.createMock(MyBean.class);
expect(bean.getBatchID()).andReturn(BigDecimal.valueOf(.0001))
.anyTimes();
expect(bean.getProcessType()).andReturn("Process Type").anyTimes();
replay(bean, dataSource, storedProcedure, con);
BigDecimal returnValue = null;
try {
returnValue = mysp.execute(bean);
} catch (Exception e) {
System.out.println("exception" + e.getStackTrace());// the Null pointer from JDBCTemplate is caught here.
}
Assert.assertEquals(BigDecimal.valueOf(-1), returnValue);
}
# 1 楼答案
你的一些模拟没有被使用,因为你没有重放它们。您应该将
replay(bean)
修改为replay(bean, datasource, storedProcedure)
另一方面,
map
不需要被嘲笑。当您希望调用storedProcedure.execute(...)
时,可以返回预填充的map