Question

    object ReadUtils {
    def readData(sqlContext: SQLContext, fileType: FileType.Value): List[DataFrame] = {
    //some logic

}

我正在为执行功能编写测试

    import com.utils.ReadUtils.readData
    
    class Logs extends Interface with NativeImplicits{
    
    override def execute(sqlContext: SQLContext){
        val inputDFs: List[DataFrame] = readData(sqlContext, FileType.PARQUET)
        //some logic
    }

在为执行功能编写测试时，如何模拟readData函数以返回虚拟值？当前，它正在调用实际功能。

test("Log Test") {
val df1 = //some dummy df
val sparkSession = SparkSession
      .builder()
      .master("local[*]")
      .appName("test")
      .getOrCreate()
    sparkSession.sparkContext.setLogLevel("ERROR")
   
val log = new Logs()
    val mockedReadUtils = mock[ReadUtils.type]
    when(mockedReadUtils.readData(sparkSession.sqlContext,FileType.PARQUET)).thenReturn(df1)
    log.execute(sparkSession.sqlContext)

Answer 1

简单的答案是-您做不到。对象基本上是scala中的单例，您不能模拟单例-这就是为什么他们说您应该尽可能避免单例的原因之一。

您可以改为模拟 sqlContext 及其所有在readData函数中调用的函数。

作为另一种方法，您可以尝试添加带有某种蛋糕模式的依赖注入-https://medium.com/rahasak/scala-cake-pattern-e0cd894dae4e

trait DataReader {
  def readData(sqlContext: SQLContext, fileType: FileType.Value): List[DataFrame]
}

trait RealDataReader {
  def readData(sqlContext: SQLContext, fileType: FileType.Value): List[DataFrame] = {
    // some code
  }
}

trait MockedDataReader {
  def readData(sqlContext: SQLContext, fileType: FileType.Value): List[DataFrame] = {
     // some moking code
  }
}

class Logs extends Interface with NativeImplicits with DataReader {

override def execute(sqlContext: SQLContext){
  val inputDFs: List[DataFrame] = readData(sqlContext, FileType.PARQUET)
    //some logic
  }
}

class RealLogs extends Logs with RealDataReader // that would be the real class

class MockedLogs extends Logs with MockedDataReader // that would be the class for tests

如何在scala中模拟函数以返回虚拟值？

1 个答案: