ServiceClass 类的 if...else 逻辑分支保持不变（为了清晰起见）。同时，无参数构造函数仍然适用。注，并不总是需要有创造性逻辑，例如 while...do 子句或 for 循环来正确地测试类的方法。只要有针对类使用的对象的方法执行，简单的模拟期望就足以测试那些执行。
您还必须更改 ServiceClassTest 类以匹配场景，如下所示：
清单 5. 经过编辑的场景 2 的 ServiceClassTest 类
...
private ServiceClass serviceClass;
private Mock mockCollaborator;
private Collaborator collaborator;
public void setUp(){
serviceClass = new ServiceClass();