Update Utests

DataDog · Dec 5, 2023 · 1bdb1cf · 1bdb1cf
1 parent e3fa8bb
commit 1bdb1cf
Showing 1 changed file with 5 additions and 1 deletion.
diff --git a/aws/logs_monitoring/tests/test_parsing.py b/aws/logs_monitoring/tests/test_parsing.py
@@ -835,6 +835,7 @@ def test_get_structured_lines_waf(self):
         lines = yield from get_structured_lines_for_s3_handler(
             data, bucket, key, source
         )
+        lines = list(lines)
 
         self.assertEqual(len(lines), 3)
         self.assertEqual(lines, expected_lines)
@@ -852,6 +853,7 @@ def test_get_structured_lines_waf_single_line(self):
         lines = yield from get_structured_lines_for_s3_handler(
             data, bucket, key, source
         )
+        lines = list(lines)
 
         self.assertEqual(len(lines), 1)
         self.assertEqual(lines, expected_lines)
@@ -865,6 +867,7 @@ def test_get_structured_lines_empty_waf(self):
         lines = yield from get_structured_lines_for_s3_handler(
             data, bucket, key, source
         )
+        lines = list(lines)
 
         self.assertEqual(len(lines), 0)
 
@@ -880,9 +883,9 @@ def test_get_structured_lines_cloudtrail(self):
         lines = yield from get_structured_lines_for_s3_handler(
             data, bucket, key, source
         )
+        lines = list(lines)
 
         self.assertEqual(len(lines), 1)
-        print(lines[0])
         self.assertEqual(lines[0], expected_line)
 
     def test_get_structured_lines_empty_cloudtrail(self):
@@ -894,6 +897,7 @@ def test_get_structured_lines_empty_cloudtrail(self):
         lines = yield from get_structured_lines_for_s3_handler(
             data, bucket, key, source
         )
+        lines = list(lines)
 
         self.assertEqual(len(lines), 0)