Drop message display() functions in favor of to_str
authorReto Buerki <reet@codelabs.ch>
Thu, 14 Sep 2017 12:17:55 +0000 (14:17 +0200)
committerReto Buerki <reet@codelabs.ch>
Fri, 15 Sep 2017 07:16:18 +0000 (09:16 +0200)
src/message.py
test/test_message.py

index c99a464..58a81d4 100644 (file)
@@ -35,8 +35,8 @@ class Message():
         self.msg = msg
         addToCount(Message)
 
-    def display(self):
-        print textWrapper.fill(self.msg)
+    def to_str(self):
+        return textWrapper.fill(self.msg)
 
 
 class WarningMessage(Message):
@@ -46,8 +46,8 @@ class WarningMessage(Message):
         self.msg = msg
         addToCount(WarningMessage)
 
-    def display(self):
-        print textWrapper.fill("* WARNING *: %s" % self.msg)
+    def to_str(self):
+        return textWrapper.fill("* WARNING *: %s" % self.msg)
 
 
 class ErrorMessage(Message):
@@ -57,8 +57,8 @@ class ErrorMessage(Message):
         self.msg = msg
         addToCount(ErrorMessage)
 
-    def display(self):
-        print textWrapper.fill("** ERROR **: %s" % self.msg)
+    def to_str(self):
+        return textWrapper.fill("** ERROR **: %s" % self.msg)
 
 
 def addToCount(classname):
@@ -70,7 +70,7 @@ def addToCount(classname):
 
 def printMessages():
     for message in messagequeue:
-        message.display()
+        print message.to_str()
         print ""
 
 
index 96174e6..5d85eea 100644 (file)
@@ -29,6 +29,13 @@ class TestMessage:
         message.reset()
         assert len(message.messagequeue) == 0
 
+    def test_str(self):
+        message.reset()
+        message.addWarning("Warning1")
+        assert message.messagequeue[0].to_str() == "* WARNING *: Warning1"
+        message.addError("Error1", False)
+        assert message.messagequeue[1].to_str() == "** ERROR **: Error1"
+
     def test_addMessage(self):
         message.reset()
         message.addWarning("Warning1")