test: Add ability to pass args to tests when using run_device_tests.py

BRANCH=none
BUG=none
TEST=With dragonclaw v0.2 connected to Segger J-Trace and servo micro:
     ./test/run_device_tests.py
       => PASS

Signed-off-by: Tom Hughes <tomhughes@chromium.org>
Change-Id: Id1f18726231593fc2ec0f9fdb12574a8be8bf5b8
Reviewed-on: https://chromium-review.googlesource.com/c/chromiumos/platform/ec/+/2231723
Commit-Queue: Craig Hesling <hesling@chromium.org>
Reviewed-by: Craig Hesling <hesling@chromium.org>

1 files changed, 7 insertions, 2 deletions

diff --git a/test/run_device_tests.py b/test/run_device_tests.py
index 51df57f3f9..be5e3b2395 100755
--- a/test/run_device_tests.py
+++ b/test/run_device_tests.py
@@ -58,13 +58,16 @@ class TestConfig:
     """Configuration for a given test."""
 
     def __init__(self, name, image_to_use=ImageType.RW, finish_regexes=None,
-                 toggle_power=False):
+                 toggle_power=False, test_args=None):
+        if test_args is None:
+            test_args = []
         if finish_regexes is None:
             finish_regexes = [ALL_TESTS_PASSED_REGEX, ALL_TESTS_FAILED_REGEX]
 
         self.name = name
         self.image_to_use = image_to_use
         self.finish_regexes = finish_regexes
+        self.test_args = test_args
         self.toggle_power = toggle_power
         self.logs = []
         self.passed = False
@@ -216,7 +219,9 @@ def run_test(test, console, executor, timeout_secs=10):
         if test.image_to_use == ImageType.RO:
             c.write('reboot ro\n'.encode())
             time.sleep(1)
-        c.write('runtest\n'.encode())
+
+        test_cmd = 'runtest ' + ' '.join(test.test_args) + '\n'
+        c.write(test_cmd.encode())
 
         while True:
             c.flush()