fisher 1.0.0

Webhooks catcher written in Rust
Documentation 
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

// Copyright (C) 2017 Pietro Albini
//
// This program is free software: you can redistribute it and/or modify
// it under the terms of the GNU General Public License as published by
// the Free Software Foundation, either version 3 of the License, or
// (at your option) any later version.
//
// This program is distributed in the hope that it will be useful,
// but WITHOUT ANY WARRANTY; without even the implied warranty of
// MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
// GNU General Public License for more details.
//
// You should have received a copy of the GNU General Public License
// along with this program.  If not, see <http://www.gnu.org/licenses/>.

use std::collections::HashMap;
use std::fs;
use std::io::Write;
use std::net::{IpAddr, Ipv4Addr};
use std::os::unix::fs::OpenOptionsExt;
use std::path::PathBuf;
use std::sync::Arc;

use tempdir::TempDir;

use common::prelude::*;
use common::state::State;
use scripts::Script;
use web::WebRequest;


pub struct TestEnv {
    state: Arc<State>,
    scripts_dir: PathBuf,
    temp_dirs: Vec<TempDir>,
}

impl TestEnv {
    fn new() -> Result<Self> {
        let scripts_dir = TempDir::new("fisher-tests")?;

        Ok(TestEnv {
            state: Arc::new(State::new()),
            scripts_dir: scripts_dir.path().to_path_buf(),
            temp_dirs: vec![scripts_dir],
        })
    }

    pub fn state(&self) -> Arc<State> {
        self.state.clone()
    }

    pub fn tempdir(&mut self) -> Result<PathBuf> {
        let dir = TempDir::new("fisher-tests")?;
        let owned = dir.path().to_path_buf();

        self.temp_dirs.push(dir);
        Ok(owned)
    }

    pub fn scripts_dir(&self) -> PathBuf {
        self.scripts_dir.clone()
    }

    pub fn create_script(&self, name: &str, content: &[&str]) -> Result<()> {
        self.create_script_into(&self.scripts_dir, name, content)
    }

    pub fn create_script_into(
        &self,
        path: &PathBuf,
        name: &str,
        content: &[&str],
    ) -> Result<()> {
        let path = path.join(name);

        let mut to_write = String::new();
        for line in content {
            to_write.push_str(line);
            to_write.push('\n');
        }

        fs::OpenOptions::new()
            .create(true)
            .write(true)
            .mode(0o755)
            .open(&path)?
            .write(to_write.as_bytes())?;

        Ok(())
    }


    pub fn load_script(&self, name: &str) -> Result<Script> {
        let path = self.scripts_dir().join(name).to_str().unwrap().to_string();
        Ok(Script::load(name.into(), path, &self.state)?)
    }
}


pub fn dummy_web_request() -> WebRequest {
    WebRequest {
        headers: HashMap::new(),
        params: HashMap::new(),
        source: IpAddr::V4(Ipv4Addr::new(127, 0, 0, 1)),
        body: String::new(),
    }
}


pub fn test_wrapper<F: Fn(&mut TestEnv) -> Result<()>>(func: F) {
    let mut env = TestEnv::new().unwrap();

    let result = func(&mut env);

    if let Err(error) = result {
        panic!("{}", error);
    }
}