72 lines
1.4 KiB
Rust
72 lines
1.4 KiB
Rust
|
use std::path::*;
|
||
|
use std::fs;
|
||
|
use std::env;
|
||
|
use rand::random;
|
||
|
|
||
|
pub struct RandomTempPath {
|
||
|
path: PathBuf
|
||
|
}
|
||
|
|
||
|
pub fn random_filename() -> String {
|
||
|
(0..8).map(|_| ((random::<f32>() * 26.0) as u8 + 97) as char).collect()
|
||
|
}
|
||
|
|
||
|
impl RandomTempPath {
|
||
|
pub fn new() -> RandomTempPath {
|
||
|
let mut dir = env::temp_dir();
|
||
|
dir.push(random_filename());
|
||
|
RandomTempPath {
|
||
|
path: dir.clone()
|
||
|
}
|
||
|
}
|
||
|
|
||
|
pub fn create_dir() -> RandomTempPath {
|
||
|
let mut dir = env::temp_dir();
|
||
|
dir.push(random_filename());
|
||
|
fs::create_dir_all(dir.as_path()).unwrap();
|
||
|
RandomTempPath {
|
||
|
path: dir.clone()
|
||
|
}
|
||
|
}
|
||
|
|
||
|
pub fn as_path(&self) -> &PathBuf {
|
||
|
&self.path
|
||
|
}
|
||
|
|
||
|
pub fn as_str(&self) -> &str {
|
||
|
self.path.to_str().unwrap()
|
||
|
}
|
||
|
}
|
||
|
|
||
|
impl Drop for RandomTempPath {
|
||
|
fn drop(&mut self) {
|
||
|
if let Err(e) = fs::remove_dir_all(self.as_path()) {
|
||
|
panic!("failed to remove temp directory, probably something failed to destroyed ({})", e);
|
||
|
}
|
||
|
}
|
||
|
}
|
||
|
|
||
|
#[test]
|
||
|
fn creates_dir() {
|
||
|
let temp = RandomTempPath::create_dir();
|
||
|
assert!(fs::metadata(temp.as_path()).unwrap().is_dir());
|
||
|
}
|
||
|
|
||
|
#[test]
|
||
|
fn destroys_dir() {
|
||
|
let path_buf = {
|
||
|
let temp = RandomTempPath::create_dir();
|
||
|
assert!(fs::metadata(temp.as_path()).unwrap().is_dir());
|
||
|
let path_buf = temp.as_path().to_path_buf();
|
||
|
path_buf
|
||
|
};
|
||
|
|
||
|
assert!(fs::metadata(&path_buf).is_err());
|
||
|
}
|
||
|
|
||
|
#[test]
|
||
|
fn provides_random() {
|
||
|
let temp = RandomTempPath::create_dir();
|
||
|
assert!(temp.as_path().to_str().is_some());
|
||
|
}
|