Scala中如何Mock单例Object的方法？测试调用类遇阻

Scala的单例object本质是静态类，直接mock这类方法会比较棘手，下面提供两种可行方案：

方案一：重构代码，使用依赖注入（推荐） #

通过抽象出特质，让单例对象实现该特质，同时让依赖类依赖特质而非直接依赖单例，这样测试时可以轻松传入mock实现。

重构步骤： #

1. 定义抽象特质

trait MyService {
  def myMethod(): String
}

2. 让原单例对象实现该特质

object MyObject extends MyService {
  override def myMethod(): String = "Original method"
}

3. 修改MyClass，通过构造函数注入依赖（默认使用原单例，不影响原有业务逻辑）

class MyClass(myService: MyService = MyObject) {
  def callMyObject(): String = myService.myMethod()
}

测试代码： #

使用Mockito创建MyService的mock实例，注入到MyClass中即可验证：

import org.scalatest.funsuite.AnyFunSuite
import org.mockito.Mockito._

class MyClassTest extends AnyFunSuite {
  test("callMyObject 返回mock值") {
    val mockService = mock(classOf[MyService])
    when(mockService.myMethod()).thenReturn("Mocked value")
    
    val myClass = new MyClass(mockService)
    assert(myClass.callMyObject() == "Mocked value")
  }
}

这种方式不仅解决了测试问题，还提升了代码的可扩展性和可维护性，是面向对象设计的最佳实践。

方案二：使用PowerMock直接Mock单例对象（不推荐） #

如果无法重构原有代码，可以借助PowerMock直接mock Scala的单例object，但这种方式依赖静态mock，可能会带来测试污染等副作用。

依赖配置（以sbt为例） #

libraryDependencies ++= Seq(
  "org.scalatest" %% "scalatest" % "3.2.15" % Test,
  "org.powermock" % "powermock-module-junit4" % "2.0.9" % Test,
  "org.powermock" % "powermock-api-mockito2" % "2.0.9" % Test,
  "org.mockito" %% "mockito-scala" % "1.17.12" % Test
)

测试代码： #

import org.scalatest.funsuite.AnyFunSuite
import org.powermock.api.mockito.PowerMockito
import org.powermock.core.classloader.annotations.PrepareForTest
import org.junit.runner.RunWith
import org.powermock.modules.junit4.PowerMockRunner

@RunWith(classOf[PowerMockRunner])
@PrepareForTest(Array(classOf[MyObject]))
class MyClassTest extends AnyFunSuite {
  
  test("callMyObject 返回mock值") {
    PowerMockito.mockStatic(classOf[MyObject])
    PowerMockito.when(MyObject.myMethod()).thenReturn("Mocked value")
    
    val myClass = new MyClass()
    assert(myClass.callMyObject() == "Mocked value")
  }
}

注意：使用PowerMock时需要确保测试之间的隔离，避免静态mock的状态污染后续测试。

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