Scala隐式类和函数

时间:2016-12-12 13:23:04

标签: scala implicit

我正在尝试重构一些使用隐式类的代码。 基本上,它是一些单元测试的代码,我在其中使用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方法在伴随对象中,并将结果和验证器作为参数。但它们并不是隐含的,我不得不放弃构建器模式,因此不会对它们进行链接。

1 个答案:

答案 0 :(得分:2)

我认为你需要的是一个函数(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(_))
  }
}