java类中的模拟scala对象方法调用

Question

我有一个java代码，我在其中调用scala对象的方法（依赖库的）。为了编写单元测试，我想模拟scala对象的方法调用

Operations.Scala

object Operations {

     def load(spark: SparkSession, path: String): Dataset[Row] = {

     }
}

public class Main {
    public void callMethod() {
         Dataset<Row>  df = Operations.load(sparkSession, path);
    }
}

Answer 1

下面是通常如何完成的...我将用scala写它，因为java语法要冗长得多（坦白地说，我不太记得它），但这是相同的想法。

   trait Operations { 
       def load(s: SparkSession, path: String): DataSet[Row]
   }
   object Operations extends Operations {
       def load(s: SparkSession, path: String) = ??? // implementation here
   }

   class Main(operations: Operations = Operations) {
       callMethod(): Unit = {
         val df = operations.load(sparkSession, path);
         ...
         // do stuff 
      }
   }

然后在测试中编写（这是使用mockito/scalatest，但对于用于测试的任何库都是同样的想法）：

    val ops = mock[Operations]
    when(ops.load(any, any)).thenReturn(whatever)
    new Main(ops).callMethod(mock[SparkSession], "/foo/bar")

这种方法称为“依赖项注入”：您的main类对operations有一个“依赖项”，它现在是在运行时“注入”的，而不是像您的代码片段那样硬编码。这允许您使用不同的依赖实例进行生产（实际的操作对象）和测试（模拟）运行。

最好是让这个方法返回实际的结果，让调用者来处理副作用。那么您就可以轻松地测试它是否产生了您所期望的结果：

   new Main(ops).callMethod(mock[SparkSession], "/foo/bar") shouldBe "ExpectedFooResult"