Tests updated

open-telemetry · Oct 9, 2024 · 171087f · 171087f
1 parent b5daf85
commit 171087f
Showing 1 changed file with 24 additions and 6 deletions.
diff --git a/opentelemetry-sdk/tests/logs/test_log_record.py b/opentelemetry-sdk/tests/logs/test_log_record.py
@@ -79,9 +79,18 @@ def test_log_record_dropped_attributes_set_limits_max_attribute(self):
             max_attributes=1,
         )
 
-        result = LogRecord(
-            timestamp=0, body="a log line", attributes=attr, limits=limits
-        )
+        with warnings.catch_warnings(record=True) as cw:
+            warnings.simplefilter("always")
+            result = LogRecord(
+                timestamp=0, body="a log line", attributes=attr, limits=limits
+            )
+
+            self.assertTrue(
+                any(
+                    issubclass(warning.category, LogDroppedAttributesWarning)
+                    for warning in cw
+                )
+            )
         self.assertTrue(result.dropped_attributes == 1)
 
     def test_log_record_dropped_attributes_set_limits_max_attribute_length(
@@ -93,9 +102,18 @@ def test_log_record_dropped_attributes_set_limits_max_attribute_length(
             max_attribute_length=1,
         )
 
-        result = LogRecord(
-            timestamp=0, body="a log line", attributes=attr, limits=limits
-        )
+        with warnings.catch_warnings(record=True) as cw:
+            warnings.simplefilter("always")
+            result = LogRecord(
+                timestamp=0, body="a log line", attributes=attr, limits=limits
+            )
+            # would not be dropping any attributes if the limit is not set
+            self.assertFalse(
+                any(
+                    issubclass(warning.category, LogDroppedAttributesWarning)
+                    for warning in cw
+                )
+            )
         self.assertTrue(result.dropped_attributes == 0)
         self.assertEqual(expected, result.attributes)