Unsetting the TaskContext which was set with empty TaskContext

Devaraj K · Devaraj K · commit 15b77dfdaa9f · 2017-06-24T01:42:43.000-07:00
diff --git a/core/src/test/scala/org/apache/spark/util/UtilsSuite.scala b/core/src/test/scala/org/apache/spark/util/UtilsSuite.scala
@@ -1108,5 +1108,6 @@ class UtilsSuite extends SparkFunSuite with ResetSystemProperties with Logging {
     assert(isErrorOccurred)
     // if the try, catch and finally blocks don't throw exceptions
     Utils.tryWithSafeFinallyAndFailureCallbacks {}(catchBlock = {}, finallyBlock = {})
+    TaskContext.unset
   }
 }

Original file line number	Diff line number	Diff line change
`@@ -1108,5 +1108,6 @@ class UtilsSuite extends SparkFunSuite with ResetSystemProperties with Logging {`
`1108`	`1108`	`assert(isErrorOccurred)`
`1109`	`1109`	`// if the try, catch and finally blocks don't throw exceptions`
`1110`	`1110`	`Utils.tryWithSafeFinallyAndFailureCallbacks {}(catchBlock = {}, finallyBlock = {})`
	`1111`	`+ TaskContext.unset`
`1111`	`1112`	`}`
`1112`	`1113`	`}`