Spark单元测试-模拟Azure SQLJDBC连接

时间:2020-08-22 07:50:25

标签: scala unit-testing apache-spark apache-spark-sql mockito

我想对下面的代码进行单元测试,以便获得良好的代码覆盖率。我正在将 FunSuite与Mockito一起使用。您能告诉我如何模拟数据库连接并进行单元测试吗?

def getSummaryConfig() : Config = {
    Config(Map(
      "url"            -> configUtil.getProperty("azure.host.name"),
      "databaseName"   -> configUtil.getProperty("azure.database.name"),
      "dbTable"        -> configUtil.getProperty("azure.summary.table"),
      "user"           -> configUtil.getProperty("azure.user.name"),
      "password"       -> configUtil.getProperty("azure.database.password")
    ))
  }

def getSummaryDF(summaryConfig : Config) : DataFrame = {
    val summaryDF = spark.read.option("driver", "com.microsoft.sqlserver.jdbc.SQLServerDriver").sqlDB(summaryConfig)
    summaryDF
  }

val summaryConfig = getSummaryConfig()
val summaryDF  = getSummaryDF(summaryConfig)

0 个答案:

没有答案