2020-01-04 19:37:57 -06:00
|
|
|
use rustc::hir::map::Map;
|
2018-12-08 13:30:23 -06:00
|
|
|
use rustc::session::{self, config, DiagnosticOutput};
|
|
|
|
use rustc::util::common::ErrorReported;
|
2020-02-29 11:37:32 -06:00
|
|
|
use rustc_ast::ast;
|
|
|
|
use rustc_ast::with_globals;
|
2019-12-22 16:42:04 -06:00
|
|
|
use rustc_data_structures::sync::Lrc;
|
|
|
|
use rustc_feature::UnstableFeatures;
|
2020-01-04 19:37:57 -06:00
|
|
|
use rustc_hir as hir;
|
2020-01-07 11:12:06 -06:00
|
|
|
use rustc_hir::intravisit;
|
2019-12-22 16:42:04 -06:00
|
|
|
use rustc_interface::interface;
|
2020-01-01 12:40:49 -06:00
|
|
|
use rustc_span::edition::Edition;
|
2020-01-01 12:25:28 -06:00
|
|
|
use rustc_span::source_map::SourceMap;
|
2020-01-01 12:30:57 -06:00
|
|
|
use rustc_span::symbol::sym;
|
2019-12-31 11:15:40 -06:00
|
|
|
use rustc_span::{BytePos, FileName, Pos, Span, DUMMY_SP};
|
2019-12-22 16:42:04 -06:00
|
|
|
use rustc_target::spec::TargetTriple;
|
2018-11-26 20:59:49 -06:00
|
|
|
use std::env;
|
2019-08-29 16:15:31 -05:00
|
|
|
use std::io::{self, Write};
|
|
|
|
use std::panic;
|
2018-12-08 13:30:23 -06:00
|
|
|
use std::path::PathBuf;
|
2019-08-29 16:15:31 -05:00
|
|
|
use std::process::{self, Command, Stdio};
|
2018-11-26 20:59:49 -06:00
|
|
|
use std::str;
|
2019-05-07 22:21:18 -05:00
|
|
|
use tempfile::Builder as TempFileBuilder;
|
2013-12-22 13:23:04 -06:00
|
|
|
|
2019-02-23 01:40:07 -06:00
|
|
|
use crate::clean::Attributes;
|
|
|
|
use crate::config::Options;
|
2019-12-22 16:42:04 -06:00
|
|
|
use crate::html::markdown::{self, ErrorCodes, Ignore, LangString};
|
2013-12-22 13:23:04 -06:00
|
|
|
|
2015-04-06 20:39:39 -05:00
|
|
|
#[derive(Clone, Default)]
|
|
|
|
pub struct TestOptions {
|
2018-03-15 17:10:09 -05:00
|
|
|
/// Whether to disable the default `extern crate my_crate;` when creating doctests.
|
2015-04-06 20:39:39 -05:00
|
|
|
pub no_crate_inject: bool,
|
2018-03-15 17:10:09 -05:00
|
|
|
/// Whether to emit compilation warnings when compiling doctests. Setting this will suppress
|
|
|
|
/// the default `#![allow(unused)]`.
|
|
|
|
pub display_warnings: bool,
|
|
|
|
/// Additional crate-level attributes to add to doctests.
|
2015-04-06 20:39:39 -05:00
|
|
|
pub attrs: Vec<String>,
|
|
|
|
}
|
|
|
|
|
2018-12-08 13:30:23 -06:00
|
|
|
pub fn run(options: Options) -> i32 {
|
2018-10-30 13:35:10 -05:00
|
|
|
let input = config::Input::File(options.input.clone());
|
2013-12-22 13:23:04 -06:00
|
|
|
|
2019-07-20 15:34:41 -05:00
|
|
|
let crate_types = if options.proc_macro_crate {
|
|
|
|
vec![config::CrateType::ProcMacro]
|
|
|
|
} else {
|
2020-01-18 17:51:06 -06:00
|
|
|
vec![config::CrateType::Rlib]
|
2019-07-20 15:34:41 -05:00
|
|
|
};
|
|
|
|
|
2014-05-06 06:38:01 -05:00
|
|
|
let sessopts = config::Options {
|
2019-06-02 06:39:50 -05:00
|
|
|
maybe_sysroot: options.maybe_sysroot.clone(),
|
2018-10-30 13:35:10 -05:00
|
|
|
search_paths: options.libs.clone(),
|
2019-07-20 15:34:41 -05:00
|
|
|
crate_types,
|
2018-10-30 13:35:10 -05:00
|
|
|
cg: options.codegen_options.clone(),
|
|
|
|
externs: options.externs.clone(),
|
2016-09-24 12:20:57 -05:00
|
|
|
unstable_features: UnstableFeatures::from_environment(),
|
2017-07-19 03:09:22 -05:00
|
|
|
lint_cap: Some(::rustc::lint::Level::Allow),
|
2017-01-18 15:31:29 -06:00
|
|
|
actually_rustdoc: true,
|
2019-12-22 16:42:04 -06:00
|
|
|
debugging_opts: config::DebuggingOptions { ..config::basic_debugging_options() },
|
2018-10-30 13:35:10 -05:00
|
|
|
edition: options.edition,
|
2019-06-12 12:49:41 -05:00
|
|
|
target_triple: options.target.clone(),
|
2018-07-26 13:36:11 -05:00
|
|
|
..config::Options::default()
|
2013-12-22 13:23:04 -06:00
|
|
|
};
|
2018-04-25 17:49:52 -05:00
|
|
|
|
2019-09-25 10:08:40 -05:00
|
|
|
let mut cfgs = options.cfgs.clone();
|
2019-11-06 11:32:51 -06:00
|
|
|
cfgs.push("doc".to_owned());
|
2019-10-07 16:08:54 -05:00
|
|
|
cfgs.push("doctest".to_owned());
|
2018-12-08 13:30:23 -06:00
|
|
|
let config = interface::Config {
|
|
|
|
opts: sessopts,
|
2019-10-11 16:48:16 -05:00
|
|
|
crate_cfg: interface::parse_cfgspecs(cfgs),
|
2018-12-08 13:30:23 -06:00
|
|
|
input,
|
|
|
|
input_path: None,
|
|
|
|
output_file: None,
|
|
|
|
output_dir: None,
|
|
|
|
file_loader: None,
|
|
|
|
diagnostic_output: DiagnosticOutput::Default,
|
|
|
|
stderr: None,
|
|
|
|
crate_name: options.crate_name.clone(),
|
|
|
|
lint_caps: Default::default(),
|
2019-10-10 18:33:00 -05:00
|
|
|
register_lints: None,
|
2019-11-11 09:09:03 -06:00
|
|
|
override_queries: None,
|
2019-11-15 12:41:50 -06:00
|
|
|
registry: rustc_driver::diagnostics_registry(),
|
2018-12-08 13:30:23 -06:00
|
|
|
};
|
|
|
|
|
|
|
|
let mut test_args = options.test_args.clone();
|
|
|
|
let display_warnings = options.display_warnings;
|
|
|
|
|
2019-12-22 16:42:04 -06:00
|
|
|
let tests = interface::run_compiler(config, |compiler| {
|
|
|
|
compiler.enter(|queries| {
|
|
|
|
let lower_to_hir = queries.lower_to_hir()?;
|
|
|
|
|
2020-02-06 06:41:37 -06:00
|
|
|
let mut opts = scrape_test_config(lower_to_hir.peek().0);
|
2019-12-22 16:42:04 -06:00
|
|
|
opts.display_warnings |= options.display_warnings;
|
|
|
|
let enable_per_target_ignores = options.enable_per_target_ignores;
|
|
|
|
let mut collector = Collector::new(
|
|
|
|
queries.crate_name()?.peek().to_string(),
|
|
|
|
options,
|
|
|
|
false,
|
|
|
|
opts,
|
|
|
|
Some(compiler.source_map().clone()),
|
|
|
|
None,
|
|
|
|
enable_per_target_ignores,
|
|
|
|
);
|
|
|
|
|
|
|
|
let mut global_ctxt = queries.global_ctxt()?.take();
|
|
|
|
|
|
|
|
global_ctxt.enter(|tcx| {
|
|
|
|
let krate = tcx.hir().krate();
|
|
|
|
let mut hir_collector = HirCollector {
|
|
|
|
sess: compiler.session(),
|
|
|
|
collector: &mut collector,
|
2020-02-06 04:59:29 -06:00
|
|
|
map: *tcx.hir(),
|
2019-12-22 16:42:04 -06:00
|
|
|
codes: ErrorCodes::from(
|
|
|
|
compiler.session().opts.unstable_features.is_nightly_build(),
|
|
|
|
),
|
|
|
|
};
|
|
|
|
hir_collector.visit_testable("".to_string(), &krate.attrs, |this| {
|
|
|
|
intravisit::walk_crate(this, krate);
|
|
|
|
});
|
2018-04-25 17:49:52 -05:00
|
|
|
});
|
2020-01-18 17:51:06 -06:00
|
|
|
compiler.session().abort_if_errors();
|
2013-12-22 13:23:04 -06:00
|
|
|
|
2019-12-22 16:42:04 -06:00
|
|
|
let ret: Result<_, ErrorReported> = Ok(collector.tests);
|
|
|
|
ret
|
|
|
|
})
|
2020-01-18 17:51:06 -06:00
|
|
|
});
|
|
|
|
let tests = match tests {
|
|
|
|
Ok(tests) => tests,
|
|
|
|
Err(ErrorReported) => return 1,
|
|
|
|
};
|
2013-12-22 13:23:04 -06:00
|
|
|
|
2018-12-08 13:30:23 -06:00
|
|
|
test_args.insert(0, "rustdoctest".to_string());
|
|
|
|
|
|
|
|
testing::test_main(
|
|
|
|
&test_args,
|
|
|
|
tests,
|
2019-12-22 16:42:04 -06:00
|
|
|
Some(testing::Options::new().display_output(display_warnings)),
|
2018-12-08 13:30:23 -06:00
|
|
|
);
|
|
|
|
|
|
|
|
0
|
2013-12-22 13:23:04 -06:00
|
|
|
}
|
|
|
|
|
2018-11-26 20:59:49 -06:00
|
|
|
// Look for `#![doc(test(no_crate_inject))]`, used by crates in the std facade.
|
2020-01-04 19:37:57 -06:00
|
|
|
fn scrape_test_config(krate: &::rustc_hir::Crate) -> TestOptions {
|
2020-01-11 10:02:46 -06:00
|
|
|
use rustc_ast_pretty::pprust;
|
2015-03-15 19:54:30 -05:00
|
|
|
|
2019-12-22 16:42:04 -06:00
|
|
|
let mut opts =
|
|
|
|
TestOptions { no_crate_inject: false, display_warnings: false, attrs: Vec::new() };
|
2015-04-06 20:39:39 -05:00
|
|
|
|
2019-12-22 16:42:04 -06:00
|
|
|
let test_attrs: Vec<_> = krate
|
|
|
|
.attrs
|
|
|
|
.iter()
|
2019-05-07 22:21:18 -05:00
|
|
|
.filter(|a| a.check_name(sym::doc))
|
2017-03-03 03:23:59 -06:00
|
|
|
.flat_map(|a| a.meta_item_list().unwrap_or_else(Vec::new))
|
2019-05-07 22:21:18 -05:00
|
|
|
.filter(|a| a.check_name(sym::test))
|
2017-03-03 03:23:59 -06:00
|
|
|
.collect();
|
|
|
|
let attrs = test_attrs.iter().flat_map(|a| a.meta_item_list().unwrap_or(&[]));
|
|
|
|
|
2015-04-06 20:39:39 -05:00
|
|
|
for attr in attrs {
|
2019-05-07 22:21:18 -05:00
|
|
|
if attr.check_name(sym::no_crate_inject) {
|
2015-04-06 20:39:39 -05:00
|
|
|
opts.no_crate_inject = true;
|
|
|
|
}
|
2019-05-07 22:21:18 -05:00
|
|
|
if attr.check_name(sym::attr) {
|
2015-04-06 20:39:39 -05:00
|
|
|
if let Some(l) = attr.meta_item_list() {
|
|
|
|
for item in l {
|
2016-08-19 20:58:14 -05:00
|
|
|
opts.attrs.push(pprust::meta_list_item_to_string(item));
|
2015-03-15 19:54:30 -05:00
|
|
|
}
|
|
|
|
}
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
2016-10-01 15:47:43 -05:00
|
|
|
opts
|
2015-03-15 19:54:30 -05:00
|
|
|
}
|
|
|
|
|
2019-05-04 14:37:12 -05:00
|
|
|
/// Documentation test failure modes.
|
|
|
|
enum TestFailure {
|
|
|
|
/// The test failed to compile.
|
|
|
|
CompileError,
|
|
|
|
/// The test is marked `compile_fail` but compiled successfully.
|
|
|
|
UnexpectedCompilePass,
|
|
|
|
/// The test failed to compile (as expected) but the compiler output did not contain all
|
|
|
|
/// expected error codes.
|
|
|
|
MissingErrorCodes(Vec<String>),
|
|
|
|
/// The test binary was unable to be executed.
|
|
|
|
ExecutionError(io::Error),
|
|
|
|
/// The test binary exited with a non-zero exit code.
|
|
|
|
///
|
|
|
|
/// This typically means an assertion in the test failed or another form of panic occurred.
|
|
|
|
ExecutionFailure(process::Output),
|
|
|
|
/// The test is marked `should_panic` but the test binary executed successfully.
|
|
|
|
UnexpectedRunPass,
|
|
|
|
}
|
|
|
|
|
|
|
|
fn run_test(
|
|
|
|
test: &str,
|
|
|
|
cratename: &str,
|
|
|
|
filename: &FileName,
|
|
|
|
line: usize,
|
2019-08-29 16:15:31 -05:00
|
|
|
options: Options,
|
2019-05-04 14:37:12 -05:00
|
|
|
should_panic: bool,
|
|
|
|
no_run: bool,
|
|
|
|
as_test_harness: bool,
|
2019-04-26 15:52:56 -05:00
|
|
|
runtool: Option<String>,
|
|
|
|
runtool_args: Vec<String>,
|
2019-06-11 13:06:34 -05:00
|
|
|
target: TargetTriple,
|
2019-05-04 14:37:12 -05:00
|
|
|
compile_fail: bool,
|
|
|
|
mut error_codes: Vec<String>,
|
|
|
|
opts: &TestOptions,
|
|
|
|
edition: Edition,
|
|
|
|
) -> Result<(), TestFailure> {
|
2020-01-02 17:34:00 -06:00
|
|
|
let (test, line_offset) = make_test(test, Some(cratename), as_test_harness, opts, edition);
|
2019-04-23 16:39:26 -05:00
|
|
|
|
2017-10-03 05:35:55 -05:00
|
|
|
// FIXME(#44940): if doctests ever support path remapping, then this filename
|
2018-11-26 20:59:49 -06:00
|
|
|
// needs to be the result of `SourceMap::span_to_unmapped_path`.
|
2018-12-04 14:18:03 -06:00
|
|
|
let path = match filename {
|
|
|
|
FileName::Real(path) => path.clone(),
|
|
|
|
_ => PathBuf::from(r"doctest.rs"),
|
|
|
|
};
|
|
|
|
|
2018-12-08 13:17:50 -06:00
|
|
|
enum DirState {
|
|
|
|
Temp(tempfile::TempDir),
|
|
|
|
Perm(PathBuf),
|
|
|
|
}
|
|
|
|
|
|
|
|
impl DirState {
|
|
|
|
fn path(&self) -> &std::path::Path {
|
|
|
|
match self {
|
|
|
|
DirState::Temp(t) => t.path(),
|
|
|
|
DirState::Perm(p) => p.as_path(),
|
|
|
|
}
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
2019-08-29 16:15:31 -05:00
|
|
|
let outdir = if let Some(mut path) = options.persist_doctests {
|
2019-12-22 16:42:04 -06:00
|
|
|
path.push(format!(
|
|
|
|
"{}_{}",
|
|
|
|
filename.to_string().rsplit('/').next().unwrap().replace(".", "_"),
|
|
|
|
line
|
|
|
|
));
|
|
|
|
std::fs::create_dir_all(&path).expect("Couldn't create directory for doctest executables");
|
2016-01-05 16:38:11 -06:00
|
|
|
|
2018-12-08 13:30:23 -06:00
|
|
|
DirState::Perm(path)
|
|
|
|
} else {
|
2019-12-22 16:42:04 -06:00
|
|
|
DirState::Temp(
|
|
|
|
TempFileBuilder::new()
|
|
|
|
.prefix("rustdoctest")
|
|
|
|
.tempdir()
|
|
|
|
.expect("rustdoc needs a tempdir"),
|
|
|
|
)
|
2018-12-08 13:30:23 -06:00
|
|
|
};
|
|
|
|
let output_file = outdir.path().join("rust_out");
|
|
|
|
|
2019-12-22 16:42:04 -06:00
|
|
|
let rustc_binary = options
|
|
|
|
.test_builder
|
|
|
|
.as_ref()
|
|
|
|
.map(|v| &**v)
|
|
|
|
.unwrap_or_else(|| rustc_interface::util::rustc_path().expect("found rustc"));
|
2019-09-09 20:11:27 -05:00
|
|
|
let mut compiler = Command::new(&rustc_binary);
|
2019-08-29 16:15:31 -05:00
|
|
|
compiler.arg("--crate-type").arg("bin");
|
|
|
|
for cfg in &options.cfgs {
|
|
|
|
compiler.arg("--cfg").arg(&cfg);
|
|
|
|
}
|
|
|
|
if let Some(sysroot) = options.maybe_sysroot {
|
|
|
|
compiler.arg("--sysroot").arg(sysroot);
|
|
|
|
}
|
|
|
|
compiler.arg("--edition").arg(&edition.to_string());
|
|
|
|
compiler.env("UNSTABLE_RUSTDOC_TEST_PATH", path);
|
2019-12-22 16:42:04 -06:00
|
|
|
compiler.env("UNSTABLE_RUSTDOC_TEST_LINE", format!("{}", line as isize - line_offset as isize));
|
2019-08-29 16:15:31 -05:00
|
|
|
compiler.arg("-o").arg(&output_file);
|
|
|
|
if as_test_harness {
|
|
|
|
compiler.arg("--test");
|
|
|
|
}
|
|
|
|
for lib_str in &options.lib_strs {
|
|
|
|
compiler.arg("-L").arg(&lib_str);
|
|
|
|
}
|
|
|
|
for extern_str in &options.extern_strs {
|
|
|
|
compiler.arg("--extern").arg(&extern_str);
|
|
|
|
}
|
2019-08-30 14:26:04 -05:00
|
|
|
compiler.arg("-Ccodegen-units=1");
|
2019-08-29 16:15:31 -05:00
|
|
|
for codegen_options_str in &options.codegen_options_strs {
|
|
|
|
compiler.arg("-C").arg(&codegen_options_str);
|
|
|
|
}
|
2019-10-10 19:00:00 -05:00
|
|
|
for debugging_option_str in &options.debugging_options_strs {
|
|
|
|
compiler.arg("-Z").arg(&debugging_option_str);
|
|
|
|
}
|
2020-01-30 08:49:21 -06:00
|
|
|
if no_run && !compile_fail {
|
2019-08-29 16:15:31 -05:00
|
|
|
compiler.arg("--emit=metadata");
|
|
|
|
}
|
2019-08-27 17:44:11 -05:00
|
|
|
compiler.arg("--target").arg(target.to_string());
|
2017-07-02 08:09:09 -05:00
|
|
|
|
2019-08-29 16:15:31 -05:00
|
|
|
compiler.arg("-");
|
|
|
|
compiler.stdin(Stdio::piped());
|
|
|
|
compiler.stderr(Stdio::piped());
|
|
|
|
|
|
|
|
let mut child = compiler.spawn().expect("Failed to spawn rustc process");
|
|
|
|
{
|
|
|
|
let stdin = child.stdin.as_mut().expect("Failed to open stdin");
|
|
|
|
stdin.write_all(test.as_bytes()).expect("could write out test sources");
|
|
|
|
}
|
|
|
|
let output = child.wait_with_output().expect("Failed to read stdout");
|
2018-07-09 19:11:06 -05:00
|
|
|
|
2019-08-29 16:15:31 -05:00
|
|
|
struct Bomb<'a>(&'a str);
|
|
|
|
impl Drop for Bomb<'_> {
|
|
|
|
fn drop(&mut self) {
|
2019-12-22 16:42:04 -06:00
|
|
|
eprint!("{}", self.0);
|
2019-08-29 16:15:31 -05:00
|
|
|
}
|
|
|
|
}
|
|
|
|
let out = str::from_utf8(&output.stderr).unwrap();
|
|
|
|
let _bomb = Bomb(&out);
|
|
|
|
match (output.status.success(), compile_fail) {
|
|
|
|
(true, true) => {
|
2019-05-04 14:37:12 -05:00
|
|
|
return Err(TestFailure::UnexpectedCompilePass);
|
2018-07-09 19:11:06 -05:00
|
|
|
}
|
2019-08-29 16:15:31 -05:00
|
|
|
(true, false) => {}
|
|
|
|
(false, true) => {
|
2019-05-04 14:37:12 -05:00
|
|
|
if !error_codes.is_empty() {
|
2020-01-03 17:28:15 -06:00
|
|
|
error_codes.retain(|err| !out.contains(&format!("error[{}]: ", err)));
|
2019-05-04 14:37:12 -05:00
|
|
|
|
|
|
|
if !error_codes.is_empty() {
|
|
|
|
return Err(TestFailure::MissingErrorCodes(error_codes));
|
|
|
|
}
|
2016-06-09 16:50:52 -05:00
|
|
|
}
|
|
|
|
}
|
2019-08-29 16:15:31 -05:00
|
|
|
(false, false) => {
|
2019-05-04 14:37:12 -05:00
|
|
|
return Err(TestFailure::CompileError);
|
2017-07-02 08:09:09 -05:00
|
|
|
}
|
2018-07-09 19:11:06 -05:00
|
|
|
}
|
2016-06-09 16:50:52 -05:00
|
|
|
|
2019-05-04 14:37:12 -05:00
|
|
|
if no_run {
|
|
|
|
return Ok(());
|
2018-07-09 19:11:06 -05:00
|
|
|
}
|
2013-12-22 13:23:04 -06:00
|
|
|
|
2014-02-28 14:33:49 -06:00
|
|
|
// Run the code!
|
2019-04-26 15:52:56 -05:00
|
|
|
let mut cmd;
|
|
|
|
|
|
|
|
if let Some(tool) = runtool {
|
|
|
|
cmd = Command::new(tool);
|
|
|
|
cmd.arg(output_file);
|
|
|
|
cmd.args(runtool_args);
|
2019-06-06 18:01:53 -05:00
|
|
|
} else {
|
2019-04-26 15:52:56 -05:00
|
|
|
cmd = Command::new(output_file);
|
|
|
|
}
|
2018-12-08 13:30:23 -06:00
|
|
|
|
2014-07-02 15:50:45 -05:00
|
|
|
match cmd.output() {
|
2019-05-04 14:37:12 -05:00
|
|
|
Err(e) => return Err(TestFailure::ExecutionError(e)),
|
2014-01-30 13:30:21 -06:00
|
|
|
Ok(out) => {
|
2015-01-31 17:08:25 -06:00
|
|
|
if should_panic && out.status.success() {
|
2019-05-04 14:37:12 -05:00
|
|
|
return Err(TestFailure::UnexpectedRunPass);
|
2015-01-31 17:08:25 -06:00
|
|
|
} else if !should_panic && !out.status.success() {
|
2019-05-04 14:37:12 -05:00
|
|
|
return Err(TestFailure::ExecutionFailure(out));
|
2013-12-22 13:23:04 -06:00
|
|
|
}
|
|
|
|
}
|
|
|
|
}
|
2019-05-04 14:37:12 -05:00
|
|
|
|
|
|
|
Ok(())
|
2013-12-22 13:23:04 -06:00
|
|
|
}
|
|
|
|
|
2019-04-23 16:39:26 -05:00
|
|
|
/// Transforms a test into code that can be compiled into a Rust binary, and returns the number of
|
|
|
|
/// lines before the test code begins.
|
2019-12-22 16:42:04 -06:00
|
|
|
pub fn make_test(
|
|
|
|
s: &str,
|
|
|
|
cratename: Option<&str>,
|
|
|
|
dont_insert_main: bool,
|
|
|
|
opts: &TestOptions,
|
|
|
|
edition: Edition,
|
|
|
|
) -> (String, usize) {
|
2018-10-20 13:45:44 -05:00
|
|
|
let (crate_attrs, everything_else, crates) = partition_source(s);
|
2018-02-09 09:24:23 -06:00
|
|
|
let everything_else = everything_else.trim();
|
2018-01-08 08:47:23 -06:00
|
|
|
let mut line_offset = 0;
|
2014-06-06 11:12:18 -05:00
|
|
|
let mut prog = String::new();
|
2015-03-12 15:01:06 -05:00
|
|
|
|
2018-03-15 17:10:09 -05:00
|
|
|
if opts.attrs.is_empty() && !opts.display_warnings {
|
2017-11-04 14:13:44 -05:00
|
|
|
// If there aren't any attributes supplied by #![doc(test(attr(...)))], then allow some
|
|
|
|
// lints that are commonly triggered in doctests. The crate-level test attributes are
|
|
|
|
// commonly used to make tests fail in case they trigger warnings, so having this there in
|
|
|
|
// that case may cause some tests to pass when they shouldn't have.
|
|
|
|
prog.push_str("#![allow(unused)]\n");
|
2018-01-08 08:47:23 -06:00
|
|
|
line_offset += 1;
|
2017-11-04 14:13:44 -05:00
|
|
|
}
|
2015-03-12 15:01:06 -05:00
|
|
|
|
2017-11-04 14:13:44 -05:00
|
|
|
// Next, any attributes that came from the crate root via #![doc(test(attr(...)))].
|
2015-04-06 20:39:39 -05:00
|
|
|
for attr in &opts.attrs {
|
|
|
|
prog.push_str(&format!("#![{}]\n", attr));
|
2018-01-08 08:47:23 -06:00
|
|
|
line_offset += 1;
|
2014-06-06 11:12:18 -05:00
|
|
|
}
|
2014-03-08 04:30:43 -06:00
|
|
|
|
2017-11-04 14:13:44 -05:00
|
|
|
// Now push any outer attributes from the example, assuming they
|
|
|
|
// are intended to be crate attributes.
|
|
|
|
prog.push_str(&crate_attrs);
|
2018-12-13 14:31:17 -06:00
|
|
|
prog.push_str(&crates);
|
2017-11-04 14:13:44 -05:00
|
|
|
|
2020-02-29 11:16:26 -06:00
|
|
|
// Uses librustc_ast to parse the doctest and find if there's a main fn and the extern
|
2018-10-05 18:22:19 -05:00
|
|
|
// crate already is included.
|
2020-01-02 17:34:00 -06:00
|
|
|
let result = rustc_driver::catch_fatal_errors(|| {
|
|
|
|
with_globals(edition, || {
|
2020-01-09 04:18:47 -06:00
|
|
|
use rustc_errors::emitter::EmitterWriter;
|
|
|
|
use rustc_errors::Handler;
|
2020-01-02 17:34:00 -06:00
|
|
|
use rustc_parse::maybe_new_parser_from_source_str;
|
2020-01-11 08:03:15 -06:00
|
|
|
use rustc_session::parse::ParseSess;
|
2020-01-02 17:34:00 -06:00
|
|
|
use rustc_span::source_map::FilePathMapping;
|
|
|
|
|
|
|
|
let filename = FileName::anon_source_code(s);
|
2020-02-28 06:36:45 -06:00
|
|
|
let source = crates + everything_else;
|
2020-01-02 17:34:00 -06:00
|
|
|
|
|
|
|
// Any errors in parsing should also appear when the doctest is compiled for real, so just
|
2020-02-29 11:16:26 -06:00
|
|
|
// send all the errors that librustc_ast emits directly into a `Sink` instead of stderr.
|
2020-02-22 08:07:05 -06:00
|
|
|
let sm = Lrc::new(SourceMap::new(FilePathMapping::empty()));
|
2020-01-02 17:34:00 -06:00
|
|
|
let emitter =
|
|
|
|
EmitterWriter::new(box io::sink(), None, false, false, false, None, false);
|
|
|
|
// FIXME(misdreavus): pass `-Z treat-err-as-bug` to the doctest parser
|
|
|
|
let handler = Handler::with_emitter(false, None, box emitter);
|
2020-02-22 08:07:05 -06:00
|
|
|
let sess = ParseSess::with_span_handler(handler, sm);
|
2020-01-02 17:34:00 -06:00
|
|
|
|
|
|
|
let mut found_main = false;
|
|
|
|
let mut found_extern_crate = cratename.is_none();
|
|
|
|
let mut found_macro = false;
|
|
|
|
|
|
|
|
let mut parser = match maybe_new_parser_from_source_str(&sess, filename, source) {
|
|
|
|
Ok(p) => p,
|
|
|
|
Err(errs) => {
|
|
|
|
for mut err in errs {
|
|
|
|
err.cancel();
|
|
|
|
}
|
2018-11-01 11:57:29 -05:00
|
|
|
|
2020-01-02 17:34:00 -06:00
|
|
|
return (found_main, found_extern_crate, found_macro);
|
|
|
|
}
|
|
|
|
};
|
|
|
|
|
|
|
|
loop {
|
|
|
|
match parser.parse_item() {
|
|
|
|
Ok(Some(item)) => {
|
|
|
|
if !found_main {
|
|
|
|
if let ast::ItemKind::Fn(..) = item.kind {
|
|
|
|
if item.ident.name == sym::main {
|
|
|
|
found_main = true;
|
|
|
|
}
|
2018-10-20 13:45:44 -05:00
|
|
|
}
|
|
|
|
}
|
2018-10-07 11:36:47 -05:00
|
|
|
|
2020-01-02 17:34:00 -06:00
|
|
|
if !found_extern_crate {
|
|
|
|
if let ast::ItemKind::ExternCrate(original) = item.kind {
|
|
|
|
// This code will never be reached if `cratename` is none because
|
|
|
|
// `found_extern_crate` is initialized to `true` if it is none.
|
|
|
|
let cratename = cratename.unwrap();
|
2018-10-20 13:45:44 -05:00
|
|
|
|
2020-01-02 17:34:00 -06:00
|
|
|
match original {
|
|
|
|
Some(name) => found_extern_crate = name.as_str() == cratename,
|
|
|
|
None => found_extern_crate = item.ident.as_str() == cratename,
|
|
|
|
}
|
2018-10-20 13:45:44 -05:00
|
|
|
}
|
|
|
|
}
|
2018-10-05 18:22:19 -05:00
|
|
|
|
2020-01-02 17:34:00 -06:00
|
|
|
if !found_macro {
|
|
|
|
if let ast::ItemKind::Mac(..) = item.kind {
|
|
|
|
found_macro = true;
|
|
|
|
}
|
2018-12-20 15:10:07 -06:00
|
|
|
}
|
|
|
|
|
2020-01-02 17:34:00 -06:00
|
|
|
if found_main && found_extern_crate {
|
|
|
|
break;
|
|
|
|
}
|
|
|
|
}
|
|
|
|
Ok(None) => break,
|
|
|
|
Err(mut e) => {
|
|
|
|
e.cancel();
|
2018-10-20 13:45:44 -05:00
|
|
|
break;
|
2018-10-05 18:22:19 -05:00
|
|
|
}
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
2020-01-02 17:34:00 -06:00
|
|
|
(found_main, found_extern_crate, found_macro)
|
|
|
|
})
|
2018-10-05 18:22:19 -05:00
|
|
|
});
|
2020-01-02 17:34:00 -06:00
|
|
|
let (already_has_main, already_has_extern_crate, found_macro) = match result {
|
|
|
|
Ok(result) => result,
|
|
|
|
Err(ErrorReported) => {
|
|
|
|
// If the parser panicked due to a fatal error, pass the test code through unchanged.
|
|
|
|
// The error will be reported during compilation.
|
|
|
|
return (s.to_owned(), 0);
|
|
|
|
}
|
|
|
|
};
|
2018-10-05 18:22:19 -05:00
|
|
|
|
2018-12-20 15:10:07 -06:00
|
|
|
// If a doctest's `fn main` is being masked by a wrapper macro, the parsing loop above won't
|
|
|
|
// see it. In that case, run the old text-based scan to see if they at least have a main
|
|
|
|
// function written inside a macro invocation. See
|
|
|
|
// https://github.com/rust-lang/rust/issues/56898
|
|
|
|
let already_has_main = if found_macro && !already_has_main {
|
|
|
|
s.lines()
|
|
|
|
.map(|line| {
|
|
|
|
let comment = line.find("//");
|
2019-12-22 16:42:04 -06:00
|
|
|
if let Some(comment_begins) = comment { &line[0..comment_begins] } else { line }
|
2018-12-20 15:10:07 -06:00
|
|
|
})
|
|
|
|
.any(|code| code.contains("fn main"))
|
|
|
|
} else {
|
|
|
|
already_has_main
|
|
|
|
};
|
|
|
|
|
2014-06-18 03:07:59 -05:00
|
|
|
// Don't inject `extern crate std` because it's already injected by the
|
|
|
|
// compiler.
|
2018-10-05 18:22:19 -05:00
|
|
|
if !already_has_extern_crate && !opts.no_crate_inject && cratename != Some("std") {
|
2016-02-28 05:11:13 -06:00
|
|
|
if let Some(cratename) = cratename {
|
2018-10-05 18:22:19 -05:00
|
|
|
// Make sure its actually used if not included.
|
2016-02-28 05:11:13 -06:00
|
|
|
if s.contains(cratename) {
|
|
|
|
prog.push_str(&format!("extern crate {};\n", cratename));
|
2018-01-08 08:47:23 -06:00
|
|
|
line_offset += 1;
|
2014-06-06 11:12:18 -05:00
|
|
|
}
|
2014-02-17 22:43:32 -06:00
|
|
|
}
|
2013-12-22 13:23:04 -06:00
|
|
|
}
|
2017-09-20 00:58:33 -05:00
|
|
|
|
2018-12-03 11:16:20 -06:00
|
|
|
// FIXME: This code cannot yet handle no_std test cases yet
|
|
|
|
if dont_insert_main || already_has_main || prog.contains("![no_std]") {
|
2018-02-09 09:24:23 -06:00
|
|
|
prog.push_str(everything_else);
|
2013-12-22 13:23:04 -06:00
|
|
|
} else {
|
2018-12-03 11:16:20 -06:00
|
|
|
let returns_result = everything_else.trim_end().ends_with("(())");
|
|
|
|
let (main_pre, main_post) = if returns_result {
|
2019-12-22 16:42:04 -06:00
|
|
|
(
|
|
|
|
"fn main() { fn _inner() -> Result<(), impl core::fmt::Debug> {",
|
|
|
|
"}\n_inner().unwrap() }",
|
|
|
|
)
|
2018-12-03 11:16:20 -06:00
|
|
|
} else {
|
|
|
|
("fn main() {\n", "\n}")
|
|
|
|
};
|
|
|
|
prog.extend([main_pre, everything_else, main_post].iter().cloned());
|
2018-01-08 08:47:23 -06:00
|
|
|
line_offset += 1;
|
2013-12-22 13:23:04 -06:00
|
|
|
}
|
|
|
|
|
2018-12-13 14:31:17 -06:00
|
|
|
debug!("final doctest:\n{}", prog);
|
|
|
|
|
2018-01-08 08:47:23 -06:00
|
|
|
(prog, line_offset)
|
2013-12-22 13:23:04 -06:00
|
|
|
}
|
|
|
|
|
2016-12-04 13:07:27 -06:00
|
|
|
// FIXME(aburka): use a real parser to deal with multiline attributes
|
2018-10-20 13:45:44 -05:00
|
|
|
fn partition_source(s: &str) -> (String, String, String) {
|
2018-12-13 14:31:42 -06:00
|
|
|
#[derive(Copy, Clone, PartialEq)]
|
|
|
|
enum PartitionState {
|
|
|
|
Attrs,
|
|
|
|
Crates,
|
|
|
|
Other,
|
|
|
|
}
|
|
|
|
let mut state = PartitionState::Attrs;
|
2015-03-12 15:01:06 -05:00
|
|
|
let mut before = String::new();
|
2018-10-20 13:45:44 -05:00
|
|
|
let mut crates = String::new();
|
2015-03-12 15:01:06 -05:00
|
|
|
let mut after = String::new();
|
|
|
|
|
|
|
|
for line in s.lines() {
|
2015-03-15 19:54:30 -05:00
|
|
|
let trimline = line.trim();
|
2018-12-13 14:31:42 -06:00
|
|
|
|
|
|
|
// FIXME(misdreavus): if a doc comment is placed on an extern crate statement, it will be
|
|
|
|
// shunted into "everything else"
|
|
|
|
match state {
|
|
|
|
PartitionState::Attrs => {
|
2019-12-22 16:42:04 -06:00
|
|
|
state = if trimline.starts_with("#![")
|
|
|
|
|| trimline.chars().all(|c| c.is_whitespace())
|
|
|
|
|| (trimline.starts_with("//") && !trimline.starts_with("///"))
|
2018-12-13 14:31:42 -06:00
|
|
|
{
|
|
|
|
PartitionState::Attrs
|
2019-12-22 16:42:04 -06:00
|
|
|
} else if trimline.starts_with("extern crate")
|
|
|
|
|| trimline.starts_with("#[macro_use] extern crate")
|
2018-12-13 14:31:42 -06:00
|
|
|
{
|
|
|
|
PartitionState::Crates
|
|
|
|
} else {
|
|
|
|
PartitionState::Other
|
|
|
|
};
|
|
|
|
}
|
|
|
|
PartitionState::Crates => {
|
2019-12-22 16:42:04 -06:00
|
|
|
state = if trimline.starts_with("extern crate")
|
|
|
|
|| trimline.starts_with("#[macro_use] extern crate")
|
|
|
|
|| trimline.chars().all(|c| c.is_whitespace())
|
|
|
|
|| (trimline.starts_with("//") && !trimline.starts_with("///"))
|
2018-12-13 14:31:42 -06:00
|
|
|
{
|
|
|
|
PartitionState::Crates
|
|
|
|
} else {
|
|
|
|
PartitionState::Other
|
|
|
|
};
|
|
|
|
}
|
|
|
|
PartitionState::Other => {}
|
|
|
|
}
|
|
|
|
|
|
|
|
match state {
|
|
|
|
PartitionState::Attrs => {
|
|
|
|
before.push_str(line);
|
|
|
|
before.push_str("\n");
|
|
|
|
}
|
|
|
|
PartitionState::Crates => {
|
2018-10-20 13:45:44 -05:00
|
|
|
crates.push_str(line);
|
2018-11-01 08:58:28 -05:00
|
|
|
crates.push_str("\n");
|
2018-10-20 13:45:44 -05:00
|
|
|
}
|
2018-12-13 14:31:42 -06:00
|
|
|
PartitionState::Other => {
|
|
|
|
after.push_str(line);
|
|
|
|
after.push_str("\n");
|
|
|
|
}
|
2015-03-12 15:01:06 -05:00
|
|
|
}
|
|
|
|
}
|
|
|
|
|
2018-12-13 14:31:42 -06:00
|
|
|
debug!("before:\n{}", before);
|
|
|
|
debug!("crates:\n{}", crates);
|
|
|
|
debug!("after:\n{}", after);
|
|
|
|
|
2018-10-20 13:45:44 -05:00
|
|
|
(before, after, crates)
|
2015-03-12 15:01:06 -05:00
|
|
|
}
|
|
|
|
|
2018-09-17 17:25:50 -05:00
|
|
|
pub trait Tester {
|
|
|
|
fn add_test(&mut self, test: String, config: LangString, line: usize);
|
|
|
|
fn get_line(&self) -> usize {
|
|
|
|
0
|
|
|
|
}
|
|
|
|
fn register_header(&mut self, _name: &str, _level: u32) {}
|
|
|
|
}
|
|
|
|
|
2013-12-22 13:23:04 -06:00
|
|
|
pub struct Collector {
|
2014-03-28 12:27:24 -05:00
|
|
|
pub tests: Vec<testing::TestDescAndFn>,
|
2017-09-16 02:59:15 -05:00
|
|
|
|
|
|
|
// The name of the test displayed to the user, separated by `::`.
|
|
|
|
//
|
|
|
|
// In tests from Rust source, this is the path to the item
|
2018-11-26 20:59:49 -06:00
|
|
|
// e.g., `["std", "vec", "Vec", "push"]`.
|
2017-09-16 02:59:15 -05:00
|
|
|
//
|
|
|
|
// In tests from a markdown file, this is the titles of all headers (h1~h6)
|
2018-11-26 20:59:49 -06:00
|
|
|
// of the sections that contain the code block, e.g., if the markdown file is
|
2017-09-16 02:59:15 -05:00
|
|
|
// written as:
|
|
|
|
//
|
|
|
|
// ``````markdown
|
|
|
|
// # Title
|
|
|
|
//
|
|
|
|
// ## Subtitle
|
|
|
|
//
|
|
|
|
// ```rust
|
|
|
|
// assert!(true);
|
|
|
|
// ```
|
|
|
|
// ``````
|
|
|
|
//
|
|
|
|
// the `names` vector of that test will be `["Title", "Subtitle"]`.
|
2014-05-22 18:57:53 -05:00
|
|
|
names: Vec<String>,
|
2017-09-16 02:59:15 -05:00
|
|
|
|
2019-08-29 16:15:31 -05:00
|
|
|
options: Options,
|
2014-03-28 12:27:24 -05:00
|
|
|
use_headers: bool,
|
2019-08-27 17:44:11 -05:00
|
|
|
enable_per_target_ignores: bool,
|
2014-05-22 18:57:53 -05:00
|
|
|
cratename: String,
|
2015-04-06 20:39:39 -05:00
|
|
|
opts: TestOptions,
|
2017-02-06 15:11:03 -06:00
|
|
|
position: Span,
|
2018-08-18 05:14:14 -05:00
|
|
|
source_map: Option<Lrc<SourceMap>>,
|
2017-12-14 01:09:19 -06:00
|
|
|
filename: Option<PathBuf>,
|
2013-12-22 13:23:04 -06:00
|
|
|
}
|
|
|
|
|
|
|
|
impl Collector {
|
2019-12-22 16:42:04 -06:00
|
|
|
pub fn new(
|
|
|
|
cratename: String,
|
|
|
|
options: Options,
|
|
|
|
use_headers: bool,
|
|
|
|
opts: TestOptions,
|
|
|
|
source_map: Option<Lrc<SourceMap>>,
|
|
|
|
filename: Option<PathBuf>,
|
|
|
|
enable_per_target_ignores: bool,
|
|
|
|
) -> Collector {
|
2014-03-06 21:31:41 -06:00
|
|
|
Collector {
|
2014-03-05 17:28:08 -06:00
|
|
|
tests: Vec::new(),
|
|
|
|
names: Vec::new(),
|
2019-08-29 16:15:31 -05:00
|
|
|
options,
|
2017-08-07 00:54:09 -05:00
|
|
|
use_headers,
|
2019-08-27 17:44:11 -05:00
|
|
|
enable_per_target_ignores,
|
2017-08-07 00:54:09 -05:00
|
|
|
cratename,
|
|
|
|
opts,
|
2017-02-06 15:11:03 -06:00
|
|
|
position: DUMMY_SP,
|
2018-08-18 05:14:14 -05:00
|
|
|
source_map,
|
2017-08-07 00:54:09 -05:00
|
|
|
filename,
|
2016-10-01 19:11:45 -05:00
|
|
|
}
|
|
|
|
}
|
|
|
|
|
2017-12-14 01:09:19 -06:00
|
|
|
fn generate_name(&self, line: usize, filename: &FileName) -> String {
|
2017-09-16 02:59:15 -05:00
|
|
|
format!("{} - {} (line {})", filename, self.names.join("::"), line)
|
2017-04-13 18:23:14 -05:00
|
|
|
}
|
|
|
|
|
2018-09-17 17:25:50 -05:00
|
|
|
pub fn set_position(&mut self, position: Span) {
|
|
|
|
self.position = position;
|
|
|
|
}
|
|
|
|
|
|
|
|
fn get_filename(&self) -> FileName {
|
|
|
|
if let Some(ref source_map) = self.source_map {
|
|
|
|
let filename = source_map.span_to_filename(self.position);
|
|
|
|
if let FileName::Real(ref filename) = filename {
|
|
|
|
if let Ok(cur_dir) = env::current_dir() {
|
|
|
|
if let Ok(path) = filename.strip_prefix(&cur_dir) {
|
|
|
|
return path.to_owned().into();
|
|
|
|
}
|
|
|
|
}
|
|
|
|
}
|
|
|
|
filename
|
|
|
|
} else if let Some(ref filename) = self.filename {
|
|
|
|
filename.clone().into()
|
|
|
|
} else {
|
|
|
|
FileName::Custom("input".to_owned())
|
|
|
|
}
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
impl Tester for Collector {
|
|
|
|
fn add_test(&mut self, test: String, config: LangString, line: usize) {
|
2018-07-21 17:54:30 -05:00
|
|
|
let filename = self.get_filename();
|
2017-04-13 18:23:14 -05:00
|
|
|
let name = self.generate_name(line, &filename);
|
2014-05-25 05:10:11 -05:00
|
|
|
let cratename = self.cratename.to_string();
|
2015-04-06 20:39:39 -05:00
|
|
|
let opts = self.opts.clone();
|
2019-08-29 16:15:31 -05:00
|
|
|
let edition = config.edition.unwrap_or(self.options.edition.clone());
|
|
|
|
let options = self.options.clone();
|
2019-08-27 17:44:11 -05:00
|
|
|
let runtool = self.options.runtool.clone();
|
|
|
|
let runtool_args = self.options.runtool_args.clone();
|
|
|
|
let target = self.options.target.clone();
|
2019-06-11 13:06:34 -05:00
|
|
|
let target_str = target.to_string();
|
2018-12-08 13:17:50 -06:00
|
|
|
|
2019-07-23 13:06:00 -05:00
|
|
|
debug!("creating test {}: {}", name, test);
|
2014-02-13 19:49:11 -06:00
|
|
|
self.tests.push(testing::TestDescAndFn {
|
|
|
|
desc: testing::TestDesc {
|
2020-01-15 08:00:25 -06:00
|
|
|
name: testing::DynTestName(name),
|
2019-04-26 15:52:56 -05:00
|
|
|
ignore: match config.ignore {
|
|
|
|
Ignore::All => true,
|
|
|
|
Ignore::None => false,
|
2019-12-22 16:42:04 -06:00
|
|
|
Ignore::Some(ref ignores) => ignores.iter().any(|s| target_str.contains(s)),
|
2019-04-26 15:52:56 -05:00
|
|
|
},
|
2015-04-06 20:39:39 -05:00
|
|
|
// compiler failures are test failures
|
|
|
|
should_panic: testing::ShouldPanic::No,
|
2018-07-21 17:54:30 -05:00
|
|
|
allow_fail: config.allow_fail,
|
2019-09-28 09:07:18 -05:00
|
|
|
test_type: testing::TestType::DocTest,
|
2013-12-22 13:23:04 -06:00
|
|
|
},
|
2017-12-10 11:58:22 -06:00
|
|
|
testfn: testing::DynTestFn(box move || {
|
2019-05-04 14:37:12 -05:00
|
|
|
let res = run_test(
|
2018-12-08 13:30:23 -06:00
|
|
|
&test,
|
|
|
|
&cratename,
|
|
|
|
&filename,
|
|
|
|
line,
|
2019-08-29 16:15:31 -05:00
|
|
|
options,
|
2018-12-08 13:30:23 -06:00
|
|
|
config.should_panic,
|
|
|
|
config.no_run,
|
|
|
|
config.test_harness,
|
2019-04-26 15:52:56 -05:00
|
|
|
runtool,
|
|
|
|
runtool_args,
|
|
|
|
target,
|
2018-12-08 13:30:23 -06:00
|
|
|
config.compile_fail,
|
|
|
|
config.error_codes,
|
|
|
|
&opts,
|
|
|
|
edition,
|
2019-05-04 14:37:12 -05:00
|
|
|
);
|
|
|
|
|
|
|
|
if let Err(err) = res {
|
|
|
|
match err {
|
|
|
|
TestFailure::CompileError => {
|
|
|
|
eprint!("Couldn't compile the test.");
|
|
|
|
}
|
|
|
|
TestFailure::UnexpectedCompilePass => {
|
|
|
|
eprint!("Test compiled successfully, but it's marked `compile_fail`.");
|
|
|
|
}
|
|
|
|
TestFailure::UnexpectedRunPass => {
|
|
|
|
eprint!("Test executable succeeded, but it's marked `should_panic`.");
|
|
|
|
}
|
|
|
|
TestFailure::MissingErrorCodes(codes) => {
|
|
|
|
eprint!("Some expected error codes were not found: {:?}", codes);
|
|
|
|
}
|
|
|
|
TestFailure::ExecutionError(err) => {
|
|
|
|
eprint!("Couldn't run the test: {}", err);
|
|
|
|
if err.kind() == io::ErrorKind::PermissionDenied {
|
|
|
|
eprint!(" - maybe your tempdir is mounted with noexec?");
|
|
|
|
}
|
|
|
|
}
|
|
|
|
TestFailure::ExecutionFailure(out) => {
|
|
|
|
let reason = if let Some(code) = out.status.code() {
|
|
|
|
format!("exit code {}", code)
|
|
|
|
} else {
|
|
|
|
String::from("terminated by signal")
|
|
|
|
};
|
|
|
|
|
|
|
|
eprintln!("Test executable failed ({}).", reason);
|
|
|
|
|
|
|
|
// FIXME(#12309): An unfortunate side-effect of capturing the test
|
|
|
|
// executable's output is that the relative ordering between the test's
|
|
|
|
// stdout and stderr is lost. However, this is better than the
|
|
|
|
// alternative: if the test executable inherited the parent's I/O
|
|
|
|
// handles the output wouldn't be captured at all, even on success.
|
|
|
|
//
|
|
|
|
// The ordering could be preserved if the test process' stderr was
|
|
|
|
// redirected to stdout, but that functionality does not exist in the
|
|
|
|
// standard library, so it may not be portable enough.
|
|
|
|
let stdout = str::from_utf8(&out.stdout).unwrap_or_default();
|
|
|
|
let stderr = str::from_utf8(&out.stderr).unwrap_or_default();
|
|
|
|
|
|
|
|
if !stdout.is_empty() || !stderr.is_empty() {
|
|
|
|
eprintln!();
|
|
|
|
|
|
|
|
if !stdout.is_empty() {
|
|
|
|
eprintln!("stdout:\n{}", stdout);
|
|
|
|
}
|
|
|
|
|
|
|
|
if !stderr.is_empty() {
|
|
|
|
eprintln!("stderr:\n{}", stderr);
|
|
|
|
}
|
|
|
|
}
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
panic::resume_unwind(box ());
|
|
|
|
}
|
2017-01-17 17:15:08 -06:00
|
|
|
}),
|
2013-12-22 13:23:04 -06:00
|
|
|
});
|
|
|
|
}
|
2014-03-06 21:31:41 -06:00
|
|
|
|
2018-09-17 17:25:50 -05:00
|
|
|
fn get_line(&self) -> usize {
|
2018-08-18 05:14:14 -05:00
|
|
|
if let Some(ref source_map) = self.source_map {
|
2017-07-31 15:04:34 -05:00
|
|
|
let line = self.position.lo().to_usize();
|
2018-08-18 05:14:14 -05:00
|
|
|
let line = source_map.lookup_char_pos(BytePos(line as u32)).line;
|
2017-01-23 15:46:18 -06:00
|
|
|
if line > 0 { line - 1 } else { line }
|
2017-01-17 16:54:51 -06:00
|
|
|
} else {
|
2017-02-06 15:11:03 -06:00
|
|
|
0
|
2017-01-17 16:54:51 -06:00
|
|
|
}
|
|
|
|
}
|
|
|
|
|
2018-09-17 17:25:50 -05:00
|
|
|
fn register_header(&mut self, name: &str, level: u32) {
|
2017-09-16 02:59:15 -05:00
|
|
|
if self.use_headers {
|
2018-11-26 20:59:49 -06:00
|
|
|
// We use these headings as test names, so it's good if
|
2014-03-06 21:31:41 -06:00
|
|
|
// they're valid identifiers.
|
2019-12-22 16:42:04 -06:00
|
|
|
let name = name
|
|
|
|
.chars()
|
|
|
|
.enumerate()
|
|
|
|
.map(|(i, c)| {
|
|
|
|
if (i == 0 && rustc_lexer::is_id_start(c))
|
|
|
|
|| (i != 0 && rustc_lexer::is_id_continue(c))
|
|
|
|
{
|
2014-03-06 21:31:41 -06:00
|
|
|
c
|
|
|
|
} else {
|
|
|
|
'_'
|
|
|
|
}
|
2019-12-22 16:42:04 -06:00
|
|
|
})
|
|
|
|
.collect::<String>();
|
2014-03-06 21:31:41 -06:00
|
|
|
|
2017-09-16 02:59:15 -05:00
|
|
|
// Here we try to efficiently assemble the header titles into the
|
|
|
|
// test name in the form of `h1::h2::h3::h4::h5::h6`.
|
|
|
|
//
|
2018-11-26 20:59:49 -06:00
|
|
|
// Suppose that originally `self.names` contains `[h1, h2, h3]`...
|
2017-09-16 02:59:15 -05:00
|
|
|
let level = level as usize;
|
|
|
|
if level <= self.names.len() {
|
|
|
|
// ... Consider `level == 2`. All headers in the lower levels
|
|
|
|
// are irrelevant in this new level. So we should reset
|
|
|
|
// `self.names` to contain headers until <h2>, and replace that
|
|
|
|
// slot with the new name: `[h1, name]`.
|
|
|
|
self.names.truncate(level);
|
|
|
|
self.names[level - 1] = name;
|
|
|
|
} else {
|
|
|
|
// ... On the other hand, consider `level == 5`. This means we
|
|
|
|
// need to extend `self.names` to contain five headers. We fill
|
|
|
|
// in the missing level (<h4>) with `_`. Thus `self.names` will
|
|
|
|
// become `[h1, h2, h3, "_", name]`.
|
|
|
|
if level - 1 > self.names.len() {
|
|
|
|
self.names.resize(level - 1, "_".to_owned());
|
|
|
|
}
|
|
|
|
self.names.push(name);
|
|
|
|
}
|
2014-03-06 21:31:41 -06:00
|
|
|
}
|
|
|
|
}
|
2013-12-22 13:23:04 -06:00
|
|
|
}
|
|
|
|
|
2019-06-14 11:39:39 -05:00
|
|
|
struct HirCollector<'a, 'hir> {
|
2017-08-05 01:38:52 -05:00
|
|
|
sess: &'a session::Session,
|
2016-11-19 19:11:20 -06:00
|
|
|
collector: &'a mut Collector,
|
2020-01-04 19:37:57 -06:00
|
|
|
map: &'a Map<'hir>,
|
2018-07-22 12:10:19 -05:00
|
|
|
codes: ErrorCodes,
|
2016-11-19 19:11:20 -06:00
|
|
|
}
|
2015-11-04 17:41:33 -06:00
|
|
|
|
2016-11-19 19:11:20 -06:00
|
|
|
impl<'a, 'hir> HirCollector<'a, 'hir> {
|
2019-12-22 16:42:04 -06:00
|
|
|
fn visit_testable<F: FnOnce(&mut Self)>(
|
|
|
|
&mut self,
|
|
|
|
name: String,
|
|
|
|
attrs: &[ast::Attribute],
|
|
|
|
nested: F,
|
|
|
|
) {
|
2017-08-05 01:38:52 -05:00
|
|
|
let mut attrs = Attributes::from_ast(self.sess.diagnostic(), attrs);
|
|
|
|
if let Some(ref cfg) = attrs.cfg {
|
2018-02-14 09:11:02 -06:00
|
|
|
if !cfg.matches(&self.sess.parse_sess, Some(&self.sess.features_untracked())) {
|
2017-08-05 01:38:52 -05:00
|
|
|
return;
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
2016-11-19 19:11:20 -06:00
|
|
|
let has_name = !name.is_empty();
|
|
|
|
if has_name {
|
|
|
|
self.collector.names.push(name);
|
|
|
|
}
|
2015-11-04 17:41:33 -06:00
|
|
|
|
2016-11-19 19:11:20 -06:00
|
|
|
attrs.collapse_doc_comments();
|
|
|
|
attrs.unindent_doc_comments();
|
2018-11-26 20:59:49 -06:00
|
|
|
// The collapse-docs pass won't combine sugared/raw doc attributes, or included files with
|
|
|
|
// anything else, this will combine them for us.
|
2017-09-21 22:37:00 -05:00
|
|
|
if let Some(doc) = attrs.collapsed_doc_value() {
|
2018-07-21 18:15:25 -05:00
|
|
|
self.collector.set_position(attrs.span.unwrap_or(DUMMY_SP));
|
2019-12-22 16:42:04 -06:00
|
|
|
markdown::find_testable_code(
|
|
|
|
&doc,
|
|
|
|
self.collector,
|
|
|
|
self.codes,
|
|
|
|
self.collector.enable_per_target_ignores,
|
|
|
|
);
|
2013-12-22 13:23:04 -06:00
|
|
|
}
|
2015-11-04 17:41:33 -06:00
|
|
|
|
2016-11-19 19:11:20 -06:00
|
|
|
nested(self);
|
|
|
|
|
|
|
|
if has_name {
|
|
|
|
self.collector.names.pop();
|
2013-12-22 13:23:04 -06:00
|
|
|
}
|
2016-11-19 19:11:20 -06:00
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
impl<'a, 'hir> intravisit::Visitor<'hir> for HirCollector<'a, 'hir> {
|
2020-01-07 10:25:33 -06:00
|
|
|
type Map = Map<'hir>;
|
|
|
|
|
|
|
|
fn nested_visit_map(&mut self) -> intravisit::NestedVisitorMap<'_, Self::Map> {
|
2016-11-29 06:07:04 -06:00
|
|
|
intravisit::NestedVisitorMap::All(&self.map)
|
2016-11-25 15:10:23 -06:00
|
|
|
}
|
|
|
|
|
2016-11-19 19:11:20 -06:00
|
|
|
fn visit_item(&mut self, item: &'hir hir::Item) {
|
2020-01-17 18:14:29 -06:00
|
|
|
let name = if let hir::ItemKind::Impl { ref self_ty, .. } = item.kind {
|
|
|
|
self.map.hir_to_pretty_string(self_ty.hir_id)
|
2016-11-19 19:11:20 -06:00
|
|
|
} else {
|
2018-11-30 20:47:08 -06:00
|
|
|
item.ident.to_string()
|
2016-11-19 19:11:20 -06:00
|
|
|
};
|
2015-11-04 17:41:33 -06:00
|
|
|
|
2016-11-19 19:11:20 -06:00
|
|
|
self.visit_testable(name, &item.attrs, |this| {
|
|
|
|
intravisit::walk_item(this, item);
|
|
|
|
});
|
|
|
|
}
|
2015-11-04 17:41:33 -06:00
|
|
|
|
2016-11-19 19:11:20 -06:00
|
|
|
fn visit_trait_item(&mut self, item: &'hir hir::TraitItem) {
|
2018-06-10 14:24:24 -05:00
|
|
|
self.visit_testable(item.ident.to_string(), &item.attrs, |this| {
|
2016-11-19 19:11:20 -06:00
|
|
|
intravisit::walk_trait_item(this, item);
|
|
|
|
});
|
|
|
|
}
|
2015-11-04 17:41:33 -06:00
|
|
|
|
2016-11-19 19:11:20 -06:00
|
|
|
fn visit_impl_item(&mut self, item: &'hir hir::ImplItem) {
|
2018-06-10 14:24:24 -05:00
|
|
|
self.visit_testable(item.ident.to_string(), &item.attrs, |this| {
|
2016-11-19 19:11:20 -06:00
|
|
|
intravisit::walk_impl_item(this, item);
|
|
|
|
});
|
|
|
|
}
|
2015-11-04 17:41:33 -06:00
|
|
|
|
2016-11-19 19:11:20 -06:00
|
|
|
fn visit_foreign_item(&mut self, item: &'hir hir::ForeignItem) {
|
2018-11-30 20:47:08 -06:00
|
|
|
self.visit_testable(item.ident.to_string(), &item.attrs, |this| {
|
2016-11-19 19:11:20 -06:00
|
|
|
intravisit::walk_foreign_item(this, item);
|
|
|
|
});
|
|
|
|
}
|
2015-11-04 17:41:33 -06:00
|
|
|
|
2019-12-22 16:42:04 -06:00
|
|
|
fn visit_variant(
|
|
|
|
&mut self,
|
|
|
|
v: &'hir hir::Variant,
|
|
|
|
g: &'hir hir::Generics,
|
|
|
|
item_id: hir::HirId,
|
|
|
|
) {
|
2019-08-13 19:40:21 -05:00
|
|
|
self.visit_testable(v.ident.to_string(), &v.attrs, |this| {
|
2016-11-19 19:11:20 -06:00
|
|
|
intravisit::walk_variant(this, v, g, item_id);
|
|
|
|
});
|
|
|
|
}
|
|
|
|
|
|
|
|
fn visit_struct_field(&mut self, f: &'hir hir::StructField) {
|
2018-05-25 18:50:15 -05:00
|
|
|
self.visit_testable(f.ident.to_string(), &f.attrs, |this| {
|
2016-11-19 19:11:20 -06:00
|
|
|
intravisit::walk_struct_field(this, f);
|
|
|
|
});
|
2013-12-22 13:23:04 -06:00
|
|
|
}
|
2016-12-08 04:48:07 -06:00
|
|
|
|
|
|
|
fn visit_macro_def(&mut self, macro_def: &'hir hir::MacroDef) {
|
|
|
|
self.visit_testable(macro_def.name.to_string(), ¯o_def.attrs, |_| ());
|
|
|
|
}
|
2013-12-22 13:23:04 -06:00
|
|
|
}
|
2018-02-09 10:40:27 -06:00
|
|
|
|
|
|
|
#[cfg(test)]
|
2019-06-05 13:33:39 -05:00
|
|
|
mod tests;
|