postgresql - 使用 Mockito 模拟具有功能参数的 Scala 方法

Question

使用 Finatra 并尝试模拟数据库调用以进行测试。我正在使用定义此方法的 finagle-postgres:

def prepareAndQuery[T](sql: String, params: Any*)(f: Row => T): Future[Seq[T]] = {...}

使用 Mockito 的其他模拟非常简单。但是，我似乎无法让这个工作。在与编译器斗争之后，我终于找到了这个，但它在运行时不起作用:

case class Foo( a: Int, b: Int, c: Int )
val client = smartMock[Client]
def f(row:Row): Foo = {
  Foo(1,2,3)
}
client.prepareAndQuery[Foo]("select 1") returns { x => f(x) }

执行测试抛出:

org.mockito.exceptions.verification.SmartNullPointerException: 
...
because this method call was *not* stubbed correctly:

我是 Mockito 的新手，我还没有找到模拟采用函数参数的函数的示例。关于我的模拟是错误的有什么想法吗？

Answer 1

如果你想要该方法的直接 stub ，那么你可以做类似的事情

val client = mock[Client]
when(client.prepareAndQuery[Foo](any, any)(any)) thenReturn Future.successful(<whatever you want>)

这将返回 <whatever you want>不管怎样