modify test

Author: aoyagikouhei

examples/oauth-web/Cargo.toml

@@ -10,11 +10,9 @@ axum = "0.8"
 serde = { version = "1", features = ["derive"] }
 tokio = { version = "1", features = ["full"] }
 tower-cookies = "0.11.0"
-twapi-v2 = { path="../..", features = ["retry"] }
+twapi-v2 = { path="../.." }
 twapi-oauth2 = "0.1.0"
 
 [features]
-retry = []
 oauth10a = []
 models = []
-upload = []

examples/post-media/Cargo.toml

@@ -14,7 +14,5 @@ tracing-subscriber = "0.3"
 twapi-v2 = { path="../..", features = [] }
 
 [features]
-retry = []
 oauth10a = []
 models = []
-upload = []

examples/streaming/Cargo.toml

@@ -11,11 +11,9 @@ futures-util = "0.3"
 reqwest = { version = "0.12", features = ["stream"] }
 serde_json = "1"
 tokio = { version = "1", features = ["full"] }
-twapi-v2 = { path="../..", features = ["retry", "oauth10a"] }
+twapi-v2 = { path="../..", features = ["oauth10a"] }
 wasm-bindgen = "0.2.92"
 
 [features]
-retry = []
 oauth10a = []
 models = []
-upload = []

src/api.rs

@@ -226,7 +226,7 @@ where
             });
         }
 
-        if count
+        if count + 1
             >= twapi_options
                 .as_ref()
                 .and_then(|options| options.try_count)

tests/get_2_tweets.rs

@@ -1,15 +1,23 @@
+use std::time::Duration;
+
 use anyhow::Result;
-use twapi_v2::api::{BearerAuthentication, execute_twitter, get_2_tweets};
+use twapi_v2::api::{BearerAuthentication, TwapiOptions, execute_twitter, get_2_tweets};
 
 //BEARER_CODE=XXXXX cargo test --test get_2_tweets --all-features -- --nocapture --test-threads=1
 
 #[tokio::test]
 async fn test_get_2_tweets_tweets() -> Result<()> {
     let bearer_code = std::env::var("BEARER_CODE").unwrap_or_default();
     let bearer_auth = BearerAuthentication::new(bearer_code);
+    let twapi_option = TwapiOptions {
+        try_count: Some(4),
+        retryable_status_codes: Some(vec![500, 502, 503, 504, 401]),
+        retry_interval_duration: Some(Duration::from_millis(1000)),
+        ..Default::default()
+    };
     let (res, rate_limit) = execute_twitter::<serde_json::Value>(
-        || get_2_tweets::Api::all("1648835862149619712,1653186227338874881").build(&bearer_auth),
-        &None,
+        || get_2_tweets::Api::all("1,2").build(&bearer_auth),
+        &Some(twapi_option),
     )
     .await?;
     println!("{}", serde_json::to_string(&res).unwrap());

tests/get_2_users_id.rs

@@ -1,6 +1,6 @@
 use anyhow::Result;
 use twapi_v2::{
-    api::{BearerAuthentication, execute_twitter, get_2_users_id},
+    api::{BearerAuthentication, TwapiOptions, execute_twitter, get_2_users_id},
     fields::user_fields::UserFields,
 };
 
@@ -10,13 +10,17 @@ use twapi_v2::{
 async fn test_get_2_users_id() -> Result<()> {
     let bearer_code = std::env::var("BEARER_CODE").unwrap_or_default();
     let bearer_auth = BearerAuthentication::new(bearer_code);
+    let twapi_option = TwapiOptions {
+        try_count: Some(4),
+        ..Default::default()
+    };
     let (res, _rate_limit) = execute_twitter::<serde_json::Value>(
         || {
-            get_2_users_id::Api::open("1660518823991336966")
+            get_2_users_id::Api::open("1")
                 .user_fields(UserFields::all())
                 .build(&bearer_auth)
         },
-        &None,
+        &Some(twapi_option),
     )
     .await?;
     println!("{}", serde_json::to_string(&res).unwrap());