From: Simon Glass <sjg@chromium.org>
Date: Sun, 3 Jul 2016 15:40:42 +0000 (-0600)
Subject: test/py: Add a helper to run a list of U-Boot commands
X-Git-Tag: v2016.09-rc1~103
X-Git-Url: https://git.sur5r.net/?a=commitdiff_plain;h=73a9054d0f33dc4612a13200c6f3af00d2a1fcda;p=u-boot

test/py: Add a helper to run a list of U-Boot commands

Some tests want to execute a sequence of commands. Add a helper for this.

Signed-off-by: Simon Glass <sjg@chromium.org>
Reviewed-by: Teddy Reed <teddy.reed@gmail.com>
---

diff --git a/test/py/u_boot_console_base.py b/test/py/u_boot_console_base.py
index 815fa64d5f..b5aad7cb94 100644
--- a/test/py/u_boot_console_base.py
+++ b/test/py/u_boot_console_base.py
@@ -216,6 +216,22 @@ class ConsoleBase(object):
             self.cleanup_spawn()
             raise
 
+    def run_command_list(self, cmds):
+        """Run a list of commands.
+
+        This is a helper function to call run_command() with default arguments
+        for each command in a list.
+
+        Args:
+            cmd: List of commands (each a string)
+        Returns:
+            Combined output of all commands, as a string
+        """
+        output = ''
+        for cmd in cmds:
+            output += self.run_command(cmd)
+        return output
+
     def ctrlc(self):
         """Send a CTRL-C character to U-Boot.