spark单元测试-sqljdbc连接

a1o7rhls  于 2021-05-27  发布在  Spark
关注(0)|答案(0)|浏览(423)

我想对下面的代码进行单元测试,这样我就可以获得良好的代码覆盖率。我在用mockito的funsuite。你能告诉我如何模拟数据库连接并进行单元测试吗。

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)

暂无答案!

目前还没有任何答案,快来回答吧!

相关问题