blob: 0f4c6bdad7a020bfbbb9ff2cb584d97a314247b0 [file] [log] [blame]
// Copyright 2013-2014 The Rust Project Developers. See the COPYRIGHT
// file at the top-level directory of this distribution and at
// http://rust-lang.org/COPYRIGHT.
//
// Licensed under the Apache License, Version 2.0 <LICENSE-APACHE or
// http://www.apache.org/licenses/LICENSE-2.0> or the MIT license
// <LICENSE-MIT or http://opensource.org/licenses/MIT>, at your
// option. This file may not be copied, modified, or distributed
// except according to those terms.
extern crate tempdir;
use std::env;
use std::fs;
use std::path::Path;
use std::sync::mpsc::channel;
use std::thread;
use tempdir::TempDir;
macro_rules! t {
($e:expr) => (match $e { Ok(n) => n, Err(e) => panic!("error: {}", e) })
}
trait PathExt {
fn exists(&self) -> bool;
fn is_dir(&self) -> bool;
}
impl PathExt for Path {
fn exists(&self) -> bool { fs::metadata(self).is_ok() }
fn is_dir(&self) -> bool {
fs::metadata(self).map(|m| m.is_dir()).unwrap_or(false)
}
}
fn test_tempdir() {
let path = {
let p = t!(TempDir::new_in(&Path::new("."), "foobar"));
let p = p.path();
assert!(p.to_str().unwrap().contains("foobar"));
p.to_path_buf()
};
assert!(!path.exists());
}
fn test_rm_tempdir() {
let (tx, rx) = channel();
let f = move|| -> () {
let tmp = t!(TempDir::new("test_rm_tempdir"));
tx.send(tmp.path().to_path_buf()).unwrap();
panic!("panic to unwind past `tmp`");
};
let _ = thread::spawn(f).join();
let path = rx.recv().unwrap();
assert!(!path.exists());
let tmp = t!(TempDir::new("test_rm_tempdir"));
let path = tmp.path().to_path_buf();
let f = move|| -> () {
let _tmp = tmp;
panic!("panic to unwind past `tmp`");
};
let _ = thread::spawn(f).join();
assert!(!path.exists());
let path;
{
let f = move || {
t!(TempDir::new("test_rm_tempdir"))
};
// FIXME(#16640) `: TempDir` annotation shouldn't be necessary
let tmp: TempDir = thread::spawn(f).join().unwrap();
path = tmp.path().to_path_buf();
assert!(path.exists());
}
assert!(!path.exists());
let path;
{
let tmp = t!(TempDir::new("test_rm_tempdir"));
path = tmp.into_path();
}
assert!(path.exists());
t!(fs::remove_dir_all(&path));
assert!(!path.exists());
}
fn test_rm_tempdir_close() {
let (tx, rx) = channel();
let f = move|| -> () {
let tmp = t!(TempDir::new("test_rm_tempdir"));
tx.send(tmp.path().to_path_buf()).unwrap();
t!(tmp.close());
panic!("panic when unwinding past `tmp`");
};
let _ = thread::spawn(f).join();
let path = rx.recv().unwrap();
assert!(!path.exists());
let tmp = t!(TempDir::new("test_rm_tempdir"));
let path = tmp.path().to_path_buf();
let f = move|| -> () {
let tmp = tmp;
t!(tmp.close());
panic!("panic when unwinding past `tmp`");
};
let _ = thread::spawn(f).join();
assert!(!path.exists());
let path;
{
let f = move || {
t!(TempDir::new("test_rm_tempdir"))
};
// FIXME(#16640) `: TempDir` annotation shouldn't be necessary
let tmp: TempDir = thread::spawn(f).join().unwrap();
path = tmp.path().to_path_buf();
assert!(path.exists());
t!(tmp.close());
}
assert!(!path.exists());
let path;
{
let tmp = t!(TempDir::new("test_rm_tempdir"));
path = tmp.into_path();
}
assert!(path.exists());
t!(fs::remove_dir_all(&path));
assert!(!path.exists());
}
// Ideally these would be in std::os but then core would need
// to depend on std
fn recursive_mkdir_rel() {
let path = Path::new("frob");
let cwd = env::current_dir().unwrap();
println!("recursive_mkdir_rel: Making: {} in cwd {} [{}]", path.display(),
cwd.display(), path.exists());
t!(fs::create_dir(&path));
assert!(path.is_dir());
t!(fs::create_dir_all(&path));
assert!(path.is_dir());
}
fn recursive_mkdir_dot() {
let dot = Path::new(".");
t!(fs::create_dir_all(&dot));
let dotdot = Path::new("..");
t!(fs::create_dir_all(&dotdot));
}
fn recursive_mkdir_rel_2() {
let path = Path::new("./frob/baz");
let cwd = env::current_dir().unwrap();
println!("recursive_mkdir_rel_2: Making: {} in cwd {} [{}]", path.display(),
cwd.display(), path.exists());
t!(fs::create_dir_all(&path));
assert!(path.is_dir());
assert!(path.parent().unwrap().is_dir());
let path2 = Path::new("quux/blat");
println!("recursive_mkdir_rel_2: Making: {} in cwd {}", path2.display(),
cwd.display());
t!(fs::create_dir("quux"));
t!(fs::create_dir_all(&path2));
assert!(path2.is_dir());
assert!(path2.parent().unwrap().is_dir());
}
// Ideally this would be in core, but needs TempFile
pub fn test_remove_dir_all_ok() {
let tmpdir = t!(TempDir::new("test"));
let tmpdir = tmpdir.path();
let root = tmpdir.join("foo");
println!("making {}", root.display());
t!(fs::create_dir(&root));
t!(fs::create_dir(&root.join("foo")));
t!(fs::create_dir(&root.join("foo").join("bar")));
t!(fs::create_dir(&root.join("foo").join("bar").join("blat")));
t!(fs::remove_dir_all(&root));
assert!(!root.exists());
assert!(!root.join("bar").exists());
assert!(!root.join("bar").join("blat").exists());
}
pub fn dont_double_panic() {
let r: Result<(), _> = thread::spawn(move|| {
let tmpdir = TempDir::new("test").unwrap();
// Remove the temporary directory so that TempDir sees
// an error on drop
t!(fs::remove_dir(tmpdir.path()));
// Panic. If TempDir panics *again* due to the rmdir
// error then the process will abort.
panic!();
}).join();
assert!(r.is_err());
}
fn in_tmpdir<F>(f: F) where F: FnOnce() {
let tmpdir = t!(TempDir::new("test"));
assert!(env::set_current_dir(tmpdir.path()).is_ok());
f();
}
pub fn pass_as_asref_path() {
let tempdir = t!(TempDir::new("test"));
takes_asref_path(&tempdir);
fn takes_asref_path<T: AsRef<Path>>(path: T) {
let path = path.as_ref();
assert!(path.exists());
}
}
#[test]
fn main() {
in_tmpdir(test_tempdir);
in_tmpdir(test_rm_tempdir);
in_tmpdir(test_rm_tempdir_close);
in_tmpdir(recursive_mkdir_rel);
in_tmpdir(recursive_mkdir_dot);
in_tmpdir(recursive_mkdir_rel_2);
in_tmpdir(test_remove_dir_all_ok);
in_tmpdir(dont_double_panic);
in_tmpdir(pass_as_asref_path);
}