Run method properly fails when executed code throws an exception in the main thread

git-svn-id: https://simple-build-tool.googlecode.com/svn/trunk@919 d89573ee-9141-11dd-94d4-bdf5e562f29c

src/main/scala/sbt/Run.scala

@@ -77,24 +77,20 @@ object Run extends ScalaRun
 	/** Runs the class 'mainClass' using the given classpath and options using the scala runner.*/
 	def run(mainClass: String, classpath: Iterable[Path], options: Seq[String], log: Logger) =
 	{
-		createSettings(log)
-		{
-			(settings: Settings) =>
+		createSettings(log) { (settings: Settings) =>
+			Control.trapUnit("Error during run: ", log)
 			{
-				Control.trapUnit("Error during run: ", log)
-				{
-					val classpathURLs = classpath.map(_.asURL).toList
-					val bootClasspath = FileUtilities.pathSplit(settings.bootclasspath.value)
-					val extraURLs =
-						for(pathString <- bootClasspath if pathString.length > 0) yield
-							(new java.io.File(pathString)).toURI.toURL
-					log.info("Running " + mainClass + " ...")
-					log.debug("  Classpath:" + (classpathURLs ++ extraURLs).mkString("\n\t", "\n\t",""))
-					def execute =
-						try { ObjectRunner.run(classpathURLs ++ extraURLs, mainClass, options.toList) }
-						catch { case e: java.lang.reflect.InvocationTargetException => throw e.getCause }
-					executeTrapExit( execute, log )
-				}
+				val classpathURLs = classpath.map(_.asURL).toList
+				val bootClasspath = FileUtilities.pathSplit(settings.bootclasspath.value)
+				val extraURLs =
+					for(pathString <- bootClasspath if pathString.length > 0) yield
+						(new java.io.File(pathString)).toURI.toURL
+				log.info("Running " + mainClass + " ...")
+				log.debug("  Classpath:" + (classpathURLs ++ extraURLs).mkString("\n\t", "\n\t",""))
+				def execute =
+					try { ObjectRunner.run(classpathURLs ++ extraURLs, mainClass, options.toList) }
+					catch { case e: java.lang.reflect.InvocationTargetException => throw e.getCause }
+				executeTrapExit( execute, log )
 			}
 		}
 	}

src/main/scala/sbt/TrapExit.scala

@@ -29,8 +29,9 @@ object TrapExit
 		* the threads that were created by 'execute'.*/
 		val originalThreads = allThreads
 		val code = new ExitCode
+		def executeMain = try { execute } catch { case x => code.set(1); throw x }
 		val customThreadGroup = new ExitThreadGroup(new ExitHandler(Thread.getDefaultUncaughtExceptionHandler, originalThreads, code, log))
-		val executionThread = new Thread(customThreadGroup, "run-main") { override def run() { execute } }
+		val executionThread = new Thread(customThreadGroup, "run-main") { override def run() { executeMain } }
 		
 		val originalSecurityManager = System.getSecurityManager
 		try

src/sbt-test/run/error/changes/RunFailureMain.scala

@@ -0,0 +1,7 @@
+object Spawn
+{
+	def main(args: Array[String])
+	{
+		error("Test error main")
+	}
+}

src/sbt-test/run/error/changes/RunSuccess.scala

@@ -0,0 +1,4 @@
+object Spawn
+{
+	def main(args: Array[String]) {}
+}

src/sbt-test/run/error/changes/ThreadRunError.scala

@@ -0,0 +1,11 @@
+object Spawn
+{
+	def main(args: Array[String])
+	{
+		(new ThreadA).start
+	}
+	class ThreadA extends Thread
+	{
+		override def run(): Unit = error("Test error thread")
+	}
+}

src/sbt-test/run/error/project/build.properties

@@ -0,0 +1,2 @@
+project.name=Test Run Errors
+project.version=10.0

src/sbt-test/run/error/test

@@ -0,0 +1,15 @@
+$ copy-file changes/RunSuccess.scala src/main/scala/Run.scala
+[success]
+> run
+[success]
+
+$ copy-file changes/RunFailureMain.scala src/main/scala/Run.scala
+[success]
+> run
+[failure]
+
+# an exception in a non-main thread should still be exit code 0
+$ copy-file changes/ThreadRunError.scala src/main/scala/Run.scala
+[success]
+> run
+[success]