catch ingest errors in Logging handler

Author: dasfmi

src/axiom_py/logging.py

@@ -64,4 +64,8 @@ def flush(self):
             return
 
         local_buffer, self.buffer = self.buffer, []
-        self.client.ingest_events(self.dataset, local_buffer)
+        try:
+            self.client.ingest_events(self.dataset, local_buffer)
+        except Exception as e:
+            print(f"Error ingesting logs: {e}")
+            self.buffer = local_buffer

tests/test_logger.py

@@ -58,3 +58,26 @@ def test_log(self):
 
         # Cleanup created dataset
         client.datasets.delete(dataset_name)
+
+    def test_handling_network_errors(self):
+        client = Client(
+            "xapt-",
+            "NO_ORG_ID",
+            "https://WRONG_URL.com",
+        )
+        # Create a dataset for that purpose
+        dataset_name = get_random_name()
+
+        axiom_handler = AxiomHandler(client, dataset_name, interval=1.0)
+
+        logger = logging.getLogger()
+        logger.addHandler(axiom_handler)
+
+        logger.warning("This is a log!")
+        axiom_handler.flush()
+
+        # wait for the flush to finish
+        time.sleep(1.5)
+
+        # ensure no exceptions were raised
+        self.assertEqual(1, len(axiom_handler.buffer))