Merge lp:~fginther/adt-result-checker/add-called-by-tarmac into lp:adt-result-checker

[Francis Ginther]

Add called-by-tarmac.py for automatic execution of unit tests by tarmac.

Add called-by-tarmac.py for automatic execution of unit tests by tarmac.

Copied from lp:adt-cloud-service.

[Paul Larson (pwlars)]

Not much for it to do right now since there are no tests, but will still be useful in the future. +1

review: Approve

[Francis Ginther (fginther)]

tarmac config has been updated, ready to land this.

=== added file 'called-by-tarmac.py'
--- called-by-tarmac.py	1970-01-01 00:00:00 +0000
+++ called-by-tarmac.py	2015-03-19 03:54:49 +0000
@@ -0,0 +1,102 @@
+#!/usr/bin/env python3
+#
+# adt-result-checker
+# Copyright (C) 2015 Canonical
+#
+# This program is free software: you can redistribute it and/or modify
+# it under the terms of the GNU General Public License as published by
+# the Free Software Foundation, either version 3 of the License, or
+# (at your option) any later version.
+#
+# This program is distributed in the hope that it will be useful,
+# but WITHOUT ANY WARRANTY; without even the implied warranty of
+# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
+# GNU General Public License for more details.
+#
+# You should have received a copy of the GNU General Public License
+# along with this program.  If not, see <http://www.gnu.org/licenses/>.
+#
+# called-by-tarmac.py is intended to be called by the tarmac merge engine
+#   to perform automatic unit testing on branch landing.
+#   It replaces the manual commands:
+#       bzr branch $PIP_CACHE_BRANCH pip_cache
+#       virtualenv -p python3 ve && \
+#           ./ve/bin/pip install --no-index --find-links=pip-cache \
+#              -r requirements.txt && \
+#           ./ve/bin/pip install -r test_requirements.txt && \
+#           ./ve/bin/python3 setup.py test && \
+#           rm -rf ve
+#
+
+import os
+import subprocess
+import sys
+import tempfile
+
+SERVICE_NAME = 'adt-result-checker'
+VENV_DIR = 'venv-{}'.format(SERVICE_NAME)
+PIP_DIR = 'pip-cache-'.format(SERVICE_NAME)
+PIP_CACHE_BRANCH = 'lp:~canonical-ci-engineering/{}/' \
+    'pip-cache'.format(SERVICE_NAME)
+
+
+def _run_command(cmd):
+    '''Run the command and return the return code or 0 on success.'''
+
+    print('INFO: Executing: {}'.format(' '.join(cmd)))
+    try:
+        subprocess.check_call(cmd, stderr=subprocess.STDOUT)
+    except subprocess.CalledProcessError as e:
+        print('ERROR: Calling `{}` failed (with code {})'.format(
+            e.cmd, e.returncode))
+        return e.returncode
+    return 0
+
+
+def main():
+    with tempfile.TemporaryDirectory(PIP_DIR) as pip_dir, \
+            tempfile.TemporaryDirectory(prefix=VENV_DIR) as ve_dir:
+        pip_cache = os.path.join(pip_dir, 'cache')
+        all_cmds = (
+            [
+                'bzr',
+                'branch',
+                PIP_CACHE_BRANCH,
+                pip_cache,
+            ],
+            [
+                'virtualenv',
+                '-p',
+                'python3',
+                ve_dir,
+            ],
+            [
+                os.path.join(ve_dir, 'bin', 'pip'),
+                'install',
+                '--no-index',
+                '--find-links={}'.format(pip_cache),
+                '-r',
+                'requirements.txt',
+            ],
+            [
+                os.path.join(ve_dir, 'bin', 'pip'),
+                'install',
+                '-r',
+                'test_requirements.txt',
+            ],
+            [
+                os.path.join(ve_dir, 'bin', 'python3'),
+                'setup.py',
+                'test',
+            ],
+        )
+        for cmd in all_cmds:
+            ret = _run_command(cmd)
+            if ret:
+                # A cmd failed, bubble up the return code to the caller
+                return ret
+    return 0
+
+
+if __name__ == '__main__':
+    sys.exit(main())