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

use std::path;

use predicates;

use fs;

/// Assert the state of files within `TempDir`.
///
/// # Examples
///
/// ```rust,ignore
/// use assert_fs::*;
///
/// let temp = assert_fs::TempDir::new().unwrap();
/// let input_file = temp.child("foo.txt");
/// input_file.touch().unwrap();
/// // ... do something with input_file ...
/// temp.child("bar.txt").assert(predicates::path::missing());
/// temp.close().unwrap();
/// ```
pub trait TempDirAssertExt {
    /// Wrap with an interface for that provides assertions on the `TempDir`.
    fn assert(&self, pred: &predicates::Predicate<path::Path>) -> &Self;
}

impl TempDirAssertExt for fs::TempDir {
    fn assert(&self, pred: &predicates::Predicate<path::Path>) -> &Self {
        if !pred.eval(self.path()) {
            panic!("Predicate failed for {:?}", self.path());
        }
        self
    }
}

impl TempDirAssertExt for fs::ChildPath {
    fn assert(&self, pred: &predicates::Predicate<path::Path>) -> &Self {
        if !pred.eval(self.path()) {
            panic!("Predicate failed for {:?}", self.path());
        }
        self
    }
}