update tests

Author: liustve

aws-opentelemetry-distro/src/amazon/opentelemetry/distro/exporter/otlp/aws/common/aws_auth_session.py

@@ -34,7 +34,7 @@ class AwsAuthSession(requests.Session):
         service (str): The AWS service name for signing (e.g., "logs" or "xray")
     """
 
-    def __init__(self, aws_region: str, service: str, session: Session = Session()):
+    def __init__(self, aws_region: str, service: str, session: Session):
         self._aws_region: str = aws_region
         self._service: str = service
         self._session: Session = session

aws-opentelemetry-distro/tests/amazon/opentelemetry/distro/exporter/otlp/aws/common/test_aws_auth_session.py

@@ -6,6 +6,7 @@
 import requests
 from botocore.credentials import Credentials
 
+from amazon.opentelemetry.distro._utils import get_aws_session
 from amazon.opentelemetry.distro.exporter.otlp.aws.common.aws_auth_session import AwsAuthSession
 
 AWS_OTLP_TRACES_ENDPOINT = "https://xray.us-east-1.amazonaws.com/v1/traces"
@@ -24,7 +25,7 @@ class TestAwsAuthSession(TestCase):
     def test_aws_auth_session_no_credentials(self, _, __):
         """Tests that aws_auth_session will not inject SigV4 Headers if retrieving credentials returns None."""
 
-        session = AwsAuthSession("us-east-1", "xray")
+        session = AwsAuthSession("us-east-1", "xray", get_aws_session())
         actual_headers = {"test": "test"}
 
         session.request("POST", AWS_OTLP_TRACES_ENDPOINT, data="", headers=actual_headers)
@@ -38,7 +39,7 @@ def test_aws_auth_session_no_credentials(self, _, __):
     def test_aws_auth_session(self, _, __):
         """Tests that aws_auth_session will inject SigV4 Headers if botocore is installed."""
 
-        session = AwsAuthSession("us-east-1", "xray")
+        session = AwsAuthSession("us-east-1", "xray", get_aws_session())
         actual_headers = {"test": "test"}
 
         session.request("POST", AWS_OTLP_TRACES_ENDPOINT, data="", headers=actual_headers)

aws-opentelemetry-distro/tests/amazon/opentelemetry/distro/exporter/otlp/aws/logs/test_otlp_aws_logs_exporter.py

@@ -201,7 +201,32 @@ def test_export_interrupted_by_shutdown(self, mock_request, mock_wait):
         # Should make one request, then get interrupted during retry wait
         self.assertEqual(mock_request.call_count, 1)
         self.assertEqual(result, LogExportResult.FAILURE)
-        mock_wait.assert_called_once()
+
+    @patch("requests.Session.post", return_value=good_response)
+    def test_export_with_log_group_and_stream_headers(self, mock_request):
+        """Tests that log_group and log_stream are properly set as headers when provided."""
+        log_group = "test-log-group"
+        log_stream = "test-log-stream"
+
+        exporter = OTLPAwsLogExporter(
+            session=get_aws_session(),
+            aws_region="us-east-1",
+            endpoint=self._ENDPOINT,
+            log_group=log_group,
+            log_stream=log_stream,
+        )
+
+        result = exporter.export(self.logs)
+
+        mock_request.assert_called_once()
+        self.assertEqual(result, LogExportResult.SUCCESS)
+
+        # Verify headers contain log group and stream
+        session_headers = exporter._session.headers
+        self.assertIn("x-aws-log-group", session_headers)
+        self.assertIn("x-aws-log-stream", session_headers)
+        self.assertEqual(session_headers["x-aws-log-group"], log_group)
+        self.assertEqual(session_headers["x-aws-log-stream"], log_stream)
 
     @staticmethod
     def generate_test_log_data(count=5):