dstest 0.1.3

Deterministic Simulation Testing for containerised services
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

use std::sync::{Arc, Mutex};

use mlua::Lua;

use crate::config::Config;
use crate::fault::FaultTree;
use crate::oracle::OracleRef;
use crate::substrate::docker::DockerSubjectData;
use crate::substrate::{Substrate, docker::Docker};

#[derive(Default)]
pub struct EngineState {
    pub subjects: Vec<(String, DockerSubjectData)>,
    pub subject_hosts: std::collections::HashMap<String, String>,
    pub seed: Option<u64>,
}

pub struct BindingContext {
    pub state: Arc<Mutex<EngineState>>,
    pub config: Arc<Mutex<Config>>,
    pub fault_tree: Arc<Mutex<Option<FaultTree>>>,
    pub oracle: OracleRef,
    pub substrate: Arc<dyn Substrate>,
    pub docker: Arc<Docker>,
    pub lua: Lua,
}

impl BindingContext {
    pub fn new() -> Self {
        let docker = Arc::new(Docker::new().expect("failed to connect to Docker daemon"));
        let substrate = Arc::clone(&docker) as Arc<dyn Substrate>;
        Self {
            state: Arc::new(Mutex::new(EngineState::default())),
            config: Arc::new(Mutex::new(Config::default())),
            fault_tree: Arc::new(Mutex::new(None)),
            oracle: Arc::new(Mutex::new(crate::oracle::Oracle::new())),
            substrate,
            docker,
            lua: Lua::new(),
        }
    }

    pub fn http_client_with_timeout(timeout_secs: u64) -> reqwest::blocking::Client {
        reqwest::blocking::Client::builder()
            .timeout(std::time::Duration::from_secs(timeout_secs))
            .build()
            .expect("failed to create HTTP client")
    }
}

impl Default for BindingContext {
    fn default() -> Self {
        Self::new()
    }
}