如何在 Mockito 中使用条件模拟

Question

我有一个用mockito模拟的scala方法调用。

让对象a并b方法，方法 b 采用类型 Map[String, String] 的参数。我想要一张满足特定条件的地图。我不知道此地图的所有键值对，但我想确保地图具有键值对，如 key1 -> value1 和 key2 -> value2当我使用这样的东西时

when(a.b(any[Map[String, String]])) thenReturn something

这里any是org.mockito中的静态成员。但是我不能在这里使用它，因为这个模拟对任何键和值Map都感到满意。在这种情况下如何实现条件模拟？

我正在寻找一个类似于where方法的实用程序 org.scalamock.matchers.Matchers

Answer 1

像这样尝试从IdiomaticMockito answers

import org.scalatest._
import org.mockito.{ArgumentMatchersSugar, IdiomaticMockito}
trait Foo {
  def bar(m: Map[String, String]): String
}
class FooSpec extends FlatSpec with Matchers with IdiomaticMockito with ArgumentMatchersSugar {
  "Foo object" should "say woohoo" in {
    val foo = mock[Foo]
    foo.bar(*) answers ((m: Map[String, String]) => if (m.exists(_ == ("key1", "value1"))) "woohoo" else "boom")
    foo.bar(Map("key1" -> "value1")) should be ("woohoo")
  }
}

在这里，我们将条件响应传递给answers

if (m.exists(_ == ("key1", "value1"))) "woohoo" else "boom")

请注意，我们使用 mockito-scala，它是香草 Mockito，除了了解 Scala 的细节：

libraryDependencies += "org.mockito" %% "mockito-scala" % "1.5.11"

这是与香草莫西托相同的解决方案

import org.scalatest._
import org.mockito.Mockito
import org.mockito.Mockito._
import org.mockito.ArgumentMatchers._
trait Foo {
  def bar(m: Map[String, String]): String
}
class FooSpec extends FlatSpec with Matchers {
  "Foo object" should "say woohoo old syntax" in {
    val foo = Mockito.mock(classOf[Foo])
    doAnswer({ args =>
      val m = args.getArgument[Map[String, String]](0)
      if (m.exists(_ == ("key1", "value1"))) "woohoo" else "boom"
    }).when(foo).bar(any[Map[String, String]])
    foo.bar(Map("key1" -> "value1")) should be ("woohoo")
  }
}