1 files changed, 19 insertions, 1 deletions
diff --git a/slog/slog_tests.py b/slog/slog_tests.py
index cc18f15..04f1cf4 100644
--- a/slog/slog_tests.py
+++ b/slog/slog_tests.py
@@ -187,7 +187,7 @@ class StructuredLogTests(unittest.TestCase):
         self.assertEqual(objs[0]['dikt'], dikt)
 
     def test_logs_to_two_files(self):
-        filename1 = os.path.join(self.tempdir, 'slog1XS')
+        filename1 = os.path.join(self.tempdir, 'slog1')
         writer1 = slog.FileSlogWriter()
         writer1.set_filename(filename1)
 
@@ -205,6 +205,24 @@ class StructuredLogTests(unittest.TestCase):
 
         self.assertEqual(objs1, objs2)
 
+    def test_massages_log_objects(self):
+
+        def add_foo(log_obj):
+            log_obj['foo'] = 'this is foo'
+            return log_obj
+
+        filename = os.path.join(self.tempdir, 'slog.log')
+        writer = slog.FileSlogWriter()
+        writer.set_filename(filename)
+
+        log = slog.StructuredLog()
+        log.add_log_writer(writer, slog.FilterAllow())
+        log.add_log_massager(add_foo)
+
+        log.log('test', msg_text='hello')
+        objs = self.read_log_entries(writer)
+        self.assertEqual(objs[0]['foo'], 'this is foo')
+
 
 class FileSlogWriterTests(unittest.TestCase):