Skip to content

Commit

Permalink
Add doc examples count for --show-coverage
Browse files Browse the repository at this point in the history
  • Loading branch information
GuillaumeGomez committed Aug 18, 2020
1 parent 441fd22 commit d9d84dc
Show file tree
Hide file tree
Showing 3 changed files with 88 additions and 26 deletions.
6 changes: 4 additions & 2 deletions src/librustdoc/html/render/cache.rs
Original file line number Diff line number Diff line change
Expand Up @@ -200,10 +200,12 @@ fn get_index_type_name(clean_type: &clean::Type, accept_generic: bool) -> Option
match *clean_type {
clean::ResolvedPath { ref path, .. } => {
let segments = &path.segments;
let path_segment = segments.iter().last().unwrap_or_else(|| panic!(
let path_segment = segments.iter().last().unwrap_or_else(|| {
panic!(
"get_index_type_name(clean_type: {:?}, accept_generic: {:?}) had length zero path",
clean_type, accept_generic
));
)
});
Some(path_segment.name.clone())
}
clean::Generic(ref s) if accept_generic => Some(s.clone()),
Expand Down
80 changes: 67 additions & 13 deletions src/librustdoc/passes/calculate_doc_coverage.rs
Original file line number Diff line number Diff line change
Expand Up @@ -2,8 +2,9 @@ use crate::clean;
use crate::config::OutputFormat;
use crate::core::DocContext;
use crate::fold::{self, DocFolder};
use crate::html::markdown::{find_testable_code, ErrorCodes};
use crate::passes::doc_test_lints::Tests;
use crate::passes::Pass;

use rustc_span::symbol::sym;
use rustc_span::FileName;
use serde::Serialize;
Expand All @@ -30,15 +31,19 @@ fn calculate_doc_coverage(krate: clean::Crate, ctx: &DocContext<'_>) -> clean::C
struct ItemCount {
total: u64,
with_docs: u64,
with_examples: u64,
}

impl ItemCount {
fn count_item(&mut self, has_docs: bool) {
fn count_item(&mut self, has_docs: bool, has_doc_example: bool) {
self.total += 1;

if has_docs {
self.with_docs += 1;
}
if has_doc_example {
self.with_examples += 1;
}
}

fn percentage(&self) -> Option<f64> {
Expand All @@ -48,20 +53,33 @@ impl ItemCount {
None
}
}

fn examples_percentage(&self) -> Option<f64> {
if self.total > 0 {
Some((self.with_examples as f64 * 100.0) / self.total as f64)
} else {
None
}
}
}

impl ops::Sub for ItemCount {
type Output = Self;

fn sub(self, rhs: Self) -> Self {
ItemCount { total: self.total - rhs.total, with_docs: self.with_docs - rhs.with_docs }
ItemCount {
total: self.total - rhs.total,
with_docs: self.with_docs - rhs.with_docs,
with_examples: self.with_examples - rhs.with_examples,
}
}
}

impl ops::AddAssign for ItemCount {
fn add_assign(&mut self, rhs: Self) {
self.total += rhs.total;
self.with_docs += rhs.with_docs;
self.with_examples += rhs.with_examples;
}
}

Expand Down Expand Up @@ -103,40 +121,73 @@ impl CoverageCalculator {
let mut total = ItemCount::default();

fn print_table_line() {
println!("+-{0:->35}-+-{0:->10}-+-{0:->10}-+-{0:->10}-+", "");
println!("+-{0:->35}-+-{0:->10}-+-{0:->10}-+-{0:->10}-+-{0:->10}-+-{0:->10}-+", "");
}

fn print_table_record(name: &str, count: ItemCount, percentage: f64) {
fn print_table_record(
name: &str,
count: ItemCount,
percentage: f64,
examples_percentage: f64,
) {
println!(
"| {:<35} | {:>10} | {:>10} | {:>9.1}% |",
name, count.with_docs, count.total, percentage
"| {:<35} | {:>10} | {:>10} | {:>9.1}% | {:>10} | {:>9.1}% |",
name,
count.with_docs,
count.total,
percentage,
count.with_examples,
examples_percentage,
);
}

print_table_line();
println!(
"| {:<35} | {:>10} | {:>10} | {:>10} |",
"File", "Documented", "Total", "Percentage"
"| {:<35} | {:>10} | {:>10} | {:>10} | {:>10} | {:>10} |",
"File", "Documented", "Total", "Percentage", "Examples", "Percentage",
);
print_table_line();

for (file, &count) in &self.items {
if let Some(percentage) = count.percentage() {
print_table_record(&limit_filename_len(file.to_string()), count, percentage);
if let (Some(percentage), Some(examples_percentage)) =
(count.percentage(), count.examples_percentage())
{
print_table_record(
&limit_filename_len(file.to_string()),
count,
percentage,
examples_percentage,
);

total += count;
}
}

print_table_line();
print_table_record("Total", total, total.percentage().unwrap_or(0.0));
print_table_record(
"Total",
total,
total.percentage().unwrap_or(0.0),
total.examples_percentage().unwrap_or(0.0),
);
print_table_line();
}
}

impl fold::DocFolder for CoverageCalculator {
fn fold_item(&mut self, i: clean::Item) -> Option<clean::Item> {
let has_docs = !i.attrs.doc_strings.is_empty();
let mut tests = Tests { found_tests: 0 };

find_testable_code(
&i.attrs.doc_strings.iter().map(|d| d.as_str()).collect::<Vec<_>>().join("\n"),
&mut tests,
ErrorCodes::No,
false,
None,
);

let has_doc_example = tests.found_tests != 0;

match i.inner {
_ if !i.def_id.is_local() => {
Expand Down Expand Up @@ -187,7 +238,10 @@ impl fold::DocFolder for CoverageCalculator {
}
_ => {
debug!("counting {:?} {:?} in {}", i.type_(), i.name, i.source.filename);
self.items.entry(i.source.filename.clone()).or_default().count_item(has_docs);
self.items
.entry(i.source.filename.clone())
.or_default()
.count_item(has_docs, has_doc_example);
}
}

Expand Down
28 changes: 17 additions & 11 deletions src/librustdoc/passes/doc_test_lints.rs
Original file line number Diff line number Diff line change
Expand Up @@ -43,6 +43,22 @@ impl<'a, 'tcx> DocFolder for PrivateItemDocTestLinter<'a, 'tcx> {
}
}

pub(crate) struct Tests {
pub(crate) found_tests: usize,
}

impl Tests {
pub(crate) fn new() -> Tests {
Tests { found_tests: 0 }
}
}

impl crate::test::Tester for Tests {
fn add_test(&mut self, _: String, _: LangString, _: usize) {
self.found_tests += 1;
}
}

pub fn look_for_tests<'tcx>(cx: &DocContext<'tcx>, dox: &str, item: &Item) {
let hir_id = match cx.as_local_hir_id(item.def_id) {
Some(hir_id) => hir_id,
Expand All @@ -52,17 +68,7 @@ pub fn look_for_tests<'tcx>(cx: &DocContext<'tcx>, dox: &str, item: &Item) {
}
};

struct Tests {
found_tests: usize,
}

impl crate::test::Tester for Tests {
fn add_test(&mut self, _: String, _: LangString, _: usize) {
self.found_tests += 1;
}
}

let mut tests = Tests { found_tests: 0 };
let mut tests = Tests::new();

find_testable_code(&dox, &mut tests, ErrorCodes::No, false, None);

Expand Down

0 comments on commit d9d84dc

Please sign in to comment.