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

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

这种方式在做单元测试的时候一般比较困难(主要是mock), 这里演示了一下使用GMock来如何做测试, 比java下的EasyMock相对来说要方便很多.
       // 录制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)
	   ...
        }

你可能感兴趣的:(spring,AOP,单元测试)