aboutsummaryrefslogtreecommitdiff
path: root/core/src/test/scala/spark/DistributedSuite.scala
diff options
context:
space:
mode:
Diffstat (limited to 'core/src/test/scala/spark/DistributedSuite.scala')
-rw-r--r--core/src/test/scala/spark/DistributedSuite.scala6
1 files changed, 6 insertions, 0 deletions
diff --git a/core/src/test/scala/spark/DistributedSuite.scala b/core/src/test/scala/spark/DistributedSuite.scala
index 2ea38d4a5d..cacc2796b6 100644
--- a/core/src/test/scala/spark/DistributedSuite.scala
+++ b/core/src/test/scala/spark/DistributedSuite.scala
@@ -182,4 +182,10 @@ class DistributedSuite extends FunSuite with ShouldMatchers with BeforeAndAfter
assert(data.count() === 4000000)
System.clearProperty("spark.storage.memoryFraction")
}
+
+ test("passing environment variables to cluster") {
+ sc = new SparkContext(clusterUrl, "test", null, Nil, Map("TEST_VAR" -> "TEST_VALUE"))
+ val values = sc.parallelize(1 to 2, 2).map(x => System.getenv("TEST_VAR")).collect()
+ assert(values.toSeq === Seq("TEST_VALUE", "TEST_VALUE"))
+ }
}