Initial test for baseruntime module

tests/cleanup.py

@@ -0,0 +1,75 @@
+"""
+cleanup docker container/images and mock root for smoke testing
+"""
+
+import logging
+import subprocess
+import re
+
+
+log = logging.getLogger('avocado.test')
+
+def cleanup_docker_and_mock(mockcfg, img_name):
+
+    # Clean-up old test artifacts (docker containers, image, mock root)
+
+    docker_containerlist_cmdline = 'docker ps --filter=ancestor=%s -a -q' % img_name
+    try:
+        containerlist = subprocess.check_output(docker_containerlist_cmdline,
+            stderr = subprocess.STDOUT, shell = True)
+    except subprocess.CalledProcessError as e:
+        log.error("command '%s' returned exit status %d; output:\n%s" %
+            (e.cmd, e.returncode, e.output))
+        raise
+    else:
+        log.info("docker container list with '%s' succeeded with output:\n%s" %
+            (docker_containerlist_cmdline, containerlist))
+
+    if containerlist:
+        containers = re.sub('[\r\n]+', ' ', containerlist)
+        log.info("docker containers using image %s need to be removed: %s\n" %
+            (img_name, containers));
+        docker_teardown_cmdline = 'docker rm -f %s' % containers
+        try:
+            docker_teardown_output = subprocess.check_output(docker_teardown_cmdline,
+                stderr = subprocess.STDOUT, shell = True)
+        except subprocess.CalledProcessError as e:
+            log.error("command '%s' returned exit status %d; output:\n%s" %
+                (e.cmd, e.returncode, e.output))
+            raise
+        else:
+            log.info("docker container teardown with '%s' succeeded with output:\n%s" %
+                (docker_teardown_cmdline, docker_teardown_output))
+    else:
+        log.info("no docker containers are using image %s\n" % img_name)
+
+    docker_teardown_cmdline = 'docker rmi %s' % img_name
+    try:
+        docker_teardown_output = subprocess.check_output(docker_teardown_cmdline,
+            stderr = subprocess.STDOUT, shell = True)
+    except subprocess.CalledProcessError as e:
+        if "No such image" not in e.output:
+            log.error("command '%s' returned exit status %d; output:\n%s" %
+                (e.cmd, e.returncode, e.output))
+            raise
+        else:
+            log.info("No existing docker image named %s" % img_name)
+    else:
+        log.info("docker teardown with '%s' succeeded with output:\n%s" %
+            (docker_teardown_cmdline, docker_teardown_output))
+
+    mock_teardown_cmdline = ['mock', '-r', mockcfg, '--scrub=all']
+    try:
+        mock_teardown_output = subprocess.check_output(mock_teardown_cmdline,
+            stderr = subprocess.STDOUT)
+    except subprocess.CalledProcessError as e:
+        log.error("command '%s' returned exit status %d; output:\n%s" %
+            (e.cmd, e.returncode, e.output))
+        raise
+    log.info("mock teardown with '%s' succeeded with output:\n%s" %
+        (mock_teardown_cmdline, mock_teardown_output))
+
+
+
+
+