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

mod common;

use common::{eval_lua, run_lua, run_lua_local};

#[tokio::test]
async fn test_require_postgres() {
    let script = r#"
        local pg = require("assay.postgres")
        assert.not_nil(pg, "postgres module should load")
        assert.not_nil(pg.client, "postgres.client should exist")
        assert.not_nil(pg.client_from_vault, "postgres.client_from_vault should exist")
        assert.not_nil(pg._quote_ident, "postgres._quote_ident should exist")
        assert.not_nil(pg._quote_literal, "postgres._quote_literal should exist")
    "#;
    run_lua(script).await.unwrap();
}

#[tokio::test]
async fn test_postgres_quote_ident_simple() {
    let result: String = eval_lua(
        r#"
        local pg = require("assay.postgres")
        return pg._quote_ident("users")
    "#,
    )
    .await;
    assert_eq!(result, "\"users\"");
}

#[tokio::test]
async fn test_postgres_quote_ident_with_quotes() {
    let result: String = eval_lua(
        r#"
        local pg = require("assay.postgres")
        return pg._quote_ident('my"table')
    "#,
    )
    .await;
    assert_eq!(result, "\"my\"\"table\"");
}

#[tokio::test]
async fn test_postgres_quote_literal_simple() {
    let result: String = eval_lua(
        r#"
        local pg = require("assay.postgres")
        return pg._quote_literal("hello")
    "#,
    )
    .await;
    assert_eq!(result, "'hello'");
}

#[tokio::test]
async fn test_postgres_quote_literal_with_quotes() {
    let result: String = eval_lua(
        r#"
        local pg = require("assay.postgres")
        return pg._quote_literal("it's")
    "#,
    )
    .await;
    assert_eq!(result, "'it''s'");
}

#[tokio::test]
async fn test_postgres_client_from_vault_missing_secret() {
    use wiremock::matchers::{method, path};
    use wiremock::{Mock, MockServer, ResponseTemplate};

    let mock_server = MockServer::start().await;

    Mock::given(method("GET"))
        .and(path("/v1/secrets/data/db/postgres"))
        .respond_with(ResponseTemplate::new(404))
        .mount(&mock_server)
        .await;

    let script = format!(
        r#"
        local vault = require("assay.vault")
        local pg = require("assay.postgres")
        
        local vault_client = vault.client("{}", "test-token")
        
        local ok, err = pcall(function()
            pg.client_from_vault(vault_client, "db/postgres", "localhost", 5432)
        end)
        
        assert.eq(ok, false, "client_from_vault should fail when secret is missing")
        assert.not_nil(err, "error message should be present")
    "#,
        mock_server.uri()
    );

    run_lua_local(&script).await.unwrap();
}