将 Scala 中的类与 Mockito 匹配

Question

我正在尝试模拟 Spark 上下文以在调用 newAPIHadoopFile 时返回模拟的 RDD。

我设置如下：

val mockedOuterRdd = mock[RDD[(NullWritable, MyProtobuf)]]

mockedSc.newAPIHadoopFile(anyString, anyObject(),classOf[org.apache.hadoop.io.NullWritable],
  classOf[MyProtobuf],anyObject()) returns mockedOuterRdd

这在编译器中很好，但是当我运行它时，我得到了

Invalid use of argument matchers!
5 matchers expected, 3 recorded:
-> at ...
This exception may occur if matchers are combined with raw values:
    //incorrect:
    someMethod(anyObject(), "raw String");
When using matchers, all arguments have to be provided by matchers.
For example:
    //correct:
    someMethod(anyObject(), eq("String by matcher"));

有没有办法可以将eq(...)（我尝试过但不起作用）之类的东西与classOf[...]一起使用？

我尝试将anyObject 用于类，但它从这些类中推断出 RDD 的类型参数，因此它们必须是正确的。

感谢阅读。

Answer 1

eq 对我有用，但是我猜你可能遗漏了测试框架的一些细节。我使用了scalatest，当我尝试使用eq 时，它一直对我大喊大叫，因为我返回了boolean 而不是Class。但是，有一个通用的eq 应该接管。因此，解决方案是使用完整路径：

mockedSc.newAPIHadoopFile(anyString, anyObject(),
  org.mockito.Matchers.eq(classOf[org.apache.hadoop.io.NullWritable]), 
  org.mockito.Matchers.eq(classOf[org.apache.hadoop.io.NullWritable]), anyObject())