Add Rust support to the Python test harness

Author: tromey

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

@@ -719,6 +719,17 @@ def skipUnlessTargetAndroid(func):
     )(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

@@ -1368,6 +1368,18 @@ def getDwarfVersion(self):
                 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()
@@ -1609,6 +1621,11 @@ def buildProgram(self, sources, exe_name):
         d = {"CXX_SOURCES": sources, "EXE": exe_name}
         self.build(dictionary=d)
 
+    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' % (