Initial commit of open source Workload Automation.

wlauto/commands/templates/UiAutomation.java

@@ -0,0 +1,25 @@
+package ${package_name};
+
+import android.app.Activity;
+import android.os.Bundle;
+import android.util.Log;
+import android.view.KeyEvent;
+
+// Import the uiautomator libraries
+import com.android.uiautomator.core.UiObject;
+import com.android.uiautomator.core.UiObjectNotFoundException;
+import com.android.uiautomator.core.UiScrollable;
+import com.android.uiautomator.core.UiSelector;
+import com.android.uiautomator.testrunner.UiAutomatorTestCase;
+
+import com.arm.wlauto.uiauto.BaseUiAutomation;
+
+public class UiAutomation extends BaseUiAutomation {   
+
+    public static String TAG = "${name}";
+
+    public void runUiAutomation() throws Exception {
+	// UI Automation code goes here
+    }
+
+}

wlauto/commands/templates/android_benchmark

@@ -0,0 +1,27 @@
+from wlauto import AndroidBenchmark, Parameter
+
+
+class ${class_name}(AndroidBenchmark):
+
+    name = '${name}'
+    # NOTE: Please do not leave these comments in the code.
+    #
+    # Replace with the package for the app in the APK file.
+    package = 'com.foo.bar'
+    # Replace with the full path to the activity to run.
+    activity = '.RunBuzz'
+    description = "This is an placeholder description"
+
+    parameters = [
+        # Workload parameters go here e.g.
+        Parameter('Example parameter', kind=int, allowed_values=[1,2,3], default=1, override=True, mandatory=False,
+                  description='This is an example parameter')
+    ]
+
+    def run(self, context):
+        pass
+
+    def update_result(self, context):
+        super(${class_name}, self).update_result(context)
+        # process results and add them using
+        # context.result.add_metric

wlauto/commands/templates/android_uiauto_benchmark

@@ -0,0 +1,24 @@
+from wlauto import AndroidUiAutoBenchmark, Parameter
+
+
+class ${class_name}(AndroidUiAutoBenchmark):
+
+    name = '${name}'
+    # NOTE: Please do not leave these comments in the code.
+    #
+    # Replace with the package for the app in the APK file.
+    package = 'com.foo.bar'
+    # Replace with the full path to the activity to run.
+    activity = '.RunBuzz'
+    description = "This is an placeholder description"
+
+    parameters = [
+        # Workload parameters go here e.g.
+        Parameter('Example parameter', kind=int, allowed_values=[1,2,3], default=1, override=True, mandatory=False,
+                  description='This is an example parameter')
+    ]
+
+    def update_result(self, context):
+        super(${class_name}, self).update_result(context)
+        # process results and add them using
+        # context.result.add_metric

wlauto/commands/templates/basic_workload

@@ -0,0 +1,28 @@
+from wlauto import Workload, Parameter
+
+
+class ${class_name}(Workload):
+
+    name = '${name}'
+    description = "This is an placeholder description"
+
+    parameters = [
+        # Workload parameters go here e.g.
+        Parameter('Example parameter', kind=int, allowed_values=[1,2,3], default=1, override=True, mandatory=False,
+                  description='This is an example parameter')
+    ]
+
+    def setup(self, context):
+        pass
+
+    def run(self, context):
+        pass
+
+    def update_result(self, context):
+        pass
+
+    def teardown(self, context):
+        pass
+
+    def validate(self):
+        pass

wlauto/commands/templates/setup.template

@@ -0,0 +1,102 @@
+import os
+import sys
+import warnings
+from multiprocessing import Process
+
+try:
+    from setuptools.command.install import install as orig_install
+    from setuptools import setup
+except ImportError:
+    from distutils.command.install import install as orig_install
+    from distutils.core import setup
+
+try:
+    import pwd
+except ImportError:
+    pwd = None
+
+warnings.filterwarnings('ignore', "Unknown distribution option: 'install_requires'")
+
+try:
+    os.remove('MANIFEST')
+except OSError:
+    pass
+
+
+packages = []
+data_files = {}
+source_dir = os.path.dirname(__file__)
+for root, dirs, files in os.walk('$package_name'):
+    rel_dir = os.path.relpath(root, source_dir)
+    data = []
+    if '__init__.py' in files:
+        for f in files:
+            if os.path.splitext(f)[1] not in ['.py', '.pyc', '.pyo']:
+                data.append(f)
+        package_name = rel_dir.replace(os.sep, '.')
+        package_dir = root
+        packages.append(package_name)
+        data_files[package_name] = data
+    else:
+        # use previous package name
+        filepaths = [os.path.join(root, f) for f in files]
+        data_files[package_name].extend([os.path.relpath(f, package_dir) for f in filepaths])
+
+params = dict(
+    name='$package_name',
+    version='0.0.1',
+    packages=packages,
+    package_data=data_files,
+    url='N/A',
+    maintainer='$user',
+    maintainer_email='$user@example.com',
+    install_requires=[
+        'wlauto',
+    ],
+    # https://pypi.python.org/pypi?%3Aaction=list_classifiers
+    classifiers=[
+        'Development Status :: 3 - Alpha',
+        'Environment :: Console',
+        'License :: Other/Proprietary License',
+        'Operating System :: Unix',
+        'Programming Language :: Python :: 2.7',
+    ],
+)
+
+
+def update_wa_packages():
+    sudo_user = os.getenv('SUDO_USER')
+    if sudo_user:
+        user_entry = pwd.getpwnam(sudo_user)
+        os.setgid(user_entry.pw_gid)
+        os.setuid(user_entry.pw_uid)
+    env_root = os.getenv('WA_USER_DIRECTORY', os.path.join(os.path.expanduser('~'), '.workload_automation'))
+    if not os.path.isdir(env_root):
+        os.makedirs(env_root)
+    wa_packages_file = os.path.join(env_root, 'packages')
+    if os.path.isfile(wa_packages_file):
+        with open(wa_packages_file, 'r') as wfh:
+            package_list = wfh.read().split()
+            if params['name'] not in package_list:
+                package_list.append(params['name'])
+    else:  # no existing package file
+        package_list = [params['name']]
+    with open(wa_packages_file, 'w') as wfh:
+        wfh.write('\n'.join(package_list))
+
+
+class install(orig_install):
+
+    def run(self):
+        orig_install.run(self)
+        # Must be done in a separate process because will drop privileges if
+        # sudo, and won't be able to reacquire them.
+        p = Process(target=update_wa_packages)
+        p.start()
+        p.join()
+
+
+params['cmdclass'] = {'install': install}
+
+
+setup(**params)

wlauto/commands/templates/uiauto_workload

@@ -0,0 +1,35 @@
+from wlauto import UiAutomatorWorkload, Parameter
+
+
+class  ${class_name}(UiAutomatorWorkload):
+
+    name = '${name}'
+    description = "This is an placeholder description"
+
+    parameters = [
+        # Workload parameters go here e.g.
+        Parameter('Example parameter', kind=int, allowed_values=[1,2,3], default=1, override=True, mandatory=False,
+                  description='This is an example parameter')
+    ]
+
+    def setup(self, context):
+        super(${class_name}, self).setup(context)
+        # Perform any necessary setup before starting the UI automation
+        # e.g. copy files to the device, start apps, reset logs, etc.
+
+
+    def update_result(self, context):
+        pass
+        # Process workload execution artifacts to extract metrics
+        # and add them to the run result using
+        # context.result.add_metric()
+
+    def teardown(self, context):
+        super(${class_name}, self).teardown(context)
+        # Preform any necessary cleanup
+
+    def validate(self):
+        pass
+        # Validate inter-parameter assumptions etc
+
+