Skip to content
Closed
Changes from 1 commit
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
Prev Previous commit
Next Next commit
[SPARK-34541][CORE] Fixed an issue where data could not be cleaned up…
… when unregisterShuffle.
  • Loading branch information
yikf committed Feb 27, 2021
commit 69467ee6c70d1074b3a79d093cde365e5ed648a3
Original file line number Diff line number Diff line change
Expand Up @@ -37,7 +37,7 @@ import org.apache.spark.serializer.{JavaSerializer, KryoSerializer, Serializer}
* Tests for the fallback logic in UnsafeShuffleManager. Actual tests of shuffling data are
* performed in other suites.
*/
class SortShuffleManagerSuite extends SparkFunSuite with Matchers {
class SortShuffleManagerSuite extends SparkFunSuite with Matchers with LocalSparkContext {

private def doReturn(value: Any) = org.mockito.Mockito.doReturn(value, Seq.empty: _*)

Expand Down Expand Up @@ -143,7 +143,7 @@ class SortShuffleManagerSuite extends SparkFunSuite with Matchers {
withTempDir { tmpDir =>
val conf = new SparkConf(loadDefaults = false)
conf.set("spark.local.dir", tmpDir.getAbsolutePath)
val sc: SparkContext = new SparkContext("local", "SPARK-34541", conf)
sc = new SparkContext("local", "SPARK-34541", conf)
val rdd = sc.parallelize(1 to 10, 1).map(x => (x, x))
// Create a shuffleRdd
val shuffledRdd = new ShuffledRDD[Int, Int, Int](rdd, new HashPartitioner(4))
Expand Down