Skip to content

Commit 08f5afc

Browse files
committed
[SPARK-6980] Added UT for constructing RpcTimeout with default value
1 parent d3754d1 commit 08f5afc

File tree

1 file changed

+7
-0
lines changed

1 file changed

+7
-0
lines changed

core/src/test/scala/org/apache/spark/rpc/RpcEnvSuite.scala

Lines changed: 7 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -564,6 +564,13 @@ abstract class RpcEnvSuite extends SparkFunSuite with BeforeAndAfterAll {
564564
val rt2 = RpcTimeout(conf, Seq("spark.ask.invalid.timeout", testProp, secondaryProp), "1s")
565565
assert( testDurationSeconds === rt2.duration.toSeconds )
566566

567+
// Construct RpcTimeout with default value,
568+
val defaultProp = "spark.ask.default.timeout"
569+
val defaultDurationSeconds = 1
570+
val rt3 = RpcTimeout(conf, Seq(defaultProp), defaultDurationSeconds.toString + "s")
571+
assert( defaultDurationSeconds === rt3.duration.toSeconds )
572+
assert( rt3.message.contains(defaultProp) )
573+
567574
// Try to construct RpcTimeout with an unconfigured property
568575
intercept[Throwable] {
569576
RpcTimeout(conf, "spark.ask.invalid.timeout")

0 commit comments

Comments
 (0)