Added shutdown hook to STC to print summary even when interrupted by user.

Change-Id: I2ddb0d46ddfd776101a27ea76fe5ae34d28ddded

@@ -45,7 +45,12 @@ public final class Main {
     private static final String BLUE = "\u001B[36m";
     private static final String BLUE = "\u001B[36m";
 
 
     private static final String SUCCESS_SUMMARY = "%sPassed! %d steps succeeded%s";
     private static final String SUCCESS_SUMMARY = "%sPassed! %d steps succeeded%s";
-    private static final String FAILURE_SUMMARY = "%sFailed! %d steps succeeded; %d steps failed; %d steps skipped%s";
+    private static final String MIXED_SUMMARY =
+            "%s%d steps succeeded; %s%d steps failed; %s%d steps skipped%s";
+    private static final String FAILURE_SUMMARY = "%sFailed! " + MIXED_SUMMARY;
+    private static final String ABORTED_SUMMARY = "%sAborted! " + MIXED_SUMMARY;
+
+    private boolean isReported = false;
 
 
     private enum Command {
     private enum Command {
         LIST, RUN, RUN_RANGE, HELP
         LIST, RUN, RUN_RANGE, HELP
@@ -181,17 +186,20 @@ public final class Main {
     // Runs the coordinator and waits for it to finish.
     // Runs the coordinator and waits for it to finish.
     private void runCoordinator() {
     private void runCoordinator() {
         try {
         try {
+            Runtime.getRuntime().addShutdownHook(new ShutdownHook());
             coordinator.start();
             coordinator.start();
             int exitCode = coordinator.waitFor();
             int exitCode = coordinator.waitFor();
             pause(100); // allow stdout to flush
             pause(100); // allow stdout to flush
-            printSummary(exitCode);
+            printSummary(exitCode, false);
             System.exit(exitCode);
             System.exit(exitCode);
         } catch (InterruptedException e) {
         } catch (InterruptedException e) {
             print("Unable to execute scenario %s", scenarioFile);
             print("Unable to execute scenario %s", scenarioFile);
         }
         }
     }
     }
 
 
-    private void printSummary(int exitCode) {
+    private synchronized void printSummary(int exitCode, boolean isAborted) {
+        if (!isReported) {
+            isReported = true;
             Set<Step> steps = coordinator.getSteps();
             Set<Step> steps = coordinator.getSteps();
             int count = steps.size();
             int count = steps.size();
             if (exitCode == 0) {
             if (exitCode == 0) {
@@ -200,7 +208,10 @@ public final class Main {
                 long success = steps.stream().filter(s -> coordinator.getStatus(s) == SUCCEEDED).count();
                 long success = steps.stream().filter(s -> coordinator.getStatus(s) == SUCCEEDED).count();
                 long failed = steps.stream().filter(s -> coordinator.getStatus(s) == FAILED).count();
                 long failed = steps.stream().filter(s -> coordinator.getStatus(s) == FAILED).count();
                 long skipped = steps.stream().filter(s -> coordinator.getStatus(s) == SKIPPED).count();
                 long skipped = steps.stream().filter(s -> coordinator.getStatus(s) == SKIPPED).count();
-            print(FAILURE_SUMMARY, color(FAILED), success, failed, skipped, color(null));
+                print(isAborted ? ABORTED_SUMMARY : FAILURE_SUMMARY,
+                      color(FAILED), color(SUCCEEDED), success,
+                      color(FAILED), failed, color(SKIPPED), skipped, color(null));
+            }
         }
         }
     }
     }
 
 
@@ -270,6 +281,14 @@ public final class Main {
         }
         }
     }
     }
 
 
+    // Shutdown hook to report status even when aborted.
+    private class ShutdownHook extends Thread {
+        @Override
+        public void run() {
+            printSummary(1, true);
+        }
+    }
+
     // Logger to quiet Jetty down
     // Logger to quiet Jetty down
     private static class NullLogger implements Logger {
     private static class NullLogger implements Logger {
         @Override
         @Override