cts: Fail script when build or flash fails

This change makes cts.py fail when building or flashing a module for
DUT or TH fails.

BUG=none
BRANCH=none
TEST=Made cts.py fail by injecting build and flash error

Change-Id: Iec1e11f4a8c261eb4c989b118df218e86cb6f5f1
Signed-off-by: Daisuke Nojiri <dnojiri@chromium.org>
Reviewed-on: https://chromium-review.googlesource.com/393326
Reviewed-by: Randall Spangler <rspangler@chromium.org>

1 files changed, 8 insertions, 4 deletions

diff --git a/cts/cts.py b/cts/cts.py
index 983b9751d2..2999e4d06f 100755
--- a/cts/cts.py
+++ b/cts/cts.py
@@ -94,14 +94,18 @@ class Cts(object):
 
   def build(self):
     """Build images for DUT and TH"""
-    self.dut.build(self.module, self.ec_dir, self.debug)
-    self.th.build(self.module, self.ec_dir, self.debug)
+    if self.dut.build(self.module, self.ec_dir, self.debug):
+      raise RuntimeError('Building module %s for DUT failed' % (self.module))
+    if self.th.build(self.module, self.ec_dir, self.debug):
+      raise RuntimeError('Building module %s for TH failed' % (self.module))
 
   def flash_boards(self):
     """Flashes th and dut boards with their most recently build ec.bin"""
     self.identify_boards()
-    self.th.flash()
-    self.dut.flash()
+    if self.th.flash():
+      raise RuntimeError('Flashing TH failed')
+    if self.dut.flash():
+      raise RuntimeError('Flashing DUT failed')
 
   def setup(self):
     """Setup boards"""