|
1 | 1 | mod support; |
2 | 2 |
|
3 | | -use std::time::Duration; |
| 3 | +use std::{ |
| 4 | + sync::{ |
| 5 | + Arc, |
| 6 | + atomic::{AtomicUsize, Ordering}, |
| 7 | + }, |
| 8 | + time::Duration, |
| 9 | +}; |
4 | 10 |
|
5 | 11 | use futures_util::future::join_all; |
6 | 12 | use hpx::{Body, Client, ClientResponseBody}; |
@@ -156,6 +162,42 @@ async fn with_connect_timeout_layer_slow() { |
156 | 162 | assert!(err.is_timeout()); |
157 | 163 | } |
158 | 164 |
|
| 165 | +#[tokio::test] |
| 166 | +async fn timeout_layer_does_not_break_follow_up_requests() { |
| 167 | + let _ = env_logger::try_init(); |
| 168 | + |
| 169 | + let request_count = Arc::new(AtomicUsize::new(0)); |
| 170 | + let server_count = Arc::clone(&request_count); |
| 171 | + let server = server::http(move |_req| { |
| 172 | + let server_count = Arc::clone(&server_count); |
| 173 | + async move { |
| 174 | + let current = server_count.fetch_add(1, Ordering::SeqCst); |
| 175 | + if current == 0 { |
| 176 | + tokio::time::sleep(Duration::from_millis(200)).await; |
| 177 | + } |
| 178 | + |
| 179 | + http::Response::new(hpx::Body::from(format!("response-{current}"))) |
| 180 | + } |
| 181 | + }); |
| 182 | + |
| 183 | + let url = format!("http://{}/slow", server.addr()); |
| 184 | + |
| 185 | + let client = Client::builder() |
| 186 | + .layer(TimeoutLayer::new(Duration::from_millis(50))) |
| 187 | + .no_proxy() |
| 188 | + .build() |
| 189 | + .unwrap(); |
| 190 | + |
| 191 | + let first_err = client.get(&url).send().await.unwrap_err(); |
| 192 | + assert!(first_err.is_timeout()); |
| 193 | + |
| 194 | + let second = client.get(&url).send().await.unwrap(); |
| 195 | + let body = second.text().await.unwrap(); |
| 196 | + |
| 197 | + assert_eq!(body, "response-1"); |
| 198 | + assert_eq!(request_count.load(Ordering::SeqCst), 2); |
| 199 | +} |
| 200 | + |
159 | 201 | #[tokio::test] |
160 | 202 | async fn multiple_timeout_layers_under_threshold() { |
161 | 203 | let _ = env_logger::try_init(); |
|
0 commit comments