[CONFIG] Remove unneeded option

[rarslave2.git] / rarslave-test.py

diff --git a/rarslave-test.py b/rarslave-test.py
index 389dcf2..f82fc12 100644 (file)
--- a/rarslave-test.py
+++ b/rarslave-test.py
@@ -3,8 +3,7 @@
 
 import os, sys, unittest
 from rarslave import *
-from RarslaveLogger import RarslaveLogger
-from RarslaveLogger import RarslaveMessage
+from RarslaveLogger import RarslaveLogger,RarslaveMessage,MessageType
 
 class rarslavetest (unittest.TestCase):
 
@@ -99,19 +98,20 @@ class rarslavetest (unittest.TestCase):
                STR1 = "Hello World"
                STR2 = "Goodbye, \nCruel World"
 
-               self.assertEquals (STR1, RarslaveMessage (STR1, True ).__repr__())
-               self.assertEquals (STR1, RarslaveMessage (STR1, False).__repr__())
-               self.assertEquals (STR2, RarslaveMessage (STR2, True ).__repr__())
-               self.assertEquals (STR2, RarslaveMessage (STR2, False).__repr__())
+               self.assertEquals (STR1, RarslaveMessage (STR1).__repr__())
+               self.assertEquals (STR1, RarslaveMessage (STR1, MessageType.Normal).__repr__())
+               self.assertEquals (STR2, RarslaveMessage (STR2, MessageType.Verbose).__repr__())
+               self.assertEquals (STR2, RarslaveMessage (STR2, MessageType.Debug).__repr__())
 
        def testisVerboseMessage (self):
                STR1 = "Hello World"
                STR2 = "Goodbye, \nCruel World"
 
-               self.assertTrue (RarslaveMessage (STR1, False).isVerbose())
-               self.assertTrue (RarslaveMessage (STR2, False).isVerbose())
-               self.assertFalse (RarslaveMessage (STR1, True).isVerbose())
-               self.assertFalse (RarslaveMessage (STR2, True).isVerbose())
+               self.assertTrue (RarslaveMessage (STR1, MessageType.Verbose).isVerbose())
+               self.assertTrue (RarslaveMessage (STR2, MessageType.Verbose).isVerbose())
+               self.assertFalse (RarslaveMessage (STR1).isVerbose())
+               self.assertFalse (RarslaveMessage (STR2, MessageType.Debug).isVerbose())
+               self.assertFalse (RarslaveMessage (STR2, MessageType.Normal).isVerbose())
 
 
 if __name__ == '__main__':