dstest 0.2.1

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

use std::sync::Arc;

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

use crate::engine::context::BindingContext;
use crate::substrate::{Subject, Substrate};

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

    let exec_fn = lua.create_async_function(move |lua, (id, cmd): (String, Vec<String>)| {
        let substrate = Arc::clone(&substrate);

        async move {
            let subject = Subject::new(id);

            let result = substrate
                .exec(&subject, &cmd)
                .await
                .map_err(mlua::Error::RuntimeError)?;

            let t = lua.create_table()?;
            t.set("exit_code", result.exit_code)?;
            t.set("stdout", result.stdout)?;
            t.set("stderr", result.stderr)?;

            Ok(t)
        }
    })?;

    dstest.set("exec", exec_fn)?;
    Ok(())
}