Make logzio handler log correctly after os.fork()

.gitignore

@@ -1,2 +1,3 @@
 *.pyc
 
+*egg-info/

logzio/sender.py

@@ -34,7 +34,9 @@ class LogzioSender:
 
         # Create a queue to hold logs
         self.queue = queue.Queue()
+        self._initialize_sending_thread()
 
+    def _initialize_sending_thread(self):
         self.sending_thread = Thread(target=self._drain_queue)
         self.sending_thread.daemon = False
         self.sending_thread.name = "logzio-sending-thread"
@@ -42,6 +44,8 @@ class LogzioSender:
 
     def append(self, logs_message):
         # Queue lib is thread safe, no issue here
+        if not self.sending_thread.is_alive():
+            self._initialize_sending_thread()
         self.queue.put(json.dumps(logs_message))
 
     def flush(self):

tests/test_logzioSender.py

@@ -106,3 +106,16 @@ class TestLogzioSender(TestCase):
         with open(failure_files[0], "r") as f:
             line = f.readline()
             self.assertTrue(log_message in line)
+
+    def test_can_send_after_fork(self):
+        childpid = os.fork()
+        log_message = 'logged from child process'
+
+        if childpid == 0:
+            # Log from the child process
+            self.logger.info(log_message)
+            time.sleep(self.logs_drain_timeout * 2)
+            os._exit(0)
+        # Wait for the child process to finish
+        os.waitpid(childpid, 0)
+        self.assertTrue(self.logzio_listener.find_log(log_message))