Add exception command (#6)

* Add exception command * Fix lint --------- Co-authored-by: Augusto Vanderley <[email protected]>
NeowayLabs · Oct 19, 2023 · b8c774d · b8c774d
1 parent 3f66eb8
commit b8c774d
Show file tree

Hide file tree

Showing 2 changed files with 21 additions and 0 deletions.
diff --git a/lodge.py b/lodge.py
@@ -118,6 +118,9 @@ def warn(self, msg, *args, **kwargs):
     def error(self, msg, *args, **kwargs):
         self._get_logger().error(msg, *args, **kwargs)
 
+    def exception(self, msg, *args, **kwargs):
+        self._get_logger().exception(msg, *args, **kwargs)
+
     def fatal(self, msg, *args, **kwargs):
         self._get_logger().fatal(msg, *args, **kwargs)
 

diff --git a/test_lodge.py b/test_lodge.py
@@ -153,3 +153,21 @@ def test_proxy_log_warn_with_level_error_should_not_log(stream, monkeypatch):
     log_entry = stream.read()
 
     assert log_entry == ""
+
+
+def test_proxy_exception_method(stream):
+    with import_lodge() as lodge:
+        log = lodge.get_logger("test")
+        try:
+            1/0
+        except ZeroDivisionError:
+            log.exception("test_message")
+
+    log_entry = stream.read()
+    json_entry = log_entry.split("\n")[0]
+    log_structured = json.loads(json_entry)
+
+    assert log_structured["message"] == "test_message"
+    assert log_structured["level"] == "ERROR"
+    assert "Traceback (most recent call last)" in log_entry
+    assert "ZeroDivisionError: division by zero" in log_entry