aboutsummaryrefslogtreecommitdiff
path: root/tests/zip.rs
blob: d02e96e8d370809f0b28dd894519cf55fef52294 (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
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
use archivator::{Archive, Zip};
use std::io::{Read, Seek, SeekFrom};

#[test]
fn test_zip_passwd() {
    let mut archive = Archive::<Zip>::read_from_file("tests/files/archive_passwd.zip").unwrap();

    assert_eq!(archive.comment(), "archive comment");
    assert_eq!(
        archive
            .files()
            .iter()
            .map(|f| &f.name)
            .collect::<Vec<&String>>(),
        vec!["store", "deflate", "bzip"]
    );

    assert!(archive.get_file_reader_by_name("store").is_err());
    assert!(archive
        .get_file_reader_by_name_with_password("store", b"wrong_passwd")
        .is_err());

    for (name, check_data) in [
        ("store", "1e643774f40510e37c6f3c451d9d"),
        ("deflate", "a70aff4b6b2754ad47852503236a"),
        ("bzip", "f7085f4f8ecc512a8c2c3cbe8227"),
    ] {
        let mut f = archive
            .get_file_reader_by_name_with_password(name, b"passwd")
            .unwrap();
        let mut data = String::new();
        f.read_to_string(&mut data).unwrap();
        assert_eq!(&data[..24], "test encrypted file data");
        assert_eq!(&data[172..], check_data);
    }
}

#[test]
fn test_zip() {
    let mut archive = Archive::<Zip>::read_from_file("tests/files/archive.zip").unwrap();

    assert_eq!(archive.comment(), "archive comment");
    assert_eq!(
        archive
            .files()
            .iter()
            .map(|f| &f.name)
            .collect::<Vec<&String>>(),
        vec!["store", "deflate", "bzip", "lzma", "zstd", "xz"]
    );

    let mut f = archive.get_file_reader_by_name("store").unwrap();

    let mut data = String::new();
    f.read_to_string(&mut data).unwrap();
    assert_eq!(data, "test file data");

    assert_eq!(f.seek(SeekFrom::Start(5)).unwrap(), 5);
    let mut data = String::new();
    f.read_to_string(&mut data).unwrap();
    assert_eq!(data, "file data");

    assert_eq!(f.seek(SeekFrom::Start(0)).unwrap(), 0);
    let mut data = vec![0; 4];
    f.read(&mut data).unwrap();
    assert_eq!(String::from_utf8(data).unwrap(), "test");

    assert_eq!(f.seek(SeekFrom::Current(1)).unwrap(), 5);
    let mut data = vec![0; 4];
    f.read(&mut data).unwrap();
    assert_eq!(String::from_utf8(data).unwrap(), "file");

    assert_eq!(f.seek(SeekFrom::End(-4)).unwrap(), 10);
    let mut data = vec![0; 4];
    f.read(&mut data).unwrap();
    assert_eq!(String::from_utf8(data).unwrap(), "data");

    f.seek(SeekFrom::Start(7)).unwrap();
    assert_eq!(f.seek(SeekFrom::Current(0)).unwrap(), 7);
    assert!(f.seek(SeekFrom::End(-100)).is_err());
    assert_eq!(f.seek(SeekFrom::Current(0)).unwrap(), 7);

    assert_eq!(f.seek(SeekFrom::Start(100)).unwrap(), 14);

    for index in 0..archive.len() {
        let mut f = archive.get_file_reader_by_index(index).unwrap();
        let mut data = String::new();
        f.read_to_string(&mut data).unwrap();
        assert_eq!(data, "test file data");
        assert!(!f.is_seekable() || f.info().name == "store")
    }
}