Fix tests

Signed-off-by: Jesse Szwedko <[email protected]>
vectordotdev · Jul 31, 2024 · 53a4923 · 53a4923
1 parent a958960
commit 53a4923
Show file tree

Hide file tree

Showing 4 changed files with 10 additions and 1 deletion.
diff --git 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
@@ -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
@@ -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
@@ -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())