Diff: mbed-os/UNITTESTS/unit_test/utils.py

Revision:

0:8fdf9a60065b

diff -r 000000000000 -r 8fdf9a60065b mbed-os/UNITTESTS/unit_test/utils.py
--- /dev/null	Thu Jan 01 00:00:00 1970 +0000
+++ b/mbed-os/UNITTESTS/unit_test/utils.py	Wed Oct 10 00:33:53 2018 +0000
@@ -0,0 +1,66 @@
+"""
+Copyright (c) 2018, Arm Limited
+SPDX-License-Identifier: Apache-2.0
+
+Licensed under the Apache License, Version 2.0 (the "License");
+you may not use this file except in compliance with the License.
+You may obtain a copy of the License at
+
+    http://www.apache.org/licenses/LICENSE-2.0
+
+Unless required by applicable law or agreed to in writing, software
+distributed under the License is distributed on an "AS IS" BASIS,
+WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+See the License for the specific language governing permissions and
+limitations under the License.
+
+
+UNIT TEST UTILITIES
+"""
+
+import platform
+import subprocess
+import logging
+import sys
+
+def is_tool(name):
+    """
+    Test if tool is found in PATH
+
+    @param name: executable name
+    @return: true if tool found, false otherwise
+    """
+    cmd = "where" if platform.system() == "Windows" else "which"
+    try:
+        subprocess.check_output([cmd, name], stderr=subprocess.STDOUT)
+        return True
+    except subprocess.CalledProcessError:
+        return False
+
+def execute_program(args, error_msg="An error occurred!", success_msg=None):
+    """
+    Execute program in a subprocess with given arguments
+
+    @param args: program and its arguments in a list
+    @param success_msg: message to show in case of success
+    @param error_msg: message to show in case of failure
+    """
+
+    try:
+        process = subprocess.Popen(args,
+                                   stdout=subprocess.PIPE,
+                                   stderr=subprocess.STDOUT)
+
+        while process.poll() is None:
+            logging.info(process.stdout.readline().decode("utf8"))
+
+        retcode = process.wait()
+
+        if retcode:
+            raise subprocess.CalledProcessError(retcode, args)
+        elif success_msg:
+            logging.info(success_msg)
+
+    except subprocess.CalledProcessError as error:
+        logging.error(error_msg)
+        sys.exit(error.returncode)