ScalaMock:如何模拟/存根方法每次调用返回不同的值?

时间:2015-12-08 17:39:07

标签: scala testing mocking scalamock

使用ScalaMock,我想模拟/存根一个类方法,这样每次调用都会返回一个不同的值(调用顺序很重要)。

我可以使用mockexpects实现这一目标,但这会迫使我验证这些调用。

我可以使用stub吗?

另外,我怎么说“第一次返回X,然后总是返回Y”(包括mockstub)?

2 个答案:

答案 0 :(得分:8)

是的,这可以做到,虽然语法有点不直观:

    trait Foo { def getInt: Int }
    val fooStub = stub[Foo]

    (fooStub.getInt _).when().returns(1).noMoreThanOnce()
    (fooStub.getInt _).when().returns(2).noMoreThanOnce()
    (fooStub.getInt _).when().returns(3)
    (fooStub.getInt _).when().returns(4)

    assert(fooStub.getInt == 1)
    assert(fooStub.getInt == 2)
    assert(fooStub.getInt == 3)
    // Note that it's fine that we don't call it a fourth time - calls are not verified.

使用.noMoreThanOnce()而不是.once()很重要,否则会导致调用被验证。还有一个.noMoreThanTwice()方法,但我认为没有.noMoreThanNTimes()或任何等价物。

以下是如何做“第一次返回X,然后总是返回Y”的模拟和存根:

    trait Bar { def getString: String }
    val barMock = mock[Bar]

    (barMock.getString _).expects().returning("X")
    (barMock.getString _).expects().returning("Y").anyNumberOfTimes()

    assert(barMock.getString == "X")
    assert(barMock.getString == "Y")
    assert(barMock.getString == "Y")

    val barStub = stub[Bar]

    (barStub.getString _).when().returns("x").noMoreThanOnce()
    (barStub.getString _).when().returns("y")

    assert(barStub.getString == "x")
    assert(barStub.getString == "y")
    assert(barStub.getString == "y")

答案 1 :(得分:2)

对我来说,编写一个不验证调用的模拟和返回值取决于输入的模拟的最佳方法是使用onCall方法 - 它会关闭你的函数。默认情况下,它只会投放第一个电话,因此请务必添加anyNumberOfTimes或某些repreted(...)

import org.scalamock.scalatest.MockFactory

trait Foo {
  def getSomeValue(param1: Any, param2: Any): String
}

class Test extends MockFactory {
  val fooMock = stub[Foo]

  val it = Iterator.single("X") ++ Iterator.continually("Y")

  (fooMock.getSomeValue _)
    .expects(*, *)
    .onCall((p1, p2) => it.next())
    .anyNumberOfTimes
}

现在,对fooMock.someValue(...)的第一次调用将返回X,并且每个Y都会返回。