X-Git-Url: https://www.irasnyder.com/gitweb/?p=rarslave2.git;a=blobdiff_plain;f=rarslave-test.py;h=e2733baa64becaee4fee862000ef771df1399b0e;hp=64d04f6ba6095c775b98d9cd1bd0411179d48767;hb=61bead019809659b0a98f2309e19b5459139659a;hpb=5ec7a887a528ecf80185ef26a2d5b754cb05ef18

diff --git a/rarslave-test.py b/rarslave-test.py
index 64d04f6..e2733ba 100644
--- a/rarslave-test.py
+++ b/rarslave-test.py
@@ -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'
@@ -51,7 +53,7 @@ class rarslavetest (unittest.TestCase):
 
 	def testFindLikelyFilesBadDir (self):
 		DIR = '/fake/dir'
-		
+
 		self.assertRaises (ValueError, find_likely_files, "fake", DIR)
 
 	def testFindAllPar2Files (self):
@@ -76,7 +78,7 @@ class rarslavetest (unittest.TestCase):
 		self.assertFalse (has_extension (FILE, 'part01'))
 
 	def testIsNewRar (self):
-		DIR = '/home/irasnyd/downloads/test_material/01/'
+		DIR = os.getcwd() + '/test_material/01/'
 
 		self.assertTrue (is_newrar (os.listdir (DIR)))
 
@@ -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 ()