tripo-api 0.1.4

Unofficial async Rust client for the Tripo 3D Generation API
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

use tripo_api::{Client, RetryPolicy};
use wiremock::matchers::{method, path};
use wiremock::{Mock, MockServer, ResponseTemplate};

#[tokio::test(flavor = "current_thread")]
#[allow(clippy::float_cmp)] // exact literals round-trip losslessly through JSON
async fn retries_on_500_then_succeeds() {
    let server = MockServer::start().await;

    // First response: 500. Subsequent: 200.
    Mock::given(method("GET"))
        .and(path("/user/balance"))
        .respond_with(ResponseTemplate::new(500).set_body_string("boom"))
        .up_to_n_times(1)
        .mount(&server)
        .await;
    Mock::given(method("GET"))
        .and(path("/user/balance"))
        .respond_with(ResponseTemplate::new(200).set_body_json(serde_json::json!({
            "code":0,"data":{"balance":1.0,"frozen":0.0}
        })))
        .mount(&server)
        .await;

    let c = Client::builder()
        .api_key("tsk_test")
        .base_url(server.uri().parse().unwrap())
        .retry(RetryPolicy {
            max_attempts: 3,
            base_delay: std::time::Duration::from_millis(1),
            max_delay: std::time::Duration::from_millis(10),
        })
        .build()
        .unwrap();
    let bal = c.get_balance().await.unwrap();
    assert_eq!(bal.balance, 1.0);
}

#[tokio::test(flavor = "current_thread")]
async fn no_retry_on_400() {
    let server = MockServer::start().await;
    Mock::given(method("GET"))
        .and(path("/user/balance"))
        .respond_with(ResponseTemplate::new(400).set_body_json(serde_json::json!({
            "code": 4000, "message": "bad"
        })))
        .expect(1) // must be called exactly once
        .mount(&server)
        .await;

    let c = Client::builder()
        .api_key("tsk_test")
        .base_url(server.uri().parse().unwrap())
        .retry(RetryPolicy {
            max_attempts: 3,
            base_delay: std::time::Duration::from_millis(1),
            max_delay: std::time::Duration::from_millis(10),
        })
        .build()
        .unwrap();
    let _ = c.get_balance().await;
}