Scala 隐式类和函数

Question

我正在尝试重构一些使用隐式类的代码。基本上，它是一些单元测试的代码，我在其中使用 Mockito 进行了一些模拟和监视。

在下面的示例中，这些函数的模拟是在隐式类 A 中完成的。每个方法“overrideX”都采用预期结果，覆盖对 myObj 中方法 X 的调用并返回 myObj(用于轻松链接多个覆盖的构建器模式稍后)。

object Outer {
  "call to abc" should {
    "fail if call to X fails" in {
      val failReason = "X failed"
      val myObjMock = B

      myObjMock.overrideX(failReason)
      myObjMock.abc mustEqual failReason
    }

    "fail if call to Y fails" in {
      val failReason = "Y failed"
      val myObjMock = B

      myObjMock.overrideY(failReason)
      myObjMock.abc mustEqual failReason
    }
  }

  implicit class A(myObj: B) {
    def overrideX(result: Result): B = {
      //override call to X in myObj with result
      myObj
    }

    def overrideY(result: Result): B = {
      //override call to Y in myObj with result
      myObj
    }
  }
}    

我要解决的问题是在测试中消除大量样板。正如您在上面的示例中看到的，这两个测试看起来非常相似。我想要实现的是为两者创建一些通用的方法，该方法将采用覆盖方法并进行实际检查。

def somethingElse(result: Result, f: Result => B) = {
    val myObjMock = B
    myObjMock.f(result).abc mustEqual result
}

测试看起来像

"call to abc" should {
    "fail if call to X fails" in {
      val failReason = "X failed"
      somethingElse(failReason, overrideX)
    }

    "fail if call to Y fails" in {
      val failReason = "Y failed"
      somethingElse(failReason, overrideY)
    }
  }

我知道 overrideX 和 overrideY 方法在类上而不是在伴随对象上。

我目前有一个版本，其中 overrideXX 方法位于伴随对象中，并将结果和验证器作为参数。但它们不是隐含的，我不得不放弃构建器模式，所以不要链接它们。

Answer 1

我认为您需要的是函数 (B, Result) => B 而不是 Result => B。如果你有一个包含函数的变量 f，你不能在像 myObjMock.f(...) 这样的对象上调用它。但是您可以将 myObjMock 传递给 f，例如 f(myObjMock, ...)。

def somethingElse(result: Result, f: (B, Result) => B) = {
  val myObjMock = B
  f(myObjMock, result).abc mustEqual result
}

"call to abc" should {
  "fail if call to X fails" in {
    val failReason = "X failed"
    somethingElse(failReason, _.overrideX(_))
  }

  "fail if call to Y fails" in {
    val failReason = "Y failed"
    somethingElse(failReason, _.overrideY(_))
  }
}