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

use mlua::{FromLua, Lua, Result, UserData, Value};
use sqlx::PgPool;

#[derive(Clone)]
pub struct LuaPgPool(pub PgPool);

impl UserData for LuaPgPool {
    fn add_methods<M: mlua::UserDataMethods<Self>>(_methods: &mut M) {}
}

impl FromLua for LuaPgPool {
    fn from_lua(value: Value, _lua: &Lua) -> Result<Self> {
        let userdata = match value {
            Value::UserData(ud) => ud,
            _ => {
                return Err(mlua::Error::FromLuaConversionError {
                    from: value.type_name(),
                    to: "LuaPgPool".to_string(),
                    message: Some("Expected UserData".to_string()),
                });
            }
        };
        let pool = userdata.borrow::<Self>()?.clone();
        Ok(pool)
    }
}