xserial/crates/xserial-client/tests/lua_tests.rs

use std::sync::Once;

use mlua::Lua;
use tokio::io::{AsyncReadExt, AsyncWriteExt};
use tokio::net::TcpListener;

static INIT: Once = Once::new();

fn init_tracing() {
    INIT.call_once(|| {
        tracing_subscriber::fmt()
            .with_env_filter("xserial=trace")
            .try_init()
            .ok();
    });
}

// ── xserial.sleep / xserial.log ──────────────────────────────────

#[tokio::test]
async fn lua_sleep_and_log() {
    let lua = Lua::new();
    xserial_client::lua::register(&lua).unwrap();

    lua.load(
        r#"
        xserial.log("test start")
        xserial.sleep(50)
        xserial.log("test end")
        "#
    )
    .exec_async()
    .await
    .unwrap();
}

// ── xserial.list_ports ───────────────────────────────────────────

#[tokio::test]
async fn lua_list_ports() {
    let lua = Lua::new();
    xserial_client::lua::register(&lua).unwrap();

    let result: Vec<String> = lua
        .load(r#"return xserial.list_ports()"#)
        .eval_async()
        .await
        .unwrap();

    // Should return a table (possibly empty if no serial ports)
    // Just verify it doesn't crash
    let _ = result;
}

// ── xserial.open + session:send / session:read / session:close ────

#[tokio::test]
async fn lua_session_open_send_read_close() {
    init_tracing();
    let listener = TcpListener::bind("127.0.0.1:0").await.unwrap();
    let addr = listener.local_addr().unwrap().to_string();

    let server = tokio::spawn(async move {
        let (mut stream, _) = listener.accept().await.unwrap();
        let mut buf = [0u8; 256];
        let n = stream.read(&mut buf).await.unwrap();
        assert_eq!(&buf[..n], b"hello\n");
        stream.write_all(b"world\n").await.unwrap();
    });

    let lua = Lua::new();
    xserial_client::lua::register(&lua).unwrap();

    let script = format!(
        r#"
        local sess = xserial.open({{
            transport = {{ Tcp = {{ addr = "{}" }} }},
            pipelines = {{{{
                name = "text",
                framer = {{ Line = {{}} }},
                decoder = {{ Text = {{}} }}
            }}}}
        }})

        sess:send("hello\n")

        local r = sess:read(5000)
        assert(r ~= nil, "expected data")
        assert(r.pipeline == "text", "expected text pipeline")
        assert(r.data == "world", "expected 'world', got " .. tostring(r.data))

        sess:close()
        "#,
        addr
    );

    lua.load(&script).exec_async().await.unwrap();
    server.await.unwrap();
}

// ── session:read with timeout ────────────────────────────────────

#[tokio::test]
async fn lua_session_read_timeout() {
    let listener = TcpListener::bind("127.0.0.1:0").await.unwrap();
    let addr = listener.local_addr().unwrap().to_string();

    let _server = tokio::spawn(async move {
        let (_stream, _) = listener.accept().await.unwrap();
        tokio::time::sleep(std::time::Duration::from_secs(10)).await;
    });

    let lua = Lua::new();
    xserial_client::lua::register(&lua).unwrap();

    let script = format!(
        r#"
        local sess = xserial.open({{
            transport = {{ Tcp = {{ addr = "{}" }} }},
            pipelines = {{{{
                name = "text",
                framer = {{ Line = {{}} }},
                decoder = {{ Text = {{}} }}
            }}}}
        }})

        local r = sess:read(100)
        assert(r == nil, "read should time out")

        sess:close()
        "#,
        addr
    );

    lua.load(&script).exec_async().await.unwrap();
}

// ── xserial.open with hex decoder ────────────────────────────────

#[tokio::test]
async fn lua_session_hex_decoder() {
    let listener = TcpListener::bind("127.0.0.1:0").await.unwrap();
    let addr = listener.local_addr().unwrap().to_string();

    let server = tokio::spawn(async move {
        let (mut stream, _) = listener.accept().await.unwrap();
        stream.write_all(b"ABC\n").await.unwrap();
    });

    let lua = Lua::new();
    xserial_client::lua::register(&lua).unwrap();

    let script = format!(
        r#"
        local sess = xserial.open({{
            transport = {{ Tcp = {{ addr = "{}" }} }},
            pipelines = {{{{
                name = "hex",
                framer = {{ Line = {{}} }},
                decoder = {{ Hex = {{ uppercase = true }} }}
            }}}}
        }})

        local r = sess:read(5000)
        assert(r ~= nil)
        assert(r.pipeline == "hex")
        assert(r.kind == "hex")
        assert(r.data == "41 42 43", "expected hex of ABC, got " .. tostring(r.data))

        sess:close()
        "#,
        addr
    );

    lua.load(&script).exec_async().await.unwrap();
    server.await.unwrap();
}

// ── session:on_data callback ─────────────────────────────────────

#[tokio::test]
async fn lua_session_on_data_callback() {
    let listener = TcpListener::bind("127.0.0.1:0").await.unwrap();
    let addr = listener.local_addr().unwrap().to_string();

    let server = tokio::spawn(async move {
        let (mut stream, _) = listener.accept().await.unwrap();
        stream.write_all(b"line1\nline2\n").await.unwrap();
    });

    let lua = Lua::new();
    xserial_client::lua::register(&lua).unwrap();

    let script = format!(
        r#"
        local received = {{}}
        local sess = xserial.open({{
            transport = {{ Tcp = {{ addr = "{}" }} }},
            pipelines = {{{{
                name = "text",
                framer = {{ Line = {{}} }},
                decoder = {{ Text = {{}} }}
            }}}}
        }})

        sess:on_data(function(name, data)
            table.insert(received, {{ name = name, data = data }})
        end)

        -- Wait for both lines
        xserial.sleep(500)

        assert(#received == 2, "expected 2 callbacks, got " .. #received)
        assert(received[1].name == "text")
        assert(received[1].data == "line1")
        assert(received[2].name == "text")
        assert(received[2].data == "line2")

        sess:close()
        "#,
        addr
    );

    lua.load(&script).exec_async().await.unwrap();
    server.await.unwrap();
}

// ── session:reconfigure from Lua ─────────────────────────────────

#[tokio::test]
async fn lua_session_reconfigure() {
    let listener = TcpListener::bind("127.0.0.1:0").await.unwrap();
    let addr = listener.local_addr().unwrap().to_string();

    let server = tokio::spawn(async move {
        let (mut stream, _) = listener.accept().await.unwrap();
        stream.write_all(b"test\n").await.unwrap();
    });

    let listener2 = TcpListener::bind("127.0.0.1:0").await.unwrap();
    let addr2 = listener2.local_addr().unwrap().to_string();

    let server2 = tokio::spawn(async move {
        let (mut stream, _) = listener2.accept().await.unwrap();
        stream.write_all(b"ABC\n").await.unwrap();
    });

    let lua = Lua::new();
    xserial_client::lua::register(&lua).unwrap();

    let script = format!(
        r#"
        local sess = xserial.open({{
            transport = {{ Tcp = {{ addr = "{}" }} }},
            pipelines = {{{{
                name = "text",
                framer = {{ Line = {{}} }},
                decoder = {{ Text = {{}} }}
            }}}}
        }})

        local r = sess:read(5000)
        assert(r.data == "test")

        -- Reconfigure to hex on different address
        sess:reconfigure({{
            transport = {{ Tcp = {{ addr = "{}" }} }},
            pipelines = {{{{
                name = "hex",
                framer = {{ Line = {{}} }},
                decoder = {{ Hex = {{ uppercase = true }} }}
            }}}}
        }})

        local r2 = sess:read(5000)
        assert(r2.pipeline == "hex")
        assert(r2.data == "41 42 43")

        sess:close()
        "#,
        addr, addr2
    );

    lua.load(&script).exec_async().await.unwrap();
    server.await.unwrap();
    server2.await.unwrap();
}