如何mock Scala单例对象方法内部调用的方法且不修改现有代码

运行时抛出空指针的核心原因有三个：

• Reader是Scala单例对象，对应生成的类是final修饰的，且你在Reader中重写的read方法也加了final关键字，默认配置下的Mockito基于子类动态代理实现，无法对final类、final方法做拦截，打桩逻辑完全不生效。
• 你使用mock[Reader.type]生成的是一个空代理对象，所有未正确打桩的方法默认返回null，调用read方法时没有执行真实逻辑，直接返回null，后续调用show()就触发空指针。
• 就算解决了final方法的代理问题，直接mock整个Reader.type也会把read方法本身的实现替换掉，根本达不到「仅mockreadTable、执行read内真实业务逻辑」的测试目标。

第一步：引入支持final类型Mock的依赖 #

普通Mockito只能代理非final类/方法，需要引入mockito-inline依赖，它基于字节码重定义实现，支持对final类、final方法、单例对象做打桩，不需要修改生产代码。
如果是sbt项目，在build.sbt中添加测试依赖：

libraryDependencies += "org.mockito" % "mockito-inline" % "4.11.0" % Test

引入该依赖后会自动启用inline mock能力，不需要额外编写配置文件。

第二步：调整测试代码逻辑 #

核心改动是用spy（部分mock）代替全量mock：spy会保留对象的所有真实逻辑，仅对显式打桩的方法做返回值替换，刚好匹配你只mockreadTable的需求。
修正后的测试代码如下：

import org.apache.spark.sql.DataFrame
import org.apache.spark.sql.SparkSession
import org.mockito.MockitoSugar._
import org.scalatest.featurespec.AnyFeatureSpec

class ReaderTest extends AnyFeatureSpec {
  // 初始化测试用的Spark会话和样例返回数据
  implicit val spark: SparkSession = SparkSession.builder()
    .master("local[*]")
    .appName("reader-test")
    .getOrCreate()

  import spark.implicits._
  private val mockReturnDf: DataFrame = Seq(
    ("sss", "test_value")
  ).toDF("closingDate", "test_col")

  scenario("测试read方法业务逻辑") {
    // 对真实的Reader单例对象做spy，而非全量mock
    val readerSpy = spy(Reader)
    // 仅对readTable方法打桩，其余逻辑全部走真实实现
    when(readerSpy.readTable("abc", "sss")).thenReturn(mockReturnDf)
    // 调用真实的read方法，执行内部业务逻辑，仅readTable返回mock数据
    val result = readerSpy.read("abc", "sss")
    result.show()
  }
}

关键注意事项 #

• 打桩方法时不需要手动传入隐式参数列表，只要当前测试作用域存在对应类型的隐式值（比如这里的spark: SparkSession），Scala会自动填充，避免隐式实例不一致导致打桩匹配失败。
• 不要对read方法做任何打桩，否则会覆盖你要测试的真实业务逻辑。
• 测试完成后记得关闭Spark会话释放资源，避免测试进程残留。

内容的提问来源于stack exchange，提问作者Pooja