Remove the stop() method

Author: thatbudakguy

mqlog/__init__.py

@@ -37,22 +37,16 @@ def __init__(
         self.capacity = capacity
         self.buffer = []
         self.will_flush = asyncio.Event()
-        self.active = True
 
     async def run(self):
         """
         Continuously publish log records via MQTT.
         This method should be scheduled as an asyncio task.
         """
-        self.active = True
-        while self.active:
+        while True:
             await self.will_flush.wait()
             await self._flush()
 
-    def stop(self):
-        """Stop the handler from processing further log records."""
-        self.active = False
-
     # Check if we should publish an MQTT message
     def _should_flush(self, record):
         return (len(self.buffer) >= self.capacity) or (

tests/test_handler.py

@@ -48,15 +48,14 @@ def test_publish(self):
         """Flushing should publish messages to MQTT topic"""
         self.handler.flush_level = logging.ERROR
 
-        async def do_test(handler, logger):
+        async def do_test(logger):
             logger.info("Test message 1")
             await asyncio.sleep(0.1)
             logger.error("Test message 2")
             await asyncio.sleep(0.1)
-            handler.stop()
 
         async def main(handler, logger):
-            await asyncio.gather(handler.run(), do_test(handler, logger))
+            await asyncio.gather(handler.run(), do_test(logger))
 
         asyncio.run(main(self.handler, self.logger))
 
@@ -68,7 +67,7 @@ def test_flush_multiple(self):
         """Flushing multiple times should publish separate messages"""
         self.handler.capacity = 2
 
-        async def do_test(handler, logger):
+        async def do_test(logger):
             logger.info("Test message 1")
             await asyncio.sleep(0.1)
             logger.info("Test message 2")
@@ -77,10 +76,9 @@ async def do_test(handler, logger):
             await asyncio.sleep(0.1)
             logger.info("Test message 4")
             await asyncio.sleep(0.1)
-            handler.stop()
 
         async def main(handler, logger):
-            await asyncio.gather(handler.run(), do_test(handler, logger))
+            await asyncio.gather(handler.run(), do_test(logger))
 
         asyncio.run(main(self.handler, self.logger))