Updated the tests according to the change

Author: Devaraj K

core/src/test/scala/org/apache/spark/deploy/SparkSubmitSuite.scala

@@ -74,20 +74,26 @@ trait TestPrematureExit {
     @volatile var exitedCleanly = false
     mainObject.exitFn = (_) => exitedCleanly = true
 
+    var message: String = null
     val thread = new Thread {
       override def run() = try {
         mainObject.main(input)
       } catch {
         // If exceptions occur after the "exit" has happened, fine to ignore them.
         // These represent code paths not reachable during normal execution.
-        case e: Exception => if (!exitedCleanly) throw e
+        case e: Exception =>
+          message = e.getMessage
+          if (!(exitedCleanly || message.contains(searchString))) {
+            throw e
+          }
       }
     }
     thread.start()
     thread.join()
     val joined = printStream.lineBuffer.mkString("\n")
-    if (!joined.contains(searchString)) {
-      fail(s"Search string '$searchString' not found in $joined")
+    if (!(joined.contains(searchString) ||
+      (message != null && message.contains(searchString)))) {
+      fail(s"Search string '$searchString' not found in $joined or in $message")
     }
   }
 }