tests/compile-test.rs

   1 #![feature(test)]
   2
   3 use compiletest_rs as compiletest;
   4 extern crate test;
   5
   6 use std::env::{set_var, var};
   7 use std::ffi::OsStr;
   8 use std::fs;
   9 use std::io;
  10 use std::path::{Path, PathBuf};
  11
  12 fn clippy_driver_path() -> PathBuf {
  13     if let Some(path) = option_env!("CLIPPY_DRIVER_PATH") {
  14         PathBuf::from(path)
  15     } else {
  16         PathBuf::from(concat!("target/", env!("PROFILE"), "/clippy-driver"))
  17     }
  18 }
  19
  20 fn host_libs() -> PathBuf {
  21     if let Some(path) = option_env!("HOST_LIBS") {
  22         PathBuf::from(path)
  23     } else {
  24         Path::new("target").join(env!("PROFILE"))
  25     }
  26 }
  27
  28 fn rustc_test_suite() -> Option<PathBuf> {
  29     option_env!("RUSTC_TEST_SUITE").map(PathBuf::from)
  30 }
  31
  32 fn rustc_lib_path() -> PathBuf {
  33     option_env!("RUSTC_LIB_PATH").unwrap().into()
  34 }
  35
  36 fn config(mode: &str, dir: PathBuf) -> compiletest::Config {
  37     let mut config = compiletest::Config::default();
  38
  39     let cfg_mode = mode.parse().expect("Invalid mode");
  40     if let Ok(name) = var::<&str>("TESTNAME") {
  41         let s: String = name.to_owned();
  42         config.filter = Some(s)
  43     }
  44
  45     if rustc_test_suite().is_some() {
  46         config.run_lib_path = rustc_lib_path();
  47         config.compile_lib_path = rustc_lib_path();
  48     }
  49     config.target_rustcflags = Some(format!(
  50         "-L {0} -L {0}/deps -Dwarnings -Zui-testing",
  51         host_libs().display()
  52     ));
  53
  54     config.mode = cfg_mode;
  55     config.build_base = if rustc_test_suite().is_some() {
  56         // we don't need access to the stderr files on travis
  57         let mut path = PathBuf::from(env!("OUT_DIR"));
  58         path.push("test_build_base");
  59         path
  60     } else {
  61         let mut path = std::env::current_dir().unwrap();
  62         path.push("target/debug/test_build_base");
  63         path
  64     };
  65     config.src_base = dir;
  66     config.rustc_path = clippy_driver_path();
  67     config
  68 }
  69
  70 fn run_mode(mode: &str, dir: PathBuf) {
  71     let cfg = config(mode, dir);
  72     // clean rmeta data, otherwise "cargo check; cargo test" fails (#2896)
  73     cfg.clean_rmeta();
  74     compiletest::run_tests(&cfg);
  75 }
  76
  77 fn run_ui_toml_tests(config: &compiletest::Config, mut tests: Vec<test::TestDescAndFn>) -> Result<bool, io::Error> {
  78     let mut result = true;
  79     let opts = compiletest::test_opts(config);
  80     for dir in fs::read_dir(&config.src_base)? {
  81         let dir = dir?;
  82         if !dir.file_type()?.is_dir() {
  83             continue;
  84         }
  85         let dir_path = dir.path();
  86         set_var("CARGO_MANIFEST_DIR", &dir_path);
  87         for file in fs::read_dir(&dir_path)? {
  88             let file = file?;
  89             let file_path = file.path();
  90             if !file.file_type()?.is_file() {
  91                 continue;
  92             }
  93             if file_path.extension() != Some(OsStr::new("rs")) {
  94                 continue;
  95             }
  96             let paths = compiletest::common::TestPaths {
  97                 file: file_path,
  98                 base: config.src_base.clone(),
  99                 relative_dir: dir_path.file_name().unwrap().into(),
 100             };
 101             let test_name = compiletest::make_test_name(&config, &paths);
 102             let index = tests
 103                 .iter()
 104                 .position(|test| test.desc.name == test_name)
 105                 .expect("The test should be in there");
 106             result &= test::run_tests_console(&opts, vec![tests.swap_remove(index)])?;
 107         }
 108     }
 109     Ok(result)
 110 }
 111
 112 fn run_ui_toml() {
 113     let path = PathBuf::from("tests/ui-toml").canonicalize().unwrap();
 114     let config = config("ui", path);
 115     let tests = compiletest::make_tests(&config);
 116
 117     let res = run_ui_toml_tests(&config, tests);
 118     match res {
 119         Ok(true) => {},
 120         Ok(false) => panic!("Some tests failed"),
 121         Err(e) => {
 122             println!("I/O failure during tests: {:?}", e);
 123         },
 124     }
 125 }
 126
 127 fn prepare_env() {
 128     set_var("CLIPPY_DISABLE_DOCS_LINKS", "true");
 129     set_var("CLIPPY_TESTS", "true");
 130     //set_var("RUST_BACKTRACE", "0");
 131 }
 132
 133 #[test]
 134 fn compile_test() {
 135     prepare_env();
 136     run_mode("run-pass", "tests/run-pass".into());
 137     run_mode("ui", "tests/ui".into());
 138     run_ui_toml();
 139 }