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):
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())
-
- ### RarslaveLogger tests
-
- def testgetUniqueID (self):
- generated = []
-
- for i in xrange(100):
- id = self.logger.getUniqueID ()
- if id in generated:
- self.fail ("Already generated ID was generated again")
-
- generated.append (id)
-
- def testisValidID (self):
- id1 = self.logger.getUniqueID ()
- id2 = self.logger.getUniqueID ()
-
- self.assertFalse (self.logger.isValidID (-100))
- self.assertFalse (self.logger.isValidID (1000))
-
- self.assertTrue (self.logger.isValidID (id1))
- self.assertTrue (self.logger.isValidID (id2))
+ 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__':