Merge with add-logger
[rarslave2.git] / rarslave-test.py
index d8336b1..e2733ba 100644 (file)
@@ -3,6 +3,8 @@
 
 import os, sys, unittest
 from rarslave import *
+from RarslaveLogger import RarslaveLogger
+from RarslaveLogger import RarslaveMessage
 
 class rarslavetest (unittest.TestCase):
 
@@ -19,10 +21,10 @@ class rarslavetest (unittest.TestCase):
                return True
 
        def setUp (self):
-               pass
+               self.logger = RarslaveLogger ()
 
        def tearDown (self):
-               pass
+               self.logger = None
 
        def testGetBasenameNone (self):
                QNAME = '[something] some names.txt'
@@ -91,6 +93,48 @@ class rarslavetest (unittest.TestCase):
 
                self.assertListEquals (find_deleteable_files (FILESN + FILESY), FILESY)
 
+       ### RarslaveMessage tests
+
+       def testRepr (self):
+               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__())
+
+       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))
+
 
 if __name__ == '__main__':
        unittest.main ()