为什么方法参数会导致 Mockito 出现 NotSerializableException？

Question

我有一些使用 spark 运行的 Scala 代码，但我对其进行了简化：

// Not Serializable
class Config

object FileReader extends FileReader
class FileReader extends Serializable {
  def read(config: Config): String = config.getClass.toString
}

object Task extends Task(FileReader)
class Task(fileReader: FileReader) extends Serializable {
  def execute(config: Config): Unit = {
    fileReader.read(config)
  }
}

不是Config 不是Serializable。

我想为它们写一些测试，Task 的实例需要是Serializable，因为它可能被序列化并发送给 spark worker。

我使用这个函数来检查一个对象是否可以被序列化：

def checkSerializable(obj: AnyRef, name: String) = {
  println("### checking " + name + ": " + obj.getClass)
  new ObjectOutputStream(new ByteArrayOutputStream()).writeObject(obj)
  println(name + " ok")
}

首先我们检查一个普通的Task实例是否可以序列化：

val task = new Task(new FileReader)
checkSerializable(task, "no-mockito")

输出：

### checking no-mockito: class Task
no-mockito ok

看起来不错。

但我想用 Mockito 模拟 FileReader，所以我的代码是：

val fileReader = Mockito.mock(classOf[FileReader])
val config = Mockito.mock(classOf[Config])
Mockito.when(fileReader.read(config)).thenReturn("mocked")
val task = new Task(fileReader)
checkSerializable(task, "with-mockito1")

报错Config不可序列化：

### checking with-mockito1: class Task
java.io.NotSerializableException: Config$$EnhancerByMockitoWithCGLIB$$c7dcb0a5
    at java.io.ObjectOutputStream.writeObject0(ObjectOutputStream.java:1165)
    at java.io.ObjectOutputStream.writeArray(ObjectOutputStream.java:1359)
    at java.io.ObjectOutputStream.writeObject0(ObjectOutputStream.java:1155)
    at java.io.ObjectOutputStream.defaultWriteFields(ObjectOutputStream.java:1535)
    at java.io.ObjectOutputStream.writeSerialData(ObjectOutputStream.java:1496)

这很奇怪，因为config 只是一些方法参数，而不是类字段！

我稍微修改了我的代码以使 config 可序列化：

val fileReader = Mockito.mock(classOf[FileReader])
val config = Mockito.mock(classOf[Config], Mockito.withSettings().serializable())
Mockito.when(fileReader.read(config)).thenReturn("mocked")
val task = new Task(fileReader)
checkSerializable(task, "with-mockito2")

它仍然失败，另一个NotSerializableException：

### checking with-mockito2: class Task
java.io.NotSerializableException: org.mockito.internal.creation.DelegatingMethod
    at java.io.ObjectOutputStream.writeObject0(ObjectOutputStream.java:1165)
    at java.io.ObjectOutputStream.defaultWriteFields(ObjectOutputStream.java:1535)
    at java.io.ObjectOutputStream.writeSerialData(ObjectOutputStream.java:1496)
    at java.io.ObjectOutputStream.writeOrdinaryObject(ObjectOutputStream.java:1413)

不确定为什么在使用 Mockito 时 config 实例会包含在 task 中？又该如何避免呢？

演示项目：https://github.com/freewind/mockito-serialization-issue，你可以直接克隆它并运行demo/MockitoDemo.scala

---

更新：

另一个值得注意的是：如果我删除了这一行

fileReader.read(config)

来自Task，这意味着Task 将是：

class Task(fileReader: FileReader) extends Serializable {
  def execute(config: Config): Unit = {
    // removed this line: fileReader.read(config)
  }
}

不会再抛出NotSerializableException（我没有更改测试代码）

Answer 1

您可能还需要使 FileReader 模拟可序列化尝试：

val fileReader = Mockito.mock(classOf[FileReader], Mockito.withSettings().serializable())