dnrops
/
smithy-rs
mirror of https://github.com/smithy-lang/smithy-rs
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

Add test to verify retry of 401s in IMDS client (#1729)

This commit is contained in:
John DiSanti 2022-09-13 10:31:30 -07:00 committed by GitHub
parent b4ac306fc2
commit 2312843197
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 27 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

27
aws/rust-runtime/aws-config/src/imds/client.rs
View File

@ -989,6 +989,33 @@ pub(crate) mod test {
connection.assert_requests_match(&[]);
}
/// 401 error during metadata retrieval must be retried
#[tokio::test]
#[traced_test]
async fn retry_metadata_401() {
let connection = TestConnection::new(vec![
(
token_request("http://169.254.169.254", 21600),
token_response(0, TOKEN_A),
),
(
imds_request("http://169.254.169.254/latest/metadata", TOKEN_A),
http::Response::builder().status(401).body("").unwrap(),
),
(
token_request("http://169.254.169.254", 21600),
token_response(21600, TOKEN_B),
),
(
imds_request("http://169.254.169.254/latest/metadata", TOKEN_B),
imds_response("ok"),
),
]);
let client = make_client(&connection).await;
assert_eq!(client.get("/latest/metadata").await.expect("success"), "ok");
connection.assert_requests_match(&[]);
}
/// 403 responses from IMDS during token acquisition MUST NOT be retried
#[tokio::test]
#[traced_test]