使用gmock对事务进行单元测试

这里的单元测试主要是需要mock掉数据库的方法, 如果是结合数据库回滚处理的集成测试不在讨论范围内.
在spring中我们使用事务通常有两种方式: 一种是采用aop的声明式事务, 一种是采用事务模板(TransactionTemplate), 对两种使用方式各有各的利弊, 这里不展开讨论, 我们通常采用第二种方式, 一般的写法也是比较template式的:
Java代码
getTemplate(...).execute(new TransactionCallback() {
public Object doInTransaction(final TransactionStatus status) {
try {
... // 使用事务的操作
} catch (Exception e) {
status.setRollbackOnly();
return e;
}
return null;
}
}

getTemplate(...).execute(new TransactionCallback() {
public Object doInTransaction(final TransactionStatus status) {
try {
... // 使用事务的操作
} catch (Exception e) {
status.setRollbackOnly();
return e;
}
return null;
}
}
这种方式在做单元测试的时候一般比较困难(主要是mock), 这里演示了一下使用GMock来如何做测试, 比java下的EasyMock相对来说要方便很多.
Java代码
// 录制mock对象的期望行为(主要是TransactionCallback内部类回调执行的动作)
...
// 对事物模板行为的录制
TransactionStatus status = mock(TransactionStatus){setRollbackOnly()}
// 使用闭包实现一个TransactionTemplate, 内部主要实现了一个回调, 这个用EasyMock实现比较困难
def template = [execute: {it.doInTransaction(status) }] as TransactionTemplate
// service是测试对象, 内部注入了TemplateManager依赖
def tm = service.TemplateManager = mock(TranscationManager)
tm.getTemplate(id).returns(template)
// 运行测试方法并验证
play{
service.update(xxx)
...
}