1 files changed, 10 insertions, 4 deletions
diff --git a/test/py/u_boot_console_base.py b/test/py/u_boot_console_base.py
index 4606ad48bf5..ee9b928756f 100644
--- a/test/py/u_boot_console_base.py
+++ b/test/py/u_boot_console_base.py
@@ -223,13 +223,14 @@ class ConsoleBase(object):
         for each command in a list.
 
         Args:
-            cmd: List of commands (each a string)
+            cmd: List of commands (each a string).
         Returns:
-            Combined output of all commands, as a string
+            A list of output strings from each command, one element for each
+            command.
         """
-        output = ''
+        output = []
         for cmd in cmds:
-            output += self.run_command(cmd)
+            output.append(self.run_command(cmd))
         return output
 
     def ctrlc(self):
@@ -393,6 +394,11 @@ class ConsoleBase(object):
             pass
         self.p = None
 
+    def restart_uboot(self):
+        """Shut down and restart U-Boot."""
+        self.cleanup_spawn()
+        self.ensure_spawned()
+
     def get_spawn_output(self):
         """Return the start-up output from U-Boot