diff options
Diffstat (limited to 'third_party/rust/tempfile/tests/tempdir.rs')
-rw-r--r-- | third_party/rust/tempfile/tests/tempdir.rs | 176 |
1 files changed, 176 insertions, 0 deletions
diff --git a/third_party/rust/tempfile/tests/tempdir.rs b/third_party/rust/tempfile/tests/tempdir.rs new file mode 100644 index 0000000000..8ca7984aca --- /dev/null +++ b/third_party/rust/tempfile/tests/tempdir.rs @@ -0,0 +1,176 @@ +// 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. + +#![deny(rust_2018_idioms)] + +use std::env; +use std::fs; +use std::path::Path; +use std::sync::mpsc::channel; +use std::thread; + +use tempfile::{Builder, TempDir}; + +fn test_tempdir() { + let path = { + let p = Builder::new().prefix("foobar").tempdir_in(".").unwrap(); + let p = p.path(); + assert!(p.to_str().unwrap().contains("foobar")); + p.to_path_buf() + }; + assert!(!path.exists()); +} + +fn test_prefix() { + let tmpfile = TempDir::with_prefix_in("prefix", ".").unwrap(); + let name = tmpfile.path().file_name().unwrap().to_str().unwrap(); + assert!(name.starts_with("prefix")); +} + +fn test_customnamed() { + let tmpfile = Builder::new() + .prefix("prefix") + .suffix("suffix") + .rand_bytes(12) + .tempdir() + .unwrap(); + let name = tmpfile.path().file_name().unwrap().to_str().unwrap(); + assert!(name.starts_with("prefix")); + assert!(name.ends_with("suffix")); + assert_eq!(name.len(), 24); +} + +fn test_rm_tempdir() { + let (tx, rx) = channel(); + let f = move || { + let tmp = TempDir::new().unwrap(); + 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 = TempDir::new().unwrap(); + 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 || TempDir::new().unwrap(); + + let tmp = thread::spawn(f).join().unwrap(); + path = tmp.path().to_path_buf(); + assert!(path.exists()); + } + assert!(!path.exists()); + + let path; + { + let tmp = TempDir::new().unwrap(); + path = tmp.into_path(); + } + assert!(path.exists()); + fs::remove_dir_all(&path).unwrap(); + assert!(!path.exists()); +} + +fn test_rm_tempdir_close() { + let (tx, rx) = channel(); + let f = move || { + let tmp = TempDir::new().unwrap(); + tx.send(tmp.path().to_path_buf()).unwrap(); + tmp.close().unwrap(); + panic!("panic when unwinding past `tmp`"); + }; + let _ = thread::spawn(f).join(); + let path = rx.recv().unwrap(); + assert!(!path.exists()); + + let tmp = TempDir::new().unwrap(); + let path = tmp.path().to_path_buf(); + let f = move || { + let tmp = tmp; + tmp.close().unwrap(); + panic!("panic when unwinding past `tmp`"); + }; + let _ = thread::spawn(f).join(); + assert!(!path.exists()); + + let path; + { + let f = move || TempDir::new().unwrap(); + + let tmp = thread::spawn(f).join().unwrap(); + path = tmp.path().to_path_buf(); + assert!(path.exists()); + tmp.close().unwrap(); + } + assert!(!path.exists()); + + let path; + { + let tmp = TempDir::new().unwrap(); + path = tmp.into_path(); + } + assert!(path.exists()); + fs::remove_dir_all(&path).unwrap(); + assert!(!path.exists()); +} + +fn dont_double_panic() { + let r: Result<(), _> = thread::spawn(move || { + let tmpdir = TempDir::new().unwrap(); + // Remove the temporary directory so that TempDir sees + // an error on drop + fs::remove_dir(tmpdir.path()).unwrap(); + // 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 = TempDir::new().unwrap(); + assert!(env::set_current_dir(tmpdir.path()).is_ok()); + + f(); +} + +fn pass_as_asref_path() { + let tempdir = TempDir::new().unwrap(); + 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_prefix); + in_tmpdir(test_customnamed); + in_tmpdir(test_rm_tempdir); + in_tmpdir(test_rm_tempdir_close); + in_tmpdir(dont_double_panic); + in_tmpdir(pass_as_asref_path); +} |