From 53a4923a2f340d4d3e01f2bc589c1ca07da19e79 Mon Sep 17 00:00:00 2001
From: Jesse Szwedko <jesse.szwedko@datadoghq.com>
Date: Wed, 31 Jul 2024 09:13:44 -0700
Subject: [PATCH] Fix tests

Signed-off-by: Jesse Szwedko <jesse.szwedko@datadoghq.com>
---
 src/aws/auth.rs                                     | 4 +++-
 src/sinks/aws_kinesis/firehose/integration_tests.rs | 3 +++
 src/sinks/elasticsearch/integration_tests.rs        | 3 +++
 src/sources/aws_sqs/integration_tests.rs            | 1 +
 4 files changed, 10 insertions(+), 1 deletion(-)

diff --git a/src/aws/auth.rs b/src/aws/auth.rs
index 4180495195af0..3f3899f4fd4cb 100644
--- a/src/aws/auth.rs
+++ b/src/aws/auth.rs
@@ -425,6 +425,7 @@ mod tests {
                 load_timeout_secs: Some(10),
                 imds: ImdsAuthentication { .. },
                 region: None,
+                sign: true,
             }
         ));
     }
@@ -467,6 +468,7 @@ mod tests {
                     connect_timeout: CONNECT_TIMEOUT,
                     read_timeout: READ_TIMEOUT,
                 },
+                sign: true,
             }
         ));
     }
@@ -702,7 +704,7 @@ mod tests {
 
         match config.auth {
             AwsAuthentication::Default { sign, .. } => {
-                assert_eq!(&no_sign, false);
+                assert!(!sign);
             }
             _ => panic!(),
         }
diff --git a/src/sinks/aws_kinesis/firehose/integration_tests.rs b/src/sinks/aws_kinesis/firehose/integration_tests.rs
index fc9f22e5c5a73..538a8895747b7 100644
--- a/src/sinks/aws_kinesis/firehose/integration_tests.rs
+++ b/src/sinks/aws_kinesis/firehose/integration_tests.rs
@@ -83,6 +83,7 @@ async fn firehose_put_records_without_partition_key() {
             load_timeout_secs: Some(5),
             imds: ImdsAuthentication::default(),
             region: None,
+            sign: true,
         })),
         endpoints: vec![elasticsearch_address()],
         bulk: BulkConfig {
@@ -195,6 +196,7 @@ async fn firehose_put_records_with_partition_key() {
             load_timeout_secs: Some(5),
             imds: ImdsAuthentication::default(),
             region: None,
+            sign: true,
         })),
         endpoints: vec![elasticsearch_address()],
         bulk: BulkConfig {
@@ -278,6 +280,7 @@ async fn ensure_elasticsearch_domain(domain_name: String) -> String {
                         &None,
                     )
                     .await
+                    .unwrap()
                     .unwrap(),
             )
             .endpoint_url(test_region_endpoint().endpoint().unwrap())
diff --git a/src/sinks/elasticsearch/integration_tests.rs b/src/sinks/elasticsearch/integration_tests.rs
index 1f627b198f8c8..53c9a088d1fa3 100644
--- a/src/sinks/elasticsearch/integration_tests.rs
+++ b/src/sinks/elasticsearch/integration_tests.rs
@@ -290,6 +290,7 @@ async fn auto_version_aws() {
                 load_timeout_secs: Some(5),
                 imds: ImdsAuthentication::default(),
                 region: None,
+                sign: true,
             },
         )),
         endpoints: vec![aws_server()],
@@ -396,6 +397,7 @@ async fn insert_events_on_aws() {
                     load_timeout_secs: Some(5),
                     imds: ImdsAuthentication::default(),
                     region: None,
+                    sign: true,
                 },
             )),
             endpoints: vec![aws_server()],
@@ -422,6 +424,7 @@ async fn insert_events_on_aws_with_compression() {
                     load_timeout_secs: Some(5),
                     imds: ImdsAuthentication::default(),
                     region: None,
+                    sign: true,
                 },
             )),
             endpoints: vec![aws_server()],
diff --git a/src/sources/aws_sqs/integration_tests.rs b/src/sources/aws_sqs/integration_tests.rs
index eec750b00566c..23bf85b41f0de 100644
--- a/src/sources/aws_sqs/integration_tests.rs
+++ b/src/sources/aws_sqs/integration_tests.rs
@@ -55,6 +55,7 @@ async fn get_sqs_client() -> aws_sdk_sqs::Client {
             AwsAuthentication::test_auth()
                 .credentials_provider(Region::new("custom"), &Default::default(), &None)
                 .await
+                .unwrap()
                 .unwrap(),
         )
         .endpoint_url(sqs_address())