assay-lua 0.10.4

General-purpose enhanced Lua runtime. Batteries-included scripting, automation, and web services.
Documentation 
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
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108

mod common;

use common::run_lua_local;
use futures_util::{SinkExt, StreamExt};

async fn start_echo_ws_server() -> u16 {
    let listener = tokio::net::TcpListener::bind("127.0.0.1:0").await.unwrap();
    let port = listener.local_addr().unwrap().port();
    tokio::spawn(async move {
        while let Ok((stream, _)) = listener.accept().await {
            tokio::spawn(async move {
                let ws_stream = tokio_tungstenite::accept_async(stream).await.unwrap();
                let (mut write, mut read) = ws_stream.split();
                while let Some(Ok(msg)) = read.next().await {
                    if (msg.is_text() || msg.is_binary()) && write.send(msg).await.is_err() {
                        break;
                    }
                }
            });
        }
    });
    port
}

#[tokio::test]
async fn test_ws_connect_send_recv() {
    let port = start_echo_ws_server().await;
    run_lua_local(&format!(
        r#"
        local conn = ws.connect("ws://127.0.0.1:{port}/")
        ws.send(conn, "hello")
        local msg = ws.recv(conn)
        assert.eq(msg, "hello")
        ws.close(conn)
    "#
    ))
    .await
    .unwrap();
}

#[tokio::test]
async fn test_ws_multiple_messages() {
    let port = start_echo_ws_server().await;
    run_lua_local(&format!(
        r#"
        local conn = ws.connect("ws://127.0.0.1:{port}/")
        ws.send(conn, "first")
        assert.eq(ws.recv(conn), "first")
        ws.send(conn, "second")
        assert.eq(ws.recv(conn), "second")
        ws.send(conn, "third")
        assert.eq(ws.recv(conn), "third")
        ws.close(conn)
    "#
    ))
    .await
    .unwrap();
}

#[tokio::test]
async fn test_ws_close() {
    let port = start_echo_ws_server().await;
    run_lua_local(&format!(
        r#"
        local conn = ws.connect("ws://127.0.0.1:{port}/")
        ws.send(conn, "ping")
        assert.eq(ws.recv(conn), "ping")
        ws.close(conn)
    "#
    ))
    .await
    .unwrap();
}

#[tokio::test]
async fn test_ws_connect_failure() {
    let result = run_lua_local(
        r#"
        ws.connect("ws://127.0.0.1:1/nonexistent")
    "#,
    )
    .await;
    assert!(result.is_err());
    let err = result.unwrap_err().to_string();
    assert!(
        err.contains("ws.connect"),
        "error should mention ws.connect: {err}"
    );
}

#[tokio::test]
async fn test_ws_send_json() {
    let port = start_echo_ws_server().await;
    run_lua_local(&format!(
        r#"
        local conn = ws.connect("ws://127.0.0.1:{port}/")
        local payload = json.encode({{type = "ping", data = 42}})
        ws.send(conn, payload)
        local msg = ws.recv(conn)
        local parsed = json.parse(msg)
        assert.eq(parsed.type, "ping")
        assert.eq(parsed.data, 42)
        ws.close(conn)
    "#
    ))
    .await
    .unwrap();
}