introduce firmware_Cr50CCDUartStress am: 3ee179e2b0
Change-Id: I1401d74a5c62713299b3aae8e78ec387fb31002d
diff --git a/server/site_tests/firmware_Cr50CCDUartStress/control b/server/site_tests/firmware_Cr50CCDUartStress/control
new file mode 100644
index 0000000..9334c9d
--- /dev/null
+++ b/server/site_tests/firmware_Cr50CCDUartStress/control
@@ -0,0 +1,32 @@
+# Copyright 2020 The Chromium OS Authors. All rights reserved.
+# Use of this source code is governed by a BSD-style license that can be
+# found in the LICENSE file.
+
+from autotest_lib.server import utils
+
+AUTHOR = "Cr50 FW team"
+NAME = "firmware_Cr50CCDUartStress"
+PURPOSE = "Uart Stress Test in ccd mode"
+ATTRIBUTES = "suite:faft_cr50_prepvt, suite:faft_cr50_pvt"
+TIME = "MEDIUM"
+TEST_TYPE = "server"
+DEPENDENCIES = "servo_state:WORKING"
+
+DOC = """
+This is a test for Uart-USB bridging qualification.
+This test runs 'uart_stress_tester.py' and checks if there are any characters
+lost.
+"""
+
+if 'args_dict' not in locals():
+ args_dict = {}
+
+args_dict.update(utils.args_to_dict(args))
+servo_args = hosts.CrosHost.get_servo_arguments(args_dict)
+
+def run(machine):
+ host = hosts.create_host(machine, servo_args=servo_args)
+ job.run_test('firmware_Cr50CCDUartStress', host=host, cmdline_args=args,
+ full_args=args_dict, duration=600)
+
+parallel_simple(run, machines)
diff --git a/server/site_tests/firmware_Cr50CCDUartStress/control.faft_cr50_tot b/server/site_tests/firmware_Cr50CCDUartStress/control.faft_cr50_tot
new file mode 100644
index 0000000..fdf26b1
--- /dev/null
+++ b/server/site_tests/firmware_Cr50CCDUartStress/control.faft_cr50_tot
@@ -0,0 +1,39 @@
+# Copyright 2020 The Chromium OS Authors. All rights reserved.
+# Use of this source code is governed by a BSD-style license that can be
+# found in the LICENSE file.
+
+from autotest_lib.server import utils
+
+AUTHOR = "Cr50 FW team"
+NAME = "firmware_Cr50CCDUartStress.faft_cr50_tot"
+PURPOSE = "Uart Stress Test in ccd mode"
+ATTRIBUTES = "suite:faft_cr50_tot"
+TIME = "MEDIUM"
+TEST_TYPE = "server"
+DEPENDENCIES = "servo_state:WORKING"
+
+DOC = """
+This is a test for Uart-USB bridging qualification.
+This test runs 'uart_stress_tester.py' and checks if there are any characters
+lost.
+
+
+This test is applicable when using ToT cr50 module.
+TODO(crbug.com/1014572) Drop this control file once suites can inject test args.
+"""
+
+if 'args_dict' not in locals():
+ args_dict = {}
+
+args_dict.update(utils.args_to_dict(args))
+servo_args = hosts.CrosHost.get_servo_arguments(args_dict)
+
+# crbug.com/1014572: Inject test args.
+args_dict['tot_test_run'] = 'true'
+
+def run(machine):
+ host = hosts.create_host(machine, servo_args=servo_args)
+ job.run_test('firmware_Cr50CCDUartStress', host=host, cmdline_args=args,
+ full_args=args_dict, duration=600)
+
+parallel_simple(run, machines)
diff --git a/server/site_tests/firmware_Cr50CCDUartStress/firmware_Cr50CCDUartStress.py b/server/site_tests/firmware_Cr50CCDUartStress/firmware_Cr50CCDUartStress.py
new file mode 100644
index 0000000..7d12881
--- /dev/null
+++ b/server/site_tests/firmware_Cr50CCDUartStress/firmware_Cr50CCDUartStress.py
@@ -0,0 +1,115 @@
+# Copyright 2020 The Chromium OS Authors. All rights reserved.
+# Use of this source code is governed by a BSD-style license that can be
+# found in the LICENSE file.
+
+
+"""The autotest performing uart_stress_tester on EC uart port. """
+import logging
+
+from autotest_lib.client.common_lib import error
+from autotest_lib.server.cros.faft.cr50_test import Cr50Test
+
+
+FLAG_FILENAME = '/tmp/chargen_testing'
+# A ChromeOS TPM command to burden CR50.
+TPM_CMD = ('trunks_client --stress_test')
+# A command line to burden Cr50 with TPM_CMD as long as FLAG_FILENAME exists.
+CR50_LOAD_GEN_CMD = 'while [ -f %s ]; do %s; done &' % (FLAG_FILENAME, TPM_CMD)
+
+# Character generator
+CHARGEN_CMD = 'chargen'
+
+class firmware_Cr50CCDUartStress(Cr50Test):
+ """A test that checks character loss with a UART and TPM stress workload."""
+
+ version = 1
+ flag_filename = '/tmp/chargen_testing'
+
+ def initialize(self, host, cmdline_args, full_args):
+ """Initialize the test
+
+ Raises:
+ TestNAError: if the test environment is not properly set.
+ e.g. the servo type doesn't support this test, or
+ EC Uart command, chargen is not available.
+ """
+ super(firmware_Cr50CCDUartStress,
+ self).initialize(host, cmdline_args, full_args)
+
+ # Check EC chargen is available.
+ if not self.ec.has_command(CHARGEN_CMD):
+ raise error.TestNAError('chargen command is not available in EC.')
+ logging.info('Checked EC has the uart command, %r.', CHARGEN_CMD)
+
+ # Check CCD is in servo_type.
+ servo_type = self.servo.get_servo_version()
+ if 'ccd_cr50' not in servo_type:
+ raise error.TestNAError('unsupported servo type: %s' % servo_type)
+ logging.info('Checked the servo type is %r.', servo_type)
+
+ # Fast open cr50 and enable testlab.
+ self.fast_open(enable_testlab=True)
+ logging.info('CCD opened.')
+
+ # Change active device as ccd_cr50.
+ if self.servo.has_control('active_v4_device'):
+ try:
+ self.active_dev = 'ccd_cr50'
+ self.servo.set('active_v4_device', self.active_dev)
+ except error.TestFail as e:
+ raise error.TestNAError('cannot change active_v4_device: %s' %
+ str(e))
+ logging.info('Set the active v4 device as %r.', self.active_dev)
+ else:
+ self.active_dev = ''
+
+ # Store the original status of EC ec3po_interp_connect.
+ self.ec_ec3po_connect = self.servo.get('ec_ec3po_interp_connect',
+ prefix=self.active_dev)
+ # turn off EC ec3po_interp_connect
+ self.servo.set('ec_ec3po_interp_connect', 'off', prefix=self.active_dev)
+ logging.info('Turned off ec3po.')
+
+ def cleanup(self):
+ """Clean up Uart stress test, then cleanup Cr50Test"""
+ try:
+ # Terminate cr50 stressing command run.
+ self.host.run('rm -f ' + FLAG_FILENAME)
+
+ # Restore EC ec3po interpreter connect config.
+ if hasattr(self, 'ec_ec3po_connect'):
+ self.servo.set('ec_ec3po_interp_connect', self.ec_ec3po_connect,
+ prefix=self.active_dev)
+ logging.info('Recovered ec3po.')
+ finally:
+ # Cleanup super class
+ super(firmware_Cr50CCDUartStress, self).cleanup()
+
+ def run_once(self, duration):
+ """The method called by the control file to start the test.
+
+ Args:
+ duration: time in seconds to run uart_stress_tester.
+
+ Raises:
+ TestFail: uart_stress_tester returned non-zero exit code for
+ character loss or other reasons.
+ """
+
+ # Run TPM command to stress cr50 in CPU.
+ logging.info('Start to stress cr50 with TPM commands.')
+ self.host.run('touch ' + FLAG_FILENAME)
+ self.host.run('nohup sh -c %r &> /dev/null' % CR50_LOAD_GEN_CMD)
+
+ # Run uart_stress_tester.
+ uart_pty = self.servo.get('raw_ec_uart_pty', prefix=self.active_dev)
+ testcmd = 'uart_stress_tester.py -t %d -d %s' % (duration, uart_pty)
+
+ logging.info('Run Uart stress tester for %d seconds.', duration)
+ logging.info(testcmd)
+ try:
+ self.servo.system(testcmd, timeout=duration*2)
+ except error.AutoservRunError:
+ raise error.TestFail('Uart stress tester failed.')
+
+ logging.info('Uart stress tester passed.')
