nblm-core 0.2.3

Core library for NotebookLM Enterprise API client
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

#[cfg(test)]
pub mod fake {
    use serde_json::json;
    use wiremock::matchers::{method, path};
    use wiremock::{Mock, MockServer, ResponseTemplate};

    /// Helper struct that spins up a fake OAuth server for tests.
    pub struct FakeOAuthServer {
        mock_server: MockServer,
    }

    impl FakeOAuthServer {
        /// Start the fake server with default token and revocation endpoints.
        pub async fn start() -> Self {
            let mock_server = MockServer::start().await;

            Mock::given(method("POST"))
                .and(path("/token"))
                .respond_with(ResponseTemplate::new(200).set_body_json(json!({
                    "access_token": "fake_access_token",
                    "refresh_token": "fake_refresh_token",
                    "token_type": "Bearer",
                    "expires_in": 3600
                })))
                .mount(&mock_server)
                .await;

            Mock::given(method("POST"))
                .and(path("/revoke"))
                .respond_with(ResponseTemplate::new(200))
                .mount(&mock_server)
                .await;

            Self { mock_server }
        }

        pub fn token_endpoint(&self) -> String {
            format!("{}/token", self.mock_server.uri())
        }

        pub fn revoke_endpoint(&self) -> String {
            format!("{}/revoke", self.mock_server.uri())
        }

        pub fn base_uri(&self) -> String {
            self.mock_server.uri()
        }
    }
}