test

Author: UebelAndre

rust/private/rustdoc_test.bzl

@@ -36,7 +36,7 @@ def _rust_doc_test_impl(ctx):
 
     # Construct rustdoc test command, which will be written to a shell script
     # to be executed to run the test.
-    flags = _build_rustdoc_flags(dep_info, crate)
+    flags = _build_rustdoc_flags(dep_info, crate, toolchain)
     if toolchain.os != "windows":
         rust_doc_test = _build_rustdoc_test_bash_script(ctx, toolchain, flags, crate)
     else:
@@ -77,12 +77,13 @@ def _dirname(path_str):
     """
     return "/".join(path_str.split("/")[:-1])
 
-def _build_rustdoc_flags(dep_info, crate):
+def _build_rustdoc_flags(dep_info, crate, toolchain):
     """Constructs the rustdoc script used to test `crate`.
 
     Args:
         dep_info (DepInfo): The DepInfo provider
         crate (CrateInfo): The CrateInfo provider
+        toolchain (rust_toolchain): The rust toolchain
 
     Returns:
         list: A list of rustdoc flags (str)
@@ -98,6 +99,13 @@ def _build_rustdoc_flags(dep_info, crate):
     link_flags += ["--extern=" + c.name + "=" + c.dep.output.short_path for c in d.direct_crates.to_list()]
     link_search_flags += ["-Ldependency={}".format(_dirname(c.output.short_path)) for c in d.transitive_crates.to_list()]
 
+    # Gets the paths to the folders containing the standard library (or libcore)
+    rust_lib_files = depset(transitive = [toolchain.rust_lib.files, toolchain.rustc_lib.files])
+    rust_lib_paths = depset([file.dirname for file in rust_lib_files.to_list()]).to_list()
+
+    # Tell Rustc where to find the standard library
+    link_search_flags.extend(["-L {}".format(lib) for lib in rust_lib_paths])
+
     # TODO(hlopko): use the more robust logic from rustc.bzl also here, through a reasonable API.
     for lib_to_link in dep_info.transitive_noncrates.to_list():
         is_static = bool(lib_to_link.static_library or lib_to_link.pic_static_library)