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
34
35
36
37
38
39
40
41
42
43
44
45
46
47

use std::sync::Arc;

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

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

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

    let inspect_fn = lua.create_async_function(move |lua, id: String| {
        let substrate = Arc::clone(&substrate);
        let state = Arc::clone(&state);

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

            let info = substrate
                .inspect(&subject)
                .await
                .map_err(mlua::Error::RuntimeError)?;

            let value = info.to_lua(&lua)?;

            // Surface the host-reachable address (host:port) when the substrate
            // mapped container ports to the host. On setups where the bridge IP
            // is not directly reachable from the host (e.g. a podman/docker
            // machine VM), this is the only address the host process can dial.
            if let Value::Table(ref t) = value
                && let Some(host) = state
                    .lock()
                    .expect("poisoned engine state lock")
                    .subject_hosts
                    .get(&id)
                    .cloned()
            {
                t.set("host", host)?;
            }

            Ok(value)
        }
    })?;

    dstest.set("inspect", inspect_fn)?;
    Ok(())
}