Skip to content

Commit 20e3585

Browse files
committed
fix tests
1 parent acbaf8b commit 20e3585

File tree

1 file changed

+2
-1
lines changed

1 file changed

+2
-1
lines changed

core/src/test/scala/org/apache/spark/executor/ExecutorSuite.scala

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -42,7 +42,7 @@ import org.apache.spark.metrics.MetricsSystem
4242
import org.apache.spark.rdd.RDD
4343
import org.apache.spark.rpc.RpcEnv
4444
import org.apache.spark.scheduler.{FakeTask, ResultTask, TaskDescription}
45-
import org.apache.spark.serializer.JavaSerializer
45+
import org.apache.spark.serializer.{JavaSerializer, SerializerManager}
4646
import org.apache.spark.shuffle.FetchFailedException
4747
import org.apache.spark.storage.BlockManagerId
4848
import org.apache.spark.util.UninterruptibleThread
@@ -234,6 +234,7 @@ class ExecutorSuite extends SparkFunSuite with LocalSparkContext with MockitoSug
234234
val mockMemoryManager = mock[MemoryManager]
235235
when(mockEnv.conf).thenReturn(conf)
236236
when(mockEnv.serializer).thenReturn(serializer)
237+
when(mockEnv.serializerManager).thenReturn(mock[SerializerManager])
237238
when(mockEnv.rpcEnv).thenReturn(mockRpcEnv)
238239
when(mockEnv.metricsSystem).thenReturn(mockMetricsSystem)
239240
when(mockEnv.memoryManager).thenReturn(mockMemoryManager)

0 commit comments

Comments
 (0)