Test issue

DataDog · Nov 5, 2024 · 49112f4 · 49112f4
1 parent 66c8708
commit 49112f4
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 2 deletions.
diff --git a/..._step_functions_customized_log_group.json → ...ctions_customized_log_group.approved.json b/..._step_functions_customized_log_group.json → ...ctions_customized_log_group.approved.json
@@ -2,7 +2,7 @@
     {
         "aws": {
             "awslogs": {
-                "logGroup": "/aws/vendedlogs/states/logs-to-traces-sequential-Logs",
+                "logGroup": "test/logs",
                 "logStream": "states/logs-to-traces-sequential/2022-11-10-15-50/7851b2d9",
                 "owner": "425362996713"
             }

diff --git a/aws/logs_monitoring/tests/test_awslogs_handler.py b/aws/logs_monitoring/tests/test_awslogs_handler.py
@@ -131,6 +131,7 @@ def test_awslogs_handler_step_functions_tags_added_properly(
 
     @patch("caching.cloudwatch_log_group_cache.CloudwatchLogGroupTagsCache.__init__")
     @patch("caching.cloudwatch_log_group_cache.send_forwarder_internal_metrics")
+    @patch.dict("os.environ", {"DD_STEP_FUNCTIONS_TRACE_ENABLED": "true"})
     def test_awslogs_handler_step_functions_customized_log_group(
         self,
         mock_forward_metrics,
@@ -165,7 +166,7 @@ def test_awslogs_handler_step_functions_customized_log_group(
             }
         }
         context = None
-        metadata = {"ddtags": "env:dev"}
+        metadata = {"ddsource": "postgresql", "ddtags": "env:dev"}
         mock_forward_metrics.side_effect = MagicMock()
         mock_cache_init.return_value = None
         cache_layer = CacheLayer("")