如何用Scalamock部分模拟类并测试Cls类的B方法？

你遇到的问题核心在于Scalamock的stub[Cls]会完全替换类中所有方法的实现，而不是仅替换你指定的A方法。当你创建stub[Cls]时，B方法的原有逻辑被覆盖成了返回类型的默认值（Int类型默认是0），所以调用f.B()不会执行原有的A(7)+1逻辑，自然断言失败。

Scalamock是否支持部分模拟？ #

是的，Scalamock支持部分模拟（Partial Mocking），常用的方式是使用spy来包装真实对象，这样未被stub的方法会保留原有实现，仅替换你指定的方法。

解决方案：用Spy实现部分模拟 #

修改你的测试代码如下：

test("test case with spy") {
  // 创建真实的Cls实例
  val realInstance = new Cls()
  // 用spy包装真实实例，保留原有方法实现
  val f = spy(realInstance)
  
  // 仅stub A方法的指定调用
  (f.A _).when(7).returns(5)
  
  // 此时调用B会执行原逻辑，使用被stub的A方法返回值
  assert(f.B() == 6)
}

这样运行测试就能得到预期结果，因为B方法的原有逻辑被保留，调用A(7)时会使用你stub的返回值5，最终计算出5+1=6。

为什么不推荐使用部分模拟？ #

虽然部分模拟能解决当前问题，但它通常是代码设计不够合理的信号，主要原因有：

• 耦合度高：类内部方法直接依赖同一类的其他方法，说明类的职责可能过于单一或混杂，违反了单一职责原则。
• 测试脆弱：测试依赖于类的内部实现细节（比如B调用了A），如果后续重构修改了内部调用逻辑（比如B不再调用A），测试会失败，但实际功能可能正常。
• 可维护性差：部分模拟的测试可读性更低，其他开发者可能难以理解测试意图。

更优的重构方案 #

更好的做法是将A方法的逻辑提取到一个独立的依赖组件中，通过依赖注入让Cls依赖这个组件，这样测试时只需mock这个依赖即可，无需部分模拟：

重构后的业务代码

// 提取A方法的逻辑到独立的特质/类
trait DataLoader {
  def load(dummy: Int): Int
}

class DefaultDataLoader extends DataLoader {
  override def load(dummy: Int): Int = 5
}

// Cls依赖DataLoader，而不是自己实现A方法
class Cls(dataLoader: DataLoader) {
  def B(): Int = dataLoader.load(7) + 1
}

对应的测试代码

test("test case with refactored code") {
  // mock DataLoader依赖
  val mockLoader = stub[DataLoader]
  (mockLoader.load _).when(7).returns(5)
  
  // 注入mock依赖到Cls
  val cls = new Cls(mockLoader)
  
  // 验证B方法的行为
  assert(cls.B() == 6)
}

这种方式更符合面向对象设计原则，测试也更清晰、稳定，不需要依赖类的内部实现细节。

内容的提问来源于stack exchange，提问作者MrMaxPayne