Skip to content

Commit

Permalink
Add test for server not responding
Browse files Browse the repository at this point in the history
  • Loading branch information
kindly committed Apr 26, 2023
1 parent b68d675 commit 747b90a
Showing 1 changed file with 13 additions and 0 deletions.
13 changes: 13 additions & 0 deletions object_store/src/client/retry.rs
Original file line number Diff line number Diff line change
Expand Up @@ -364,6 +364,19 @@ mod tests {
assert_eq!(e.retries, retry.max_retries);
assert_eq!(e.message, "502 Bad Gateway");

// Panic results in a incomplete message error in the client
mock.push_fn(|_| {panic!()});
let r = do_request().await.unwrap();
assert_eq!(r.status(), StatusCode::OK);

// Gives up after retrying mulitiple panics
for _ in 0..=retry.max_retries {
mock.push_fn(|_| {panic!()});
}
let e = do_request().await.unwrap_err();
assert_eq!(e.retries, retry.max_retries);
assert_eq!(e.message, "request error");

// Shutdown
mock.shutdown().await
}
Expand Down

0 comments on commit 747b90a

Please sign in to comment.