~gbmor/clinte

ref: 06af54fd6068780daa44a004dd62f194f41c07b0 clinte/src/logging.rs -rw-r--r-- 1.2 KiB
06af54fdBenjamin Morrison removed panic on error in favor of exit(1) and clean msg 1 year, 4 months ago
                                                                                
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
use std::fs::OpenOptions;

use simplelog::*;

use crate::error;
use crate::user;

pub fn checked_init() {
    let logfile = format!("/tmp/clinte_{}.log", *user::NAME);
    error::helper(init(&logfile), "Couldn't initialize logging");
}

fn init(path: &str) -> error::Result<()> {
    let logfile = OpenOptions::new().append(true).create(true).open(path)?;
    WriteLogger::init(LevelFilter::Info, Config::default(), logfile)?;

    Ok(())
}

#[cfg(test)]
mod tests {
    use super::*;

    use crate::user;
    use std::fs;

    #[test]
    fn init_logs() {
        let file = "clinte_test.log";
        let blank = " ".bytes().collect::<Vec<u8>>();
        fs::write(&file, &blank).unwrap();
        init("clinte_test.log").unwrap();

        log::info!("TEST LOG MESSAGE");
        let logfile = fs::read_to_string(&file).unwrap();
        assert!(logfile.contains("TEST LOG MESSAGE"));

        fs::remove_file("clinte_test.log").unwrap();
    }

    #[test]
    #[ignore]
    fn checked_init_logs() {
        checked_init();
        log::info!("CHECKED INIT TEST");

        let path = format!("/tmp/clinte_{}.log", *user::NAME);
        let logfile = fs::read_to_string(&path).unwrap();
        assert!(logfile.contains("CHECKED INIT TEST"));
    }
}