forked from steveklabnik/rustdoc
-
Notifications
You must be signed in to change notification settings - Fork 0
/
build.rs
78 lines (60 loc) · 1.97 KB
/
build.rs
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
//! This build script creates source tests from the files in the `tests/source` directory.
#![recursion_limit = "1024"]
#[macro_use]
extern crate quote;
use std::env;
use std::fs::File;
use std::fs;
use std::io::prelude::*;
use std::io;
use std::path::Path;
use std::process;
use quote::Ident;
const SOURCE_TEST_DIR: &str = "tests/source";
fn run() -> io::Result<()> {
let out_dir = env::var("OUT_DIR").unwrap();
let mut generated_code = File::create(Path::new(&out_dir).join("source_generated.rs"))?;
let source_dir = Path::new(SOURCE_TEST_DIR);
for source_file in fs::read_dir(&source_dir)? {
let source_file = source_file?.path();
let is_rust_file = source_file
.extension()
.map(|ext| ext == "rs")
.unwrap_or_default();
if !is_rust_file {
continue;
}
let test_name = Ident::new(
source_file
.file_stem()
.and_then(|stem| stem.to_str())
.expect("invalid file stem"),
);
let source_file_path = source_file.to_str().unwrap();
let test = quote! {
#[test]
fn #test_name() {
use std::env;
use tempdir::TempDir;
let tempdir = TempDir::new("rustdoc-test").unwrap();
let source_file = env::current_dir().unwrap().join(#source_file_path);
let host = ::generate_analysis(&source_file, tempdir.path()).unwrap();
if let Err(err) = ::check(&source_file, &host) {
for cause in err.causes() {
println!("cause: {}", cause);
}
println!("{}", err.backtrace());
panic!();
}
}
};
write!(generated_code, "{}", test.as_str())?;
}
Ok(())
}
fn main() {
if let Err(ref e) = run() {
eprintln!("Error: {}", e);
process::exit(1);
}
}