dstest 0.1.2

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
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

use std::sync::Arc;

use mlua::{Lua, Result, Table};

use crate::bindings::context::BindingContext;
use crate::config::AccumulationMode;
use crate::fault::FaultTree;

pub fn register(lua: &Lua, dstest: &Table, ctx: &BindingContext) -> Result<()> {
    let state = Arc::clone(&ctx.state);
    let config = Arc::clone(&ctx.config);
    let fault_tree = Arc::clone(&ctx.fault_tree);
    let substrate = Arc::clone(&ctx.substrate);
    let oracle = Arc::clone(&ctx.oracle);

    let step_fn = lua.create_function(move |lua, ()| {
        let (require_seed, accumulation_mode, step_delay) = {
            let cfg = config.lock().expect("poisoned config lock");
            (cfg.require_seed, cfg.accumulation_mode, cfg.step_delay_ms)
        };

        if require_seed {
            let s = state.lock().expect("poisoned engine state lock");
            if s.seed.is_none() {
                return Err(mlua::Error::RuntimeError(
                    "dstest.config({ seed = n }) must be called before dstest.step()".into(),
                ));
            }
        }

        let mut tree_guard = fault_tree.lock().expect("poisoned fault tree lock");

        if tree_guard.is_none() {
            let s = state.lock().expect("poisoned engine state lock");
            if let Some(seed) = s.seed {
                let subject_ids: Vec<String> =
                    s.subjects.iter().map(|(id, _)| id.clone()).collect();
                if subject_ids.is_empty() {
                    return Err(mlua::Error::RuntimeError(
                        "no subjects available for fault injection".into(),
                    ));
                }
                let cfg = config.lock().expect("poisoned config lock");
                *tree_guard = Some(FaultTree::new(seed, subject_ids, &cfg));
            } else {
                return Err(mlua::Error::RuntimeError(
                    "dstest.config({ seed = n }) must be called before dstest.step()".into(),
                ));
            }
        }

        let tree = tree_guard
            .as_mut()
            .ok_or_else(|| mlua::Error::RuntimeError("fault tree not initialized".into()))?;
        let result = tree.step();

        let Some(step_result) = result else {
            let t = lua.create_table()?;
            t.set("more", false)?;
            return Ok(t);
        };

        let subject = crate::substrate::Subject::new(step_result.subject_id.clone());

        match accumulation_mode {
            AccumulationMode::Single => {
                substrate
                    .clear_faults(&subject)
                    .map_err(mlua::Error::RuntimeError)?;
                std::thread::sleep(std::time::Duration::from_millis(step_delay));
            }
            AccumulationMode::Accumulate => {}
        }

        substrate
            .affect(&subject, &step_result.fault)
            .map_err(mlua::Error::RuntimeError)?;

        let t = lua.create_table()?;
        t.set("fault", step_result.fault.to_string())?;
        t.set("subject", step_result.subject_id.clone())?;
        t.set("round", step_result.round)?;
        t.set("total_rounds", step_result.total_rounds)?;
        t.set("remaining", step_result.remaining)?;
        t.set("more", step_result.more)?;

        {
            let mut o = oracle.lock().expect("poisoned oracle lock");
            if o.enabled {
                let report = o.check_all(
                    lua,
                    &step_result.subject_id,
                    &step_result.fault.to_string(),
                    step_result.round,
                );
                o.report.merge(&report);
                t.set("oracle", {
                    let ot = lua.create_table()?;
                    ot.set("passed", report.passed)?;
                    ot.set("total_checks", report.total_checks)?;
                    ot.set("passed_checks", report.passed_checks)?;
                    ot.set("failed_checks", report.failed_checks)?;
                    ot
                })?;
            }
        }

        Ok(t)
    })?;

    dstest.set("step", step_fn)?;
    Ok(())
}