assay-lua 0.10.3

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
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163

mod common;

use common::run_lua;
use wiremock::matchers::{body_json, method, path};
use wiremock::{Mock, MockServer, ResponseTemplate};

#[tokio::test]
async fn test_require_oauth2() {
    let script = r#"
        local mod = require("assay.oauth2")
        assert.not_nil(mod)
        assert.not_nil(mod.from_file)
    "#;
    run_lua(script).await.unwrap();
}

#[tokio::test]
async fn test_oauth2_from_file() {
    let script = r#"
        local oauth2 = require("assay.oauth2")
        local tmpdir = fs.tempdir()
        fs.write(tmpdir .. "/credentials.json", json.encode({
            installed = {
                client_id = "test-client",
                client_secret = "test-secret",
            }
        }))
        fs.write(tmpdir .. "/token.json", json.encode({
            access_token = "token-123",
            refresh_token = "refresh-123",
        }))

        local client = oauth2.from_file(tmpdir .. "/credentials.json", tmpdir .. "/token.json")
        assert.eq(client._credentials.client_id, "test-client")
        assert.eq(client._token_data.refresh_token, "refresh-123")
        assert.eq(client._token_file, tmpdir .. "/token.json")
    "#;
    run_lua(script).await.unwrap();
}

#[tokio::test]
async fn test_oauth2_access_token() {
    let script = r#"
        local oauth2 = require("assay.oauth2")
        local tmpdir = fs.tempdir()
        fs.write(tmpdir .. "/credentials.json", json.encode({
            client_id = "test-client",
            client_secret = "test-secret",
        }))
        fs.write(tmpdir .. "/token.json", json.encode({
            access_token = "token-abc",
            refresh_token = "refresh-abc",
        }))

        local client = oauth2.from_file(tmpdir .. "/credentials.json", tmpdir .. "/token.json")
        assert.eq(client:access_token(), "token-abc")
    "#;
    run_lua(script).await.unwrap();
}

#[tokio::test]
async fn test_oauth2_refresh() {
    let server = MockServer::start().await;
    Mock::given(method("POST"))
        .and(path("/token"))
        .and(body_json(serde_json::json!({
            "grant_type": "refresh_token",
            "refresh_token": "refresh-xyz",
            "client_id": "test-client",
            "client_secret": "test-secret"
        })))
        .respond_with(ResponseTemplate::new(200).set_body_json(serde_json::json!({
            "access_token": "fresh-token",
            "refresh_token": "refresh-next"
        })))
        .mount(&server)
        .await;

    let script = format!(
        r#"
        local oauth2 = require("assay.oauth2")
        local tmpdir = fs.tempdir()
        fs.write(tmpdir .. "/credentials.json", json.encode({{
            client_id = "test-client",
            client_secret = "test-secret",
        }}))
        fs.write(tmpdir .. "/token.json", json.encode({{
            access_token = "stale-token",
            refresh_token = "refresh-xyz",
        }}))

        local client = oauth2.from_file(tmpdir .. "/credentials.json", tmpdir .. "/token.json", {{
            token_url = "{}/token",
        }})
        local refreshed = client:refresh()
        assert.eq(refreshed, "fresh-token")
        assert.eq(client:access_token(), "fresh-token")
        assert.eq(client._token_data.refresh_token, "refresh-next")
        "#,
        server.uri()
    );
    run_lua(&script).await.unwrap();
}

#[tokio::test]
async fn test_oauth2_refresh_401() {
    let server = MockServer::start().await;
    Mock::given(method("POST"))
        .and(path("/token"))
        .respond_with(ResponseTemplate::new(401).set_body_json(serde_json::json!({
            "error": "invalid_grant"
        })))
        .mount(&server)
        .await;

    let script = format!(
        r#"
        local oauth2 = require("assay.oauth2")
        local tmpdir = fs.tempdir()
        fs.write(tmpdir .. "/credentials.json", json.encode({{
            client_id = "test-client",
            client_secret = "test-secret",
        }}))
        fs.write(tmpdir .. "/token.json", json.encode({{
            access_token = "stale-token",
            refresh_token = "refresh-xyz",
        }}))

        local client = oauth2.from_file(tmpdir .. "/credentials.json", tmpdir .. "/token.json", {{
            token_url = "{}/token",
        }})
        local ok, err = pcall(function()
            client:refresh()
        end)
        assert.eq(ok, false)
        assert.contains(err, "oauth2: token refresh failed HTTP 401")
        "#,
        server.uri()
    );
    run_lua(&script).await.unwrap();
}

#[tokio::test]
async fn test_oauth2_headers() {
    let script = r#"
        local oauth2 = require("assay.oauth2")
        local tmpdir = fs.tempdir()
        fs.write(tmpdir .. "/credentials.json", json.encode({
            client_id = "test-client",
            client_secret = "test-secret",
        }))
        fs.write(tmpdir .. "/token.json", json.encode({
            access_token = "header-token",
            refresh_token = "refresh-abc",
        }))

        local client = oauth2.from_file(tmpdir .. "/credentials.json", tmpdir .. "/token.json")
        local headers = client:headers()
        assert.eq(headers["Authorization"], "Bearer header-token")
        assert.eq(headers["Content-Type"], "application/json")
    "#;
    run_lua(script).await.unwrap();
}