dstest 0.2.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

//! Lua bindings: the adapter layer that renders the application API to a
//! Lua script surface.
//!
//! These are deliberately thin. Each binding calls an application use case or
//! reaches a port, then renders the structured result to a Lua table. They
//! never mutate raw application state or implement domain logic.

pub mod clock;
pub mod core;
pub mod dst;
pub mod log;
pub mod net;
pub mod pg;
pub mod random;
pub mod storage;
pub mod subs;
pub mod workload;

use mlua::{Lua, Table};

use crate::application::BindingContext;

pub trait LuaModule {
    fn register(lua: &Lua, dstest: &Table, ctx: &BindingContext) -> mlua::Result<()>;
}

/// Set up the script surface and register every Lua module on the `dstest`
/// global. Called from the composition root; this is where the Lua globals
/// (`print`, `dstest`) are born, so the engine itself never wires its own
/// scripting surface.
pub fn register_all(lua: &Lua, ctx: &BindingContext) -> mlua::Result<()> {
    let globals = lua.globals();

    // Route Lua `print` through tracing so script output lands in the logs.
    globals.set(
        "print",
        lua.create_function(|_, msg: String| {
            tracing::info!("lua: {}", msg);
            Ok(())
        })?,
    )?;

    let dstest = lua.create_table()?;
    globals.set("dstest", dstest.clone())?;

    net::Net::register(lua, &dstest, ctx)?;
    dst::Dst::register(lua, &dstest, ctx)?;
    subs::Subs::register(lua, &dstest, ctx)?;
    log::Log::register(lua, &dstest, ctx)?;
    clock::Clock::register(lua, &dstest, ctx)?;
    core::Core::register(lua, &dstest, ctx)?;
    pg::Pg::register(lua, &dstest, ctx)?;
    storage::Storage::register(lua, &dstest, ctx)?;
    random::Random::register(lua, &dstest, ctx)?;
    workload::Workload::register(lua, &dstest, ctx)?;

    Ok(())
}