rustdoc_test: fix and test OUT_DIR (#1779)

It needs the same treatment as SYSROOT.

Co-authored-by: Daniel Wagner-Hall <[email protected]>
Co-authored-by: UebelAndre <[email protected]>

Author: 3 people

rust/private/rustdoc.bzl

@@ -132,6 +132,8 @@ def rustdoc_compile_action(
     if is_test:
         if "SYSROOT" in env:
             env.update({"SYSROOT": "${{pwd}}/{}".format(toolchain.sysroot_short_path)})
+        if "OUT_DIR" in env:
+            env.update({"OUT_DIR": "${{pwd}}/{}".format(build_info.out_dir.short_path)})
 
         # `rustdoc` does not support the SYSROOT environment variable. To account
         # for this, the flag must be explicitly passed to the `rustdoc` binary.

test/out_dir_in_tests/BUILD.bazel

@@ -1,5 +1,5 @@
 load("//cargo:defs.bzl", "cargo_build_script")
-load("//rust:defs.bzl", "rust_library", "rust_test", "rust_test_suite")
+load("//rust:defs.bzl", "rust_doc_test", "rust_library", "rust_test", "rust_test_suite")
 
 cargo_build_script(
     name = "build_script",
@@ -22,6 +22,11 @@ rust_test(
     edition = "2018",
 )
 
+rust_doc_test(
+    name = "demo_lib_doc_test",
+    crate = ":demo_lib",
+)
+
 rust_test_suite(
     name = "suite",
     srcs = glob(["tests/**"]),

test/out_dir_in_tests/src/lib.rs

@@ -1,3 +1,8 @@
+//! ```
+//! let contents = include_str!(concat!(env!("OUT_DIR"), "/test_content.txt"));
+//! assert_eq!("Test content", contents);
+//! ```
+
 #[cfg(test)]
 mod tests {
     use std::env;