ktest: Added reboot on success

Signed-off-by: Steven Rostedt <rostedt@goodmis.org>
diff --git a/tools/testing/ktest/ktest.pl b/tools/testing/ktest/ktest.pl
index 9eaf8d0..f344fd0 100644
--- a/tools/testing/ktest/ktest.pl
+++ b/tools/testing/ktest/ktest.pl
@@ -21,6 +21,7 @@
 $opt{"BUILD_NOCLEAN"}		= 0;
 $opt{"REBOOT_ON_ERROR"}		= 0;
 $opt{"POWEROFF_ON_ERROR"}	= 0;
+$opt{"REBOOT_ON_SUCCESS"}	= 1;
 $opt{"POWEROFF_ON_SUCCESS"}	= 0;
 $opt{"BUILD_OPTIONS"}		= "";
 $opt{"BISECT_SLEEP_TIME"}	= 10;   # sleep time between bisects
@@ -644,6 +645,10 @@
 for (my $i = 1; $i <= $opt{"NUM_BUILDS"}; $i++) {
     my $type = "BUILD_TYPE[$i]";
 
+    if (!defined($opt{$type})) {
+	$opt{$type} = $opt{"DEFAULT_BUILD_TYPE"};
+    }
+
     $noclean = set_build_option("BUILD_NOCLEAN", $i);
     $minconfig = set_build_option("MIN_CONFIG", $i);
     $run_test = set_build_option("TEST", $i);
@@ -674,7 +679,7 @@
 
 if ($opt{"POWEROFF_ON_SUCCESS"}) {
     halt;
-} else {
+} elsif ($opt{"REBOOT_ON_SUCCESS"}) {
     reboot;
 }