IQ.Pilot Release Commit @ b36e2e3

system/tests/test_logmessaged.py

@@ -0,0 +1,68 @@
+import glob
+import json
+import os
+import time
+
+import cereal.messaging as messaging
+from openpilot.system.manager.process_config import managed_processes
+from openpilot.system.hardware.hw import Paths
+from openpilot.common.swaglog import cloudlog, ipchandler
+
+
+class TestLogmessaged:
+  def setup_method(self):
+    # clear the IPC buffer in case some other tests used cloudlog and filled it
+    ipchandler.close()
+    ipchandler.connect()
+
+    managed_processes['logmessaged'].start()
+    self.sock = messaging.sub_sock("logMessage", timeout=1000, conflate=False)
+    self.error_sock = messaging.sub_sock("errorLogMessage", timeout=1000, conflate=False)
+
+    # ensure sockets are connected
+    time.sleep(0.5)
+    messaging.drain_sock(self.sock)
+    messaging.drain_sock(self.error_sock)
+
+  def teardown_method(self):
+    del self.sock
+    del self.error_sock
+    managed_processes['logmessaged'].stop(block=True)
+
+  def _get_log_files(self):
+    return list(glob.glob(os.path.join(Paths.swaglog_root(), "swaglog.*")))
+
+  def test_simple_log(self):
+    msgs = [f"abc {i}" for i in range(10)]
+    for m in msgs:
+      cloudlog.error(m)
+    time.sleep(0.5)
+    m = messaging.drain_sock(self.sock)
+    assert len(m) == len(msgs)
+    assert len(self._get_log_files()) >= 1
+
+  def test_big_log(self):
+    n = 10
+    msg = "a"*3*1024*1024
+    for _ in range(n):
+      cloudlog.info(msg)
+    time.sleep(0.5)
+
+    msgs = messaging.drain_sock(self.sock)
+    assert len(msgs) == 0
+
+    logsize = sum([os.path.getsize(f) for f in self._get_log_files()])
+    assert (n*len(msg)) < logsize < (n*(len(msg)+1024))
+
+  def test_large_log_below_publish_limit(self):
+    msg = "a" * (256 * 1024)
+    cloudlog.error(msg)
+    time.sleep(0.5)
+
+    msgs = messaging.drain_sock(self.sock)
+    error_msgs = messaging.drain_sock(self.error_sock)
+
+    assert len(msgs) == 1
+    assert json.loads(msgs[0].logMessage)["msg"] == msg
+    assert len(error_msgs) == 1
+    assert json.loads(error_msgs[0].errorLogMessage)["msg"] == msg