Groovy高效编程——使用Mock & Stub简化测试

对程序进行单元测试时，有时我们会遇到一些小麻烦，比如某些操作十分耗时，有些依赖的对象不易获取等等，这时我们就可以通过Groovy Mock或Groovy Stub来解决这些问题。Mock对方法调用的次数十分敏感，如果程序中调用的次数不在期望的调用次数范围内，则会抛出异常以致测试失败；而Stub仅仅是起到‘替身’的作用，它并不关心方法调用与否，即使方法一次都没被调用，也不会抛出异常导致测试失败。

下面让我们来看一个实际的例子吧：
import groovy.mock.interceptor.*

class User {
String name
String address
String toString() {
"Name: $name, Address: $address"
}
}

class UserDao {
User findUserByName(String name) {
if ('山风小子' == name)
return new User(name:name, address:'Shanghai')
else
return new User(name:'Anonymous User', address:'Unknown')
}
}

class ShowUserInfoService {
UserDao userDao = new UserDao()
def showInfo(String name) {
User user = userDao.findUserByName('山风小子')
println user
}
}


/*
平时下面这些代码写于单元测试中，为了简洁起见，单独抽取出来
*/


def userDaoMock = new MockFor(UserDao)
/*
UserDao对象中的findUserByName方法必须被调用1～2次，否则测试失败
同时将findUserByName方法的实现替换为return new User(name:'mock user', address:'Unknown')
*/
userDaoMock.demand.findUserByName(1..2) { String name ->
return new User(name:'mock user', address:'Unknown')
}

userDaoMock.use {
def showUserInfoService = new ShowUserInfoService()
showUserInfoService.showInfo('山风小子')
showUserInfoService.showInfo('Daniel.Sun')
//showUserInfoService.showInfo('BlueSUN')
}

def userDaoStub = new StubFor(UserDao)
/*
仅仅将findUserByName方法的实现替换为return new User(name:'stub user', address:'Unknown')
即使findUserByName方法不被调用，测试也不会失败
*/
userDaoStub.demand.findUserByName { String name ->
return new User(name:'stub user', address:'Unknown')
}

userDaoStub.use {
def showUserInfoService = new ShowUserInfoService()
showUserInfoService.showInfo('山风小子')
}
运行结果:
Name: mock user, Address: Unknown
Name: mock user, Address: Unknown
Name: stub user, Address: Unknown