From daf4b8de04b60dfa908d6699c0d75b8143f1acd6 Mon Sep 17 00:00:00 2001
From: Russell Cohen <rcoh@amazon.com>
Date: Tue, 4 Jan 2022 21:13:12 -0500
Subject: [PATCH] Add missing recursion detection test (#1030)

---
 .../s3/tests/recursion-detection.rs           | 36 +++++++++++++++++++
 1 file changed, 36 insertions(+)
 create mode 100644 aws/sdk/integration-tests/s3/tests/recursion-detection.rs

diff --git a/aws/sdk/integration-tests/s3/tests/recursion-detection.rs b/aws/sdk/integration-tests/s3/tests/recursion-detection.rs
new file mode 100644
index 000000000..0bd7b44a7
--- /dev/null
+++ b/aws/sdk/integration-tests/s3/tests/recursion-detection.rs
@@ -0,0 +1,36 @@
+/*
+ * Copyright Amazon.com, Inc. or its affiliates. All Rights Reserved.
+ * SPDX-License-Identifier: Apache-2.0.
+ */
+
+use aws_sdk_s3::{Credentials, Region};
+use aws_smithy_client::test_connection::capture_request;
+use http::HeaderValue;
+
+#[tokio::test]
+async fn recursion_detection_applied() {
+    std::env::set_var("AWS_LAMBDA_FUNCTION_NAME", "some-function");
+    std::env::set_var("_X_AMZ_TRACE_ID", "traceid");
+    let (conn, captured_request) = capture_request(None);
+
+    let creds = Credentials::new(
+        "ANOTREAL",
+        "notrealrnrELgWzOk3IfjzDKtFBhDby",
+        Some("notarealsessiontoken".to_string()),
+        None,
+        "test",
+    );
+    let conf = aws_sdk_s3::Config::builder()
+        .credentials_provider(creds)
+        .region(Region::new("us-east-1"))
+        .build();
+    let client = aws_sdk_s3::Client::from_conf_conn(conf, conn);
+    let _response = client.list_objects_v2().bucket("test-bucket").send().await;
+    assert_eq!(
+        captured_request
+            .expect_request()
+            .headers()
+            .get("x-amzn-trace-id"),
+        Some(&HeaderValue::from_static("traceid"))
+    );
+}
-- 
GitLab