ScalaMock - 模拟高阶函数

时间:2018-06-15 19:28:58

标签: scala unit-testing mocking higher-order-functions scalamock

我需要一些帮助,了解如何使用ScalaMock

在类中模拟高阶函数
import org.scalamock.scalatest.MockFactory
import org.scalatest.{FlatSpec, Matchers}

class TestSpec extends FlatSpec with MockFactory with Matchers {

  class Foo {
    def foo(f: () ⇒ String) = "this is foo"
  }

  val fooMock = mock[Foo]

  "Failing test" should "pass but doesn't" in {
    (fooMock
      .foo(_: () ⇒ String))
      .expects({ () ⇒
        "bar"
      })
      .returns("this is the test")

    // will fail here
    val result = fooMock.foo({ () ⇒
      "bar"
    })

    assert(true)
  }

  "Passing test" should "that passes but I can't do it this way" in {
    val f = { () ⇒
      "bar"
    }

    (fooMock.foo(_: () ⇒ String)).expects(f).returns("this is the test")

    val result = fooMock.foo(f)

    result shouldBe "this is the test"
  }
}

正如您在上面的代码中看到的那样,当您传入具有更高阶函数的值时,被模拟的函数可以正常工作,但如果您在每个点上键入它,则不会。在我的用例中,我无法按照第二次测试中的方式进行操作

以下是有关用例的更多信息,但并非完全有必要回答此问题

这是一个简化的例子,但我需要一种让前者工作的方法。理由是(我会尽力解释这个)我有一个正在测试的A类。内部A是一个传递模拟类B的函数,基本上foo函数(如下所示)在这个模拟B中,我不能像我在下面的第二个例子中那样传入f。如果这没有任何意义,我可以尝试完全复制它。

TL; DR我需要第一次测试才能工作lol

为什么会发生这种情况的任何想法?

如果您对我为什么需要这样做感到好奇,这里有一个更精确的示例,说明我如何使用它:

import org.scalamock.scalatest.MockFactory
import org.scalatest.{FlatSpec, Matchers}

class TestSpec extends FlatSpec with MockFactory with Matchers {

  class Foo {
    def foo(f: () ⇒ String) = s"you sent in: ${f()}"
  }

  object Bar {

    def bar(foo: Foo): String = {
      val f = { () ⇒ "bar" }

      foo.foo(f)
    }
  }

  val fooMock = mock[Foo]

  "Failing test" should "pass but doesn't" in {
    (fooMock.foo(_: () ⇒ String))
      .expects({ () ⇒
        "bar"
      })
      .returns("this is the test")

    // will fail here
    val result = Bar.bar(fooMock)

    assert(true)
  }
}

1 个答案:

答案 0 :(得分:0)

这不起作用的原因归结于Scala本身。 例如看到这个scala repl:

scala> val f = { () ⇒ "bar" }
f: () => String = $$Lambda$1031/294651011@14a049f9

scala> val g = { () ⇒ "bar" }
g: () => String = $$Lambda$1040/1907228381@9301672

scala> g == f
res0: Boolean = false

因此,ScalaMock无法在没有帮助的情况下比较函数是否相等。 我的建议是,如果您仅对函数的结果感兴趣,请使用谓词匹配,如此处的文档所述:http://scalamock.org/user-guide/matching/

在谓词中,您可以评估捕获的参数并检查返回值确实为bar。或者,如果您的函数更复杂,也许您甚至可以执行mock[() => String]并在匹配器中比较引用相等性?