summaryrefslogtreecommitdiffstats
path: root/third_party/rust/nix/test/test_dir.rs
blob: 2af4aa5c0aa6f1d642a66f70fbfbb8f484ee8bfd (plain)
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
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
use nix::dir::{Dir, Type};
use nix::fcntl::OFlag;
use nix::sys::stat::Mode;
use std::fs::File;
use tempfile::tempdir;

#[cfg(test)]
fn flags() -> OFlag {
    #[cfg(target_os = "illumos")]
    let f = OFlag::O_RDONLY | OFlag::O_CLOEXEC;

    #[cfg(not(target_os = "illumos"))]
    let f = OFlag::O_RDONLY | OFlag::O_CLOEXEC | OFlag::O_DIRECTORY;

    f
}

#[test]
#[allow(clippy::unnecessary_sort_by)] // False positive
fn read() {
    let tmp = tempdir().unwrap();
    File::create(tmp.path().join("foo")).unwrap();
    std::os::unix::fs::symlink("foo", tmp.path().join("bar")).unwrap();
    let mut dir = Dir::open(tmp.path(), flags(), Mode::empty()).unwrap();
    let mut entries: Vec<_> = dir.iter().map(|e| e.unwrap()).collect();
    entries.sort_by(|a, b| a.file_name().cmp(b.file_name()));
    let entry_names: Vec<_> = entries
        .iter()
        .map(|e| e.file_name().to_str().unwrap().to_owned())
        .collect();
    assert_eq!(&entry_names[..], &[".", "..", "bar", "foo"]);

    // Check file types. The system is allowed to return DT_UNKNOWN (aka None here) but if it does
    // return a type, ensure it's correct.
    assert!(&[Some(Type::Directory), None].contains(&entries[0].file_type())); // .: dir
    assert!(&[Some(Type::Directory), None].contains(&entries[1].file_type())); // ..: dir
    assert!(&[Some(Type::Symlink), None].contains(&entries[2].file_type())); // bar: symlink
    assert!(&[Some(Type::File), None].contains(&entries[3].file_type())); // foo: regular file
}

#[test]
fn rewind() {
    let tmp = tempdir().unwrap();
    let mut dir = Dir::open(tmp.path(), flags(), Mode::empty()).unwrap();
    let entries1: Vec<_> = dir
        .iter()
        .map(|e| e.unwrap().file_name().to_owned())
        .collect();
    let entries2: Vec<_> = dir
        .iter()
        .map(|e| e.unwrap().file_name().to_owned())
        .collect();
    let entries3: Vec<_> = dir
        .into_iter()
        .map(|e| e.unwrap().file_name().to_owned())
        .collect();
    assert_eq!(entries1, entries2);
    assert_eq!(entries2, entries3);
}

#[cfg(not(target_os = "haiku"))]
#[test]
fn ebadf() {
    assert_eq!(Dir::from_fd(-1).unwrap_err(), nix::Error::EBADF);
}