mlua-socket 0.2.7

A Rust-native implementation of LuaSocket for mlua.
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

use super::tcp_client::TcpClient;
use mlua::{Error, Lua, MultiValue};

pub(super) fn handle(lua: &Lua, args: MultiValue) -> Result<TcpClient, Error> {
    let tcp = super::tcp4::handle(lua, args.clone())?;
    super::tcp_connect::handle(lua, &tcp, args).map_err(|err| Error::RuntimeError(err.to_string()))?;
    Ok(TcpClient { tcp })
}

#[cfg(test)]
mod tests {
    use mlua::Lua;
    use std::error::Error;
    use std::net::TcpListener;

    #[test]
    fn connect() -> Result<(), Box<dyn Error>> {
        // Setup a listener to connect to
        let socket = TcpListener::bind("127.0.0.1:0")?;

        // Test
        let lua = Lua::new();
        crate::preload(&lua)?;
        let script = r#"
                local socket = require('socket')
                local client = socket.connect('127.0.0.1', _port_)
                return client:getsockname()
            "#
        .replace("_port_", format!("{}", socket.local_addr()?.port()).as_str());
        let (ip_addr, port, family): (String, i32, String) = lua.load(script).eval()?;
        assert_eq!(ip_addr, "127.0.0.1");
        assert!(port > 0);
        assert_eq!(family, "inet");
        Ok(())
    }
}