graphANNIS/cli/tests/cli.rs at d6ae21abfa805070215bb2722c9fd80672367d0a · korpling/graphANNIS · GitHub

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
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
use assert_cmd::cargo;
use insta::Settings;
use insta::assert_snapshot;
use serial_test::serial;
use std::path::Path;

fn standard_filter() -> Settings {
    let mut settings = insta::Settings::clone_current();
    // Remove any color ASCII codes
    settings.add_filter("\x1b", "");
    settings.add_filter("\\[[0-9]+m", "");

    // Filter out the time stamps
    settings.add_filter("[0-9]+:[0-9]+:[0-9]+ ", "12:00:00");
    // The loaded and also total available RAM size can vary
    settings.add_filter("[0-9.]+[MG]B / [0-9.]+[MG]B", "100MB / 300MB");
    // The loading and time can vary
    settings.add_filter("in [0-9]+ ms", "in 10 ms");
    settings
}

fn mimic_insta_snapshot(
    output: std::process::Output,
) -> Result<String, Box<dyn std::error::Error>> {
    let success = output.status.success();
    let exit_code = output.status.code().unwrap();
    let stdout = str::from_utf8(&output.stdout)?;
    let stderr = str::from_utf8(&output.stderr)?;
    Ok(format!(
        r#"
success: {success}
exit_code: {exit_code}
----- stdout -----
{stdout}
----- stderr -----
{stderr}
    "#
    ))
}

#[test]
#[serial]
fn show_corpus_info() -> Result<(), Box<dyn std::error::Error>> {
    let mut cmd = cargo::cargo_bin_cmd!("annis");

    cmd.arg("../graphannis/tests/data/")
        .arg("-c")
        .arg("corpus sample-disk-based-3.8")
        .arg("-c")
        .arg("preload")
        .arg("-c")
        .arg("info");

    let output = cmd.output().unwrap();
    let actual = mimic_insta_snapshot(output)?;
    let settings = standard_filter();
    settings.bind(|| assert_snapshot!(actual));

    Ok(())
}

#[test]
#[serial]
fn list_corpora_not_loaded() -> Result<(), Box<dyn std::error::Error>> {
    let mut cmd = cargo::cargo_bin_cmd!("annis");

    cmd.arg("../graphannis/tests/data/").arg("-c").arg("list");

    let output = cmd.output().unwrap();
    let actual = mimic_insta_snapshot(output)?;
    let settings = standard_filter();
    settings.bind(|| assert_snapshot!(actual));

    Ok(())
}

#[test]
#[serial]
fn list_corpora_fully_loaded() -> Result<(), Box<dyn std::error::Error>> {
    let mut cmd = cargo::cargo_bin_cmd!("annis");

    cmd.arg("../graphannis/tests/data/")
        .arg("-c")
        .arg("corpus sample-disk-based-3.3")
        .arg("-c")
        .arg("preload")
        .arg("-c")
        .arg("list");

    let output = cmd.output().unwrap();
    let actual = mimic_insta_snapshot(output)?;
    let settings = standard_filter();
    settings.bind(|| assert_snapshot!(actual));

    Ok(())
}

#[test]
#[serial]
fn list_corpora_partially_loaded() -> Result<(), Box<dyn std::error::Error>> {
    let mut cmd = cargo::cargo_bin_cmd!("annis");

    cmd.arg("../graphannis/tests/data/")
        .arg("-c")
        .arg("corpus sample-disk-based-3.3")
        .arg("-c")
        .arg("count tok")
        .arg("-c")
        .arg("list");

    let output = cmd.output().unwrap();
    let actual = mimic_insta_snapshot(output)?;
    let settings = standard_filter();
    settings.bind(|| assert_snapshot!(actual));

    Ok(())
}

#[test]
#[serial]
fn export_to_zip_file() -> Result<(), Box<dyn std::error::Error>> {
    let mut cmd = cargo::cargo_bin_cmd!("annis");

    cmd.arg("../graphannis/tests/data/")
        .arg("-c")
        .arg("corpus sample-disk-based-3.3")
        .arg("-c")
        .arg("export sample-disk-based-3.3.zip");

    let output = cmd.output().unwrap();
    let actual = mimic_insta_snapshot(output)?;
    let settings = standard_filter();
    settings.bind(|| assert_snapshot!(actual));

    // Check that the file has been created
    let p = Path::new("sample-disk-based-3.3.zip");
    assert_eq!(true, p.is_file());
    // Cleanup created file
    std::fs::remove_file(p)?;
    Ok(())
}