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

use std::env;
use std::fs;
use std::path;
use std::path::Path;

pub const TMP_DIR: &str = "/tmp/wake-tmp-folder";

fn create_tmp_folder() {
    //Create a temporary folder
    //Create the temporary directory if it doesn't exist
    let path = path::Path::new(TMP_DIR);
    if !path.exists() {
        fs::create_dir(TMP_DIR).unwrap();
    }

    //Set test mode
    env::set_var("WAKE_TEST_MODE", "1");
}

fn delete_tmp_folder() {
    let _ = fs::remove_dir_all(TMP_DIR); //silence the error
}

pub fn setup() {
    delete_tmp_folder();
    create_tmp_folder();
    move_to_tmp_folder();
}

pub fn teardown() {
    delete_tmp_folder();
    env::remove_var("WAKE_TEST_MODE");
}

fn move_to_tmp_folder() {
    env::set_current_dir(Path::new(TMP_DIR)).unwrap();
}