dt-extensions-sdk 1.1.20__tar.gz → 1.1.22__tar.gz

{dt_extensions_sdk-1.1.20 → dt_extensions_sdk-1.1.22}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.3
 Name: dt-extensions-sdk
-Version: 1.1.20
+Version: 1.1.22
 Project-URL: Documentation, https://github.com/dynatrace-extensions/dt-extensions-python-sdk#readme
 Project-URL: Issues, https://github.com/dynatrace-extensions/dt-extensions-python-sdk/issues
 Project-URL: Source, https://github.com/dynatrace-extensions/dt-extensions-python-sdk

{dt_extensions_sdk-1.1.20 → dt_extensions_sdk-1.1.22}/dynatrace_extension/__about__.py

@@ -2,4 +2,4 @@
 #
 # SPDX-License-Identifier: MIT
 
-__version__ = "1.1.20"
+__version__ = "1.1.22"

{dt_extensions_sdk-1.1.20 → dt_extensions_sdk-1.1.22}/dynatrace_extension/sdk/communication.py

@@ -415,6 +415,8 @@ class DebugClient(CommunicationClient):
 
     def send_events(self, events: dict | list[dict], eec_enrichment: bool = True) -> list[dict | None]:
         self.logger.info(f"send_events (enrichment = {eec_enrichment}): {len(events)} events")
+        if isinstance(events, dict):
+            events = [events]
         if self.print_metrics:
             for event in events:
                 self.logger.info(f"send_event: {event}")
@@ -444,7 +446,9 @@ def divide_into_batches(items: Sequence[dict | str], max_size_bytes: int, join_w
 
     if join_with is not None:
         items = join_with.join(items)
-    encoded = f"{items}".encode(errors="replace")
+        encoded = f"{items}".encode(errors="replace")
+    else:
+        encoded = json.dumps(items).encode(errors="replace")
     size = len(encoded)
     if size <= max_size_bytes:
         yield encoded

{dt_extensions_sdk-1.1.20 → dt_extensions_sdk-1.1.22}/tests/sdk/test_communication.py

@@ -1,3 +1,4 @@
+import json
 import unittest
 from unittest.mock import MagicMock, mock_open, patch
 
@@ -53,13 +54,13 @@ class TestCommunication(unittest.TestCase):
 
     def test_large_metric_chunk(self):
 
-        metrics = ['my.metric,dim="dim" 10'] * 500 * 100  # 1_300_000 bytes, but becomes 1_149_999 with the newlines
+        metrics = ['my.metric,dim="dim" 10'] * 500 * 100
 
         # it needs to be divided into 2 lists, each with 650_000 bytes
         chunks = list(divide_into_batches(metrics, MAX_METRIC_REQUEST_SIZE, "\n"))
         self.assertEqual(len(chunks), 2)
-        self.assertEqual(len(chunks[0]), 574999)
-        self.assertEqual(len(chunks[1]), 575000)
+        self.assertEqual(len(chunks[0]), 650000)
+        self.assertEqual(len(chunks[1]), 650002)
 
     def test_small_metric_chunk(self):
         metrics = ['my.metric,dim="dim" 10'] * 100
@@ -73,3 +74,19 @@ class TestCommunication(unittest.TestCase):
 
         chunks = list(divide_into_batches(metrics, MAX_METRIC_REQUEST_SIZE, "\n"))
         self.assertEqual(len(chunks), 0)
+
+    def test_large_log_chunk_valid_json(self):
+
+        events = []
+        for i in range(5000):
+            attributes = {}
+            for j in range(150):
+                attributes[f"attribute{j}"] = j
+            events.append(attributes)
+
+        # it needs to be divided into 4 lists, each with 3_665_000 bytes
+        chunks = list(divide_into_batches(events, MAX_LOG_REQUEST_SIZE))
+        self.assertEqual(len(chunks), 4)
+
+        for chunk in chunks:
+            json.loads(chunk)