Adding log tests

mathom · mathom · commit 32d165a37ea2 · 2014-10-15T13:19:51.000-05:00
diff --git a/tests/test_formats.py b/tests/test_formats.py
@@ -39,6 +39,18 @@ def __init__(self, color):
 
         self.assertEqual(json.loads(result), {'children': [{'color': 'red'}, {'color': 'green'}]})
 
+    def test_nonempty_attributes(self):
+        '''Only nonempty attributes are serialized'''
+        class MyTest(JSONObject):
+            def __init__(self):
+                self.a = '1'
+                self.b = False
+                self.c = None
+                self.d = []
+        result = MyTest().toJSON()
+
+        self.assertEqual(json.loads(result), {'a': '1', 'b': False, 'd': []})
+
 
 if __name__=='__main__':
     unittest.main()
diff --git a/tests/test_log.py b/tests/test_log.py
@@ -0,0 +1,44 @@
+#!/usr/bin/env python
+"""
+Test the stackify.log module
+"""
+
+import unittest
+from mock import patch, Mock
+import json
+
+import stackify.log
+
+from stackify.log import LogMsg
+import logging
+import json
+import time
+#logging.LogRecord('name','level','pathname','lineno','msg','args','exc_info','func')
+
+
+class TestLogPopulate(unittest.TestCase):
+    '''
+    Test populating log objects with data
+    '''
+
+    def test_record_to_error(self):
+        '''LogMsgs can load logger records'''
+        record = logging.LogRecord('name',logging.WARNING,'pathname',32,
+                'message',(),(),'func')
+        record.my_extra = [1,2,3]
+        msg = LogMsg()
+        msg.from_record(record)
+
+        curr_ms = time.time() * 1000
+
+        self.assertEqual(msg.SrcMethod, 'func')
+        self.assertEqual(msg.SrcLine, 32)
+        self.assertEqual(msg.Th, 'MainThread')
+        self.assertEqual(msg.Msg, 'message')
+        self.assertTrue(msg.EpochMs <= curr_ms)
+        self.assertEqual(json.loads(msg.data), {'my_extra':[1,2,3]})
+
+
+if __name__=='__main__':
+    unittest.main()
+
