Scala模拟部分残茬

Question

class TeamService(val dep1: D1) {

  def method1(param: Int) = param * dep1.magicNumber()

  def method2(param: Int) = {
    method1(param) * 2
  }
}

val teamService = ??? // creates a stub
(teamService.method1 _).when(33).returns(22)
teamService.method2(33).should be(44)

有没有办法做到这一点？

Answer 1

正如其他问题中所建议的，这里和这里，我们可以将stub与final结合起来。来自ScalaMock常见问题：

我可以模拟final/private方法或类吗？

这是不受支持的，因为用宏生成的模拟是作为要模拟的类型的子类实现的。因此不能重写private和final方法

it should "test" in {
  val teamService = stub[TeamService]
  (teamService.method1 _).when(33).returns(22)
  teamService.method2(33) shouldBe 44
}

it should "test" in {
  class PartFinalTeamService(dep: D1) extends TeamService(dep) {
    override final def method2(param: Int): Int = super.method2(param)
  }

  val teamService = stub[PartFinalTeamService]
  (teamService.method1 _).when(33).returns(22)
  teamService.method2(33) shouldBe 44
}