Add Rust support to the Python test harness

Author: tromey

lldb/packages/Python/lldbsuite/test/decorators.py

@@ -626,6 +626,17 @@ def skipUnlessTargetAndroid(func):
                                 "requires target to be Android")(func)
 
 
+def skipUnlessRustInstalled(func):
+    """Decorate the item to skip tests when no Rust compiler is available."""
+
+    def is_rust_missing(self):
+        compiler = self.getRustCompilerVersion()
+        if not compiler:
+            return "skipping because rust compiler not found"
+        return None
+    return skipTestIfFn(is_rust_missing)(func)
+
+
 def skipIfHostIncompatibleWithRemote(func):
     """Decorate the item to skip tests if binaries built on this host are incompatible."""
 

lldb/packages/Python/lldbsuite/test/lldbtest.py

@@ -1331,6 +1331,18 @@ def getDwarfVersion(self):
             except: pass
         return '0'
 
+    def getRustCompilerVersion(self):
+        """ Returns a string that represents the rust compiler version, or None if rust is not found.
+        """
+        compiler = which("rustc")
+        if compiler:
+            version_output = system([[compiler, "--version"]])[0]
+            for line in version_output.split(os.linesep):
+                m = re.search('rustc ([0-9\.]+)', line)
+                if m:
+                    return m.group(1)
+        return None
+
     def platformIsDarwin(self):
         """Returns true if the OS triple for the selected platform is any valid apple OS"""
         return lldbplatformutil.platformIsDarwin()
@@ -1599,6 +1611,11 @@ def buildGModules(
                                     dictionary, testdir, testname):
             raise Exception("Don't know how to build binary with gmodules")
 
+    def buildRust(self):
+        """Build the default rust binary.
+        """
+        system([[which('rustc'), '-g main.rs']])
+
     def signBinary(self, binary_path):
         if sys.platform.startswith("darwin"):
             codesign_cmd = "codesign --force --sign \"%s\" %s" % (