如何在Scala 3中使用EasyMock模拟Scala伴生对象？

你遇到的错误是因为Scala伴生对象编译后会生成一个final类（比如这里的CoreDataSource$），而EasyMock这类基于继承实现Mock的框架无法创建final类的子类，所以直接mock[CoreDataSource.type]会失败。下面提供两种可行的解决方案：

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

最稳妥的方式是把伴生对象里的数据源逻辑抽离成可注入的接口/特质，这样测试时可以轻松Mock这个依赖，同时代码也更符合面向对象设计原则。

具体步骤： #

1. 定义一个数据源提供者特质，封装核心逻辑：

trait DataSourceProvider {
  def getDataSource: Option[DataSource]
  def getConnection: Option[Connection] = getDataSource.flatMap(ds => Option(ds.getConnection))
}

2. 实现默认的数据源提供者类，把原来伴生对象里的Hikari配置逻辑移进去：

class DefaultDataSourceProvider extends DataSourceProvider {
  override def getDataSource: Option[DataSource] = {
    val config = new HikariConfig()
    // 这里添加你的配置参数代码
    val pool = new HikariDataSource(config)
    Some(pool)
  }
}

3. 修改原伴生对象，让它依赖这个提供者（设计成可替换的，方便测试）：

object CoreDataSource {
  // 用var方便测试时替换，生产环境也可以用依赖注入框架管理
  private var provider: DataSourceProvider = new DefaultDataSourceProvider()
  
  def getDataSource: Option[DataSource] = provider.getDataSource
  def getConnection: Option[Connection] = provider.getConnection
  
  // 测试专用方法，用于替换提供者（限定包访问权限避免被误用）
  private[yourpackage] def setProvider(newProvider: DataSourceProvider): Unit = {
    provider = newProvider
  }
}

4. 编写测试代码，Mock DataSourceProvider即可：

class CoreDataSourceSpec extends AnyFunSpec with EasyMockSugar {
  describe("Core Data Source") {
    it("能正确获取连接") {
      val mockProvider = mock[DataSourceProvider]
      val mockConn = mock[Connection]
      
      // 替换伴生对象的数据源提供者
      CoreDataSource.setProvider(mockProvider)
      
      // 设定Mock预期行为
      expect(mockProvider.getConnection).andReturn(Some(mockConn))
      replay(mockProvider)
      
      // 执行测试逻辑
      val conn = CoreDataSource.getConnection
      assert(conn.contains(mockConn))
      
      // 验证Mock方法是否被正确调用
      verify(mockProvider)
    }
  }
}

方案2：使用PowerMock扩展EasyMock（无需重构） #

如果不想改动原有代码，可以用PowerMock来Mock final类和静态方法，它能绕过Scala伴生对象的final限制。

具体步骤： #

1. 添加PowerMock相关依赖（以sbt为例）：

libraryDependencies ++= Seq(
  "org.powermock" % "powermock-module-junit4" % "2.0.9" % Test,
  "org.powermock" % "powermock-api-easymock" % "2.0.9" % Test,
  "org.easymock" % "easymock" % "4.3" % Test
)

2. 编写测试代码，用PowerMock来Mock伴生对象的静态方法：

import org.powermock.api.easymock.PowerMock
import org.powermock.core.classloader.annotations.PrepareForTest
import org.scalatest.BeforeAndAfterEach
import org.scalatest.funspec.AnyFunSpec
import org.scalatestplus.junit.JUnitRunner
import org.junit.runner.RunWith

@RunWith(classOf[JUnitRunner])
// 指定要Mock的伴生对象对应的类
@PrepareForTest(Array(classOf[CoreDataSource]))
class CoreDataSourceSpec extends AnyFunSpec with EasyMockSugar with BeforeAndAfterEach {

  describe("Core Data Source") {
    it("能Mock伴生对象的getConnection方法") {
      val mockConn = mock[Connection]
      
      // 初始化静态Mock
      PowerMock.mockStatic(classOf[CoreDataSource])
      // 设置预期调用行为
      expect(CoreDataSource.getConnection).andReturn(Some(mockConn))
      
      // 开启Mock模式
      PowerMock.replay(classOf[CoreDataSource])
      
      // 执行测试逻辑
      val conn = CoreDataSource.getConnection
      assert(conn.contains(mockConn))
      
      // 验证静态方法是否被正确调用
      PowerMock.verify(classOf[CoreDataSource])
    }
  }
}

两种方案对比： #

• 依赖注入方案：代码结构更清晰，易于维护和扩展，测试逻辑也更直观，是长期维护的首选。
• PowerMock方案：无需改动原有代码，适合快速解决测试问题，但会引入额外依赖，测试逻辑相对复杂，且PowerMock可能和部分框架存在兼容性问题。

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