diff options
author | Davies Liu <davies@databricks.com> | 2016-03-03 17:46:28 -0800 |
---|---|---|
committer | Davies Liu <davies.liu@gmail.com> | 2016-03-03 17:46:28 -0800 |
commit | d062587dd2c4ed13998ee8bcc9d08f29734df228 (patch) | |
tree | ac1605fc1a82d0d41c51c20ab9a9ce1bfa241766 | |
parent | b373a888621ba6f0dd499f47093d4e2e42086dfc (diff) | |
download | spark-d062587dd2c4ed13998ee8bcc9d08f29734df228.tar.gz spark-d062587dd2c4ed13998ee8bcc9d08f29734df228.tar.bz2 spark-d062587dd2c4ed13998ee8bcc9d08f29734df228.zip |
[SPARK-13601] [TESTS] use 1 partition in tests to avoid race conditions
## What changes were proposed in this pull request?
Fix race conditions when cleanup files.
## How was this patch tested?
Existing tests.
Author: Davies Liu <davies@databricks.com>
Closes #11507 from davies/flaky.
-rw-r--r-- | sql/hive/src/test/scala/org/apache/spark/sql/sources/CommitFailureTestRelationSuite.scala | 7 |
1 files changed, 4 insertions, 3 deletions
diff --git a/sql/hive/src/test/scala/org/apache/spark/sql/sources/CommitFailureTestRelationSuite.scala b/sql/hive/src/test/scala/org/apache/spark/sql/sources/CommitFailureTestRelationSuite.scala index 64c27da475..2058705393 100644 --- a/sql/hive/src/test/scala/org/apache/spark/sql/sources/CommitFailureTestRelationSuite.scala +++ b/sql/hive/src/test/scala/org/apache/spark/sql/sources/CommitFailureTestRelationSuite.scala @@ -51,7 +51,7 @@ class CommitFailureTestRelationSuite extends SQLTestUtils with TestHiveSingleton withTempPath { file => // fail the job in the middle of writing val divideByZero = udf((x: Int) => { x / (x - 1)}) - val df = sqlContext.range(0, 10).select(divideByZero(col("id"))) + val df = sqlContext.range(0, 10).coalesce(1).select(divideByZero(col("id"))) SimpleTextRelation.callbackCalled = false intercept[SparkException] { @@ -69,7 +69,8 @@ class CommitFailureTestRelationSuite extends SQLTestUtils with TestHiveSingleton withTempPath { file => // fail the job in the middle of writing val divideByZero = udf((x: Int) => { x / (x - 1)}) - val df = sqlContext.range(0, 10).select(col("id").mod(2).as("key"), divideByZero(col("id"))) + val df = sqlContext.range(0, 10).coalesce(1) + .select(col("id").mod(2).as("key"), divideByZero(col("id"))) SimpleTextRelation.callbackCalled = false intercept[SparkException] { @@ -87,7 +88,7 @@ class CommitFailureTestRelationSuite extends SQLTestUtils with TestHiveSingleton SimpleTextRelation.failCommitter = false withTempPath { file => // fail the job in the middle of writing - val df = sqlContext.range(0, 10).select(col("id").mod(2).as("key"), col("id")) + val df = sqlContext.range(0, 10).coalesce(1).select(col("id").mod(2).as("key"), col("id")) SimpleTextRelation.callbackCalled = false SimpleTextRelation.failWriter = true |