add unit test for rustdoc

Author: TroyKomodo

test/unit/rustdoc/rustdoc_features.rs

@@ -0,0 +1,23 @@
+// Copyright 2022 The Bazel Authors. All rights reserved.
+//
+// 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.
+
+//!
+//! Checkout [inc]
+//! 
+
+/// Increments the input.
+#[cfg(feature = "docs")]
+pub fn inc(n: u32) -> u32 {
+    n + 1
+}

test/unit/rustdoc/rustdoc_unit_test.bzl

@@ -97,6 +97,18 @@ def _rustdoc_for_bin_with_transitive_proc_macro_test_impl(ctx):
 
     return analysistest.end(env)
 
+def _rustdoc_for_lib_with_features_test_impl(ctx):
+    env = analysistest.begin(ctx)
+    tut = analysistest.target_under_test(env)
+
+    _common_rustdoc_checks(env, tut)
+
+    action = _get_rustdoc_action(env, tut)
+
+    assert_argv_contains(env, action, "--cfg=feature=\"docs\"")
+
+    return analysistest.end(env)
+
 def _rustdoc_for_lib_with_cc_lib_test_impl(ctx):
     env = analysistest.begin(ctx)
     tut = analysistest.target_under_test(env)
@@ -164,6 +176,7 @@ rustdoc_for_lib_with_proc_macro_in_docs_test = analysistest.make(_rustdoc_for_li
 rustdoc_for_bin_with_transitive_proc_macro_test = analysistest.make(_rustdoc_for_bin_with_transitive_proc_macro_test_impl)
 rustdoc_for_lib_with_cc_lib_test = analysistest.make(_rustdoc_for_lib_with_cc_lib_test_impl)
 rustdoc_with_args_test = analysistest.make(_rustdoc_with_args_test_impl)
+rustdoc_for_lib_with_features_test = analysistest.make(_rustdoc_for_lib_with_features_test_impl)
 rustdoc_zip_output_test = analysistest.make(_rustdoc_zip_output_test_impl)
 rustdoc_with_json_error_format_test = analysistest.make(_rustdoc_with_json_error_format_test_impl, config_settings = {
     str(Label("//rust/settings:error_format")): "json",
@@ -283,6 +296,18 @@ def _define_targets():
         crate_features = ["with_proc_macro"],
     )
 
+    _target_maker(
+        rust_library,
+        name = "lib_with_features",
+        srcs = ["rustdoc_features.rs"],
+        crate_features = ["docs"],
+    )
+
+    rust_doc(
+        name = "rustdoc_lib_with_features",
+        crate = ":lib_with_features",
+    )
+
     cc_library(
         name = "cc_lib",
         hdrs = ["rustdoc.h"],
@@ -424,6 +449,11 @@ def rustdoc_test_suite(name):
         target_under_test = ":lib_with_cc_doc",
     )
 
+    rustdoc_for_lib_with_features_test(
+        name = "rustdoc_for_lib_with_features_test",
+        target_under_test = ":rustdoc_lib_with_features",
+    )
+
     rustdoc_with_args_test(
         name = "rustdoc_with_args_test",
         target_under_test = ":rustdoc_with_args",
@@ -455,6 +485,7 @@ def rustdoc_test_suite(name):
             ":rustdoc_for_lib_with_proc_macro_in_docs_test",
             ":rustdoc_for_lib_with_proc_macro_test",
             ":rustdoc_for_lib_with_cc_lib_test",
+            ":rustdoc_for_lib_with_features_test",
             ":rustdoc_with_args_test",
             ":rustdoc_with_json_error_format_test",
             ":rustdoc_zip_output_test",