test: assert exactly one capture loss line per failed batch

Author: eli-r-ph

posthog/test/test_async_client.py

@@ -844,7 +844,14 @@ async def test_failed_capture_logs_one_loss_line_without_response_detail(
         await client.shutdown()
 
     assert "server-secret" not in caplog.text
-    assert f"1 event(s) not persisted by {path}: APIError (status=400)" in caplog.text
+    loss_lines = [
+        r.getMessage()
+        for r in caplog.records
+        if "event(s) not persisted by" in r.getMessage()
+    ]
+    assert loss_lines == [
+        f"[PostHog] 1 event(s) not persisted by {path}: APIError (status=400)"
+    ]
 
 
 @pytest.mark.asyncio

posthog/test/test_client.py

@@ -2789,9 +2789,14 @@ def test_sync_send_failure_without_on_error_logs_aggregate_line(
 
         self.assertIsNone(result)
         output = "\n".join(logs.output)
-        self.assertIn(
-            f"1 event(s) not persisted by {path}: APIError (status=400)",
-            output,
+        loss_lines = [
+            r.getMessage()
+            for r in logs.records
+            if "event(s) not persisted by" in r.getMessage()
+        ]
+        self.assertEqual(
+            loss_lines,
+            [f"[PostHog] 1 event(s) not persisted by {path}: APIError (status=400)"],
         )
         self.assertNotIn("server-secret", output)