dstest 0.1.4

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
114
115
116
117
118
119
120

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 run_steps_fn = lua.create_function(move |lua, n: usize| {
        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.run_steps()".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.run_steps()".into(),
                ));
            }
        }

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

        let mut results = Vec::new();
        for _ in 0..n {
            let Some(step_result) = tree.step() else {
                break;
            };

            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
                    })?;
                }
            }

            results.push(t);
        }

        let result_table = lua.create_table()?;
        for (i, t) in results.into_iter().enumerate() {
            result_table.set(i + 1, t)?;
        }

        Ok(result_table)
    })?;

    dstest.set("run_steps", run_steps_fn)?;
    Ok(())
}