grr-cli 2.0.0

Google tools from the terminal, at maximum performance: 401 methods across 14 Google APIs in one generated command tree, zero-config, over HTTP/3
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
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
179
180
181
182

//! Offline regression tests for refresh-token failure handling.
//!
//! Regression: a rejected refresh token (Google 400 `invalid_grant`) used to
//! silently fall through into the implicit interactive OAuth flow, which binds
//! port 3434 mid-API-call and hangs callers for the full callback timeout.
//! The fix makes `get_access_token` return a fast, actionable `Auth` error
//! carrying Google's underlying reason; the implicit flow is reserved for the
//! fresh-install / explicit-login case (no stored token at all).

use std::time::Duration;

use grr_cli::core::{GoogleAuth, GrrConfig, GrrError, TokenStorage};
use wiremock::matchers::{method, path};
use wiremock::{Mock, MockServer, ResponseTemplate};

fn expired_storage_with_refresh_token() -> TokenStorage {
    TokenStorage {
        access_token: "stale-access-token".into(),
        refresh_token: Some("stale-refresh-token".into()),
        expires_at: 1, // long expired
        token_type: "Bearer".into(),
        scope: "test-scope".into(),
    }
}

#[tokio::test]
async fn rejected_refresh_token_returns_auth_error_without_oauth_flow() {
    let server = MockServer::start().await;
    let guard = Mock::given(method("POST"))
        .and(path("/token"))
        .respond_with(ResponseTemplate::new(400).set_body_json(serde_json::json!({
            "error": "invalid_grant",
            "error_description": "Token has been expired or revoked."
        })))
        .expect(1)
        .mount_as_scoped(&server)
        .await;

    let config = GrrConfig::default();
    let auth = GoogleAuth::with_token(config.oauth.clone(), expired_storage_with_refresh_token())
        .await
        .unwrap()
        .with_token_endpoint(format!("{}/token", server.uri()));

    let result = tokio::time::timeout(Duration::from_secs(5), auth.get_access_token())
        .await
        .expect("get_access_token must fail fast, not wait on an OAuth callback timeout");

    let err = result.expect_err("a rejected refresh token must yield an error");
    assert!(
        matches!(err, GrrError::Auth(_)),
        "unexpected error: {err:?}"
    );
    let message = err.to_string();
    assert!(
        message.contains("invalid_grant"),
        "underlying reason not surfaced: {message}"
    );
    assert!(
        message.contains("grr auth login"),
        "remediation guidance missing: {message}"
    );

    // Panics unless the token endpoint saw exactly one request, i.e. no
    // second attempt and no callback-server detour happened.
    drop(guard);
}

#[tokio::test]
async fn refresh_persists_to_overridden_token_path_only() {
    let server = MockServer::start().await;
    Mock::given(method("POST"))
        .and(path("/token"))
        .respond_with(ResponseTemplate::new(200).set_body_json(serde_json::json!({
            "access_token": "fresh-access-token",
            "expires_in": 3600,
            "token_type": "Bearer"
        })))
        .mount(&server)
        .await;

    let dir = tempfile::tempdir().unwrap();
    let token_path = dir.path().join("nested").join("token.json");

    let config = GrrConfig::default();
    let auth = GoogleAuth::with_token(config.oauth.clone(), expired_storage_with_refresh_token())
        .await
        .unwrap()
        .with_token_endpoint(format!("{}/token", server.uri()))
        .with_token_path(token_path.clone());

    let token = tokio::time::timeout(Duration::from_secs(5), auth.get_access_token())
        .await
        .expect("refresh must complete promptly")
        .unwrap();
    assert_eq!(token, "fresh-access-token");

    // The refreshed credential must land at the overridden path — never at
    // the user's real cache-dir token.json.
    let persisted = tokio::fs::read_to_string(&token_path).await.unwrap();
    assert!(
        persisted.contains("fresh-access-token"),
        "refreshed token not persisted to overridden path: {persisted}"
    );
}

#[tokio::test]
async fn refresh_response_without_refresh_token_preserves_stored_refresh_token() {
    let server = MockServer::start().await;
    // Google's refresh-grant response may omit refresh_token (it is only
    // rotated sometimes). The stored refresh token must survive such a
    // response so subsequent expiries can still refresh.
    Mock::given(method("POST"))
        .and(path("/token"))
        .respond_with(ResponseTemplate::new(200).set_body_json(serde_json::json!({
            "access_token": "rotating-access-token",
            "expires_in": 0, // immediately expired again
            "token_type": "Bearer"
        })))
        .mount(&server)
        .await;

    let config = GrrConfig::default();
    let dir = tempfile::tempdir().unwrap();
    let auth = GoogleAuth::with_token(config.oauth.clone(), expired_storage_with_refresh_token())
        .await
        .unwrap()
        .with_token_endpoint(format!("{}/token", server.uri()))
        .with_token_path(dir.path().join("token.json"));

    let first = tokio::time::timeout(Duration::from_secs(5), auth.get_access_token())
        .await
        .expect("first refresh must complete promptly")
        .unwrap();

    // expires_in=0 makes the refreshed token immediately stale: this second
    // call must refresh AGAIN using the preserved refresh token. With the
    // bug, the omitted refresh_token overwrote storage with None and this
    // call fails fast with an Auth error instead.
    let second = tokio::time::timeout(Duration::from_secs(5), auth.get_access_token())
        .await
        .expect("second refresh must complete promptly, not hit a missing refresh token")
        .unwrap();

    assert_eq!(first, "rotating-access-token");
    assert_eq!(second, "rotating-access-token");

    // Exactly two refresh round-trips: the second one is only possible when
    // the stored refresh token survived the first (refresh_token-less)
    // response.
    assert_eq!(server.received_requests().await.unwrap().len(), 2);
}

#[tokio::test]
async fn successful_refresh_still_updates_the_access_token() {
    let server = MockServer::start().await;
    Mock::given(method("POST"))
        .and(path("/token"))
        .respond_with(ResponseTemplate::new(200).set_body_json(serde_json::json!({
            "access_token": "fresh-access-token",
            "expires_in": 3600,
            "token_type": "Bearer"
        })))
        .mount(&server)
        .await;

    let config = GrrConfig::default();
    let dir = tempfile::tempdir().unwrap();
    let auth = GoogleAuth::with_token(config.oauth.clone(), expired_storage_with_refresh_token())
        .await
        .unwrap()
        .with_token_endpoint(format!("{}/token", server.uri()))
        // Never persist the fake refreshed credential over the real
        // user token at the platform cache dir.
        .with_token_path(dir.path().join("token.json"));

    let token = tokio::time::timeout(Duration::from_secs(5), auth.get_access_token())
        .await
        .expect("refresh must complete promptly")
        .unwrap();
    assert_eq!(token, "fresh-access-token");
}