add test for them

test/logger/formatter.py

@@ -4,14 +4,32 @@
 #  All rights reserved
 #  -------------------------------
 
+import inspect
 import unittest
 
-from lib_not_dr.logger.formatter import BaseFormatter, TimeFormatter, LevelFormatter
+from lib_not_dr.logger.formatter import BaseFormatter, TimeFormatter, LevelFormatter, TraceFormatter
 from lib_not_dr.logger.structers import LogMessage
 
 
 class FormatterTest(unittest.TestCase):
 
+    def test_create_message(self):
+        message = LogMessage(messages=['test'],
+                             level=20,
+                             logger_name='root',
+                             logger_tag='root',
+                             stack_trace=inspect.currentframe())
+        self.assertEqual(message.messages, ['test'])
+        self.assertEqual(message.level, 20)
+        self.assertEqual(message.logger_name, 'root')
+        self.assertEqual(message.logger_tag, 'root')
+
+    def test_create_empty_tag_message(self):
+        message = LogMessage()
+
+        self.assertIsNone(message.logger_tag)
+        self.assertEqual(message.format_for_message()['logger_tag'], '   ')
+
     def test_formatter(self):
         formatter = BaseFormatter()
         formatter.info()
@@ -27,3 +45,7 @@ class FormatterTest(unittest.TestCase):
     def test_std_formatter(self):
         formatter = BaseFormatter()
         formatter.info()
+
+    def test_trace_formatter(self):
+        formatter = TraceFormatter()
+        formatter.info()