Merge ~kevinbecker/+git/autotest-client-tests:kevinbecker/oslat into ~canonical-kernel-team/+git/autotest-client-tests:master

[Kevin Becker]

UBUNTU: SAUCE: Added oslat test

For now this does not have a pass/fail and is only added for data gathering purposes.

Signed-off-by: Kevin Becker <email address hidden>

[Francis Ginther (fginther)]

Looks good to start running.

review: Approve

[Po-Hsu Lin (cypressyew)]

review: Approve

[Po-Hsu Lin (cypressyew)]

Applied and pushed, thanks.

diff --git a/rt_tests_oslat/control b/rt_tests_oslat/control
new file mode 100644
index 0000000..1ec0969
--- /dev/null
+++ b/rt_tests_oslat/control
@@ -0,0 +1,19 @@
+AUTHOR = '''
+Authors: Peter Xu <[e-mail]>
+ '''
+NAME = "rt_tests_oslat"
+DOC = '''
+description rt test utils
+URL https://git.kernel.org/pub/scm/utils/rt-tests/rt-tests.git
+'''
+SUITE = "None"
+TIME = "MEDIUM"
+TEST_CLASS = 'kernel'
+TEST_CATEGORY = 'Functional'
+TEST_TYPE = 'client'
+
+result = job.run_test_detail('rt_tests_oslat', test_name='setup', tag='setup', timeout=60*5)
+if result == 'GOOD':
+    job.run_test('rt_tests_oslat', test_name='rt_tests_oslat', tag='rt_tests_oslat')
+else:
+    print("ERROR: test failed to build")
diff --git a/rt_tests_oslat/rt_tests_oslat.py b/rt_tests_oslat/rt_tests_oslat.py
new file mode 100644
index 0000000..6e64cac
--- /dev/null
+++ b/rt_tests_oslat/rt_tests_oslat.py
@@ -0,0 +1,74 @@
+import multiprocessing
+import os
+import platform
+import re
+import shutil
+from autotest.client import canonical, test, utils
+from autotest.client.shared import error
+
+class rt_tests_oslat(test.test):
+    version = 1
+
+    def initialize(self):
+        self.flavour = re.split('-\d*-', platform.uname()[2])[-1]
+        self.arch = platform.processor()
+
+    def install_required_pkgs(self):
+        try:
+            series = platform.dist()[2]
+        except AttributeError:
+            import distro
+            series = distro.codename()
+
+        pkgs = [
+            'build-essential',
+            'git',
+            'libnuma-dev',
+        ]
+        gcc = 'gcc' if self.arch in ['ppc64le', 'aarch64', 's390x', 'riscv64'] else 'gcc-multilib'
+        pkgs.append(gcc)
+
+        cmd = 'yes "" | DEBIAN_FRONTEND=noninteractive apt-get install --yes --force-yes ' + ' '.join(pkgs)
+        self.results = utils.system_output(cmd, retain_output=True)
+
+    # setup
+    #
+    #    Automatically run when there is no autotest/client/tmp/<test-suite> directory
+    #
+    def setup(self):
+        self.install_required_pkgs()
+        self.job.require_gcc()
+        os.chdir(self.srcdir)
+        shutil.rmtree('rt-tests', ignore_errors=True)
+        canonical.setup_proxy()
+        branch = 'main'
+        cmd = 'git clone -b {} https://git.kernel.org/pub/scm/utils/rt-tests/rt-tests.git'.format(branch)
+        utils.system_output(cmd, retain_output=True)
+
+        # Print test suite HEAD SHA1 commit id for future reference
+        os.chdir(os.path.join(self.srcdir, 'rt-tests'))
+        title_local = utils.system_output("git log --oneline -1 | sed 's/(.*)//'", retain_output=False, verbose=False)
+        title_upstream = utils.system_output("git log --oneline | grep -v SAUCE | head -1", retain_output=False, verbose=False)
+        print("Latest commit in '{}' branch: {}".format(branch, title_local))
+        print("Latest upstream commit: {}".format(title_upstream))
+
+        try:
+            nprocs = '-j' + str(multiprocessing.cpu_count())
+        except:
+            nprocs = ''
+        utils.make(nprocs)
+
+
+    # run_once
+    #
+    #    Driven by the control file for each individual test.
+    #
+    #    Runs oslat for 60 seconds. This is just for data gathering purposes for now.
+    #
+    def run_once(self, test_name, args='--cpu-list 2-9 --rtprio 1 --duration 60 -q', exit_on_error=True):
+        if test_name == 'setup':
+            return
+
+        utils.system_output(self.srcdir + '/rt-tests/oslat ' + args, retain_output=True)
+
+        return