dstest 0.1.5

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

use std::sync::Arc;

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

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

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

    let inspect_fn = lua.create_function(move |lua, id: String| {
        let subject = Subject::new(id);

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

        let t = lua.create_table()?;
        t.set(
            "state",
            match info.state {
                ContainerState::Running => "running",
                ContainerState::Paused => "paused",
                ContainerState::Exited => "exited",
                ContainerState::Dead => "dead",
            },
        )?;
        t.set("pid", info.pid)?;
        t.set("ip", info.ip)?;
        t.set("memory_limit", info.memory_limit)?;
        t.set("cpu_quota", info.cpu_quota)?;

        Ok(t)
    })?;

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