dstest 0.1.6

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

use std::sync::Arc;

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

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

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

    let logs_fn = lua.create_function(move |lua, (id, opts): (String, Option<Table>)| {
        let subject = Subject::new(id);

        let log_opts = if let Some(opts) = opts {
            LogOptions {
                stdout: opts.get("stdout").unwrap_or(true),
                stderr: opts.get("stderr").unwrap_or(true),
                tail: opts.get("tail").ok(),
                since: opts.get("since").ok(),
                timestamps: opts.get("timestamps").unwrap_or(false),
            }
        } else {
            LogOptions::default()
        };

        let entries = substrate
            .logs(&subject, log_opts)
            .map_err(mlua::Error::RuntimeError)?;

        let result = lua.create_table()?;
        for (i, entry) in entries.into_iter().enumerate() {
            let t = lua.create_table()?;
            t.set(
                "stream",
                match entry.stream {
                    Stream::StdOut => "stdout",
                    Stream::StdErr => "stderr",
                },
            )?;
            t.set("message", entry.message)?;
            result.set(i + 1, t)?;
        }

        Ok(result)
    })?;

    dstest.set("logs", logs_fn)?;
    Ok(())
}