From 08f5afc8ce9df2ad94225b0cabf1ed1a8222ac26 Mon Sep 17 00:00:00 2001 From: Bryan Cutler Date: Thu, 11 Jun 2015 16:58:48 -0700 Subject: [PATCH] [SPARK-6980] Added UT for constructing RpcTimeout with default value --- core/src/test/scala/org/apache/spark/rpc/RpcEnvSuite.scala | 7 +++++++ 1 file changed, 7 insertions(+) diff --git a/core/src/test/scala/org/apache/spark/rpc/RpcEnvSuite.scala b/core/src/test/scala/org/apache/spark/rpc/RpcEnvSuite.scala index 634df4ea7bfc4..a806259e6499c 100644 --- a/core/src/test/scala/org/apache/spark/rpc/RpcEnvSuite.scala +++ b/core/src/test/scala/org/apache/spark/rpc/RpcEnvSuite.scala @@ -564,6 +564,13 @@ abstract class RpcEnvSuite extends SparkFunSuite with BeforeAndAfterAll { val rt2 = RpcTimeout(conf, Seq("spark.ask.invalid.timeout", testProp, secondaryProp), "1s") assert( testDurationSeconds === rt2.duration.toSeconds ) + // Construct RpcTimeout with default value, + val defaultProp = "spark.ask.default.timeout" + val defaultDurationSeconds = 1 + val rt3 = RpcTimeout(conf, Seq(defaultProp), defaultDurationSeconds.toString + "s") + assert( defaultDurationSeconds === rt3.duration.toSeconds ) + assert( rt3.message.contains(defaultProp) ) + // Try to construct RpcTimeout with an unconfigured property intercept[Throwable] { RpcTimeout(conf, "spark.ask.invalid.timeout")