Use the remoteBlockSize param in the tests instead of conditioning on…

… if we're testing shuffles or not
apache · Jun 1, 2020 · e81aa5a · e81aa5a
1 parent 4cfeb8e
commit e81aa5a
Showing 1 changed file with 2 additions and 4 deletions.
diff --git a/core/src/test/scala/org/apache/spark/storage/BlockManagerDecommissionSuite.scala b/core/src/test/scala/org/apache/spark/storage/BlockManagerDecommissionSuite.scala
@@ -66,10 +66,8 @@ class BlockManagerDecommissionSuite extends SparkFunSuite with LocalSparkContext
     // workload we need to worry about.
       .set(config.STORAGE_DECOMMISSION_REPLICATION_REATTEMPT_INTERVAL, 1L)
 
-    // Force fetching to local disk
-    if (shuffle) {
-      conf.set("spark.network.maxRemoteBlockSizeFetchToMem", "1")
-    }
+    // Allow force fetching to local disk
+    conf.set("spark.network.maxRemoteBlockSizeFetchToMem", remoteBlockSize)
 
     sc = new SparkContext(master, "test", conf)