fix unit test error

DataDog · Nov 4, 2024 · acb99c6 · acb99c6
1 parent f951348
commit acb99c6
Showing 1 changed file with 5 additions and 1 deletion.
diff --git a/aws/logs_monitoring/tests/test_awslogs_handler.py b/aws/logs_monitoring/tests/test_awslogs_handler.py
@@ -123,6 +123,10 @@ def test_awslogs_handler_step_functions_tags_added_properly(
         self.assertEqual(
             awslogs_handler.metadata[DD_SOURCE], AwsEventSource.STEPFUNCTION.value
         )
+        self.assertEqual(
+            awslogs_handler.metadata[DD_HOST],
+            "arn:aws:states:us-east-1:12345678910:stateMachine:StepFunction1",
+        )
 
         # SF customized log group
         eventFromCustomizedLogGroup = {
@@ -158,7 +162,7 @@ def test_awslogs_handler_step_functions_tags_added_properly(
         )
         self.assertEqual(
             awslogs_handler.metadata[DD_HOST],
-            "arn:aws:states:us-east-1:12345678910:statemachine:stepfunction1",
+            "arn:aws:states:us-east-1:12345678910:stateMachine:StepFunction1",
         )
 
         eventFromCustomizedLogGroup = {