使用scalamock部分模拟类

Question

我试图测试一个类cls，它具有两个函数:a和b。a加载DataFrame，B调用a，然后执行一些操作并返回新的DataFrame。为举例起见：

class Cls {
    def A(dummy: Int): Int = 5
    def B(): Int = A(7) + 1
}

使用scalamock如何编写测试代码？

我试过：

test("test case") {
  val f = stub[Cls]
  f.A _ when 7 returns 5
  assert(f.B() == 6)
}

我希望测试成功通过，并且得到0不等于6(MyTestCase.scala:24)（我知道scalamock用mock替换了所有现有函数，但这不是预期的行为）

编辑：我在mockito中找到了这个引用这个概念的答案，但我不确定scalamock是否支持这种嘲讽，以及为什么要反对这种嘲讽。

Answer 1

ScalaMock不重写/stub final方法。因此您的解决方案可以是创建一个子类,方法的部分标记为final：

import org.scalamock.scalatest.MockFactory
import org.scalatest.FunSuite

class PartialMockingTest extends FunSuite with MockFactory {

  test("test case") {

    class PartFinalCls extends Cls {
      override final def B(): Int = super.B()
    }

    val f = stub[PartFinalCls]
    f.A _ when 7 returns 5
    assert(f.B() == 6)
  }

}

class Cls {
  def A(dummy: Int): Int = 5
  def B(): Int = A(7) + 1
}