X-Git-Url: https://www.irasnyder.com/gitweb/?p=rarslave2.git;a=blobdiff_plain;f=rarslave-test.py;h=57906816de75ddd1692f0bf5e92a1dc8d92e4e53;hp=21aadb8fc55903ca539b4ca1b17824cc4ba7bca2;hb=a41f0878f0745f3026f4f610eeb52cba42f03da1;hpb=aebb7adb2c800b1efc7c7d1877a16609cfe76b65

diff --git a/rarslave-test.py b/rarslave-test.py
index 21aadb8..5790681 100644
--- a/rarslave-test.py
+++ b/rarslave-test.py
@@ -3,13 +3,27 @@
 
 import os, sys, unittest
 from rarslave import *
+from RarslaveLogger import RarslaveLogger,RarslaveMessage,MessageType
 
 class rarslavetest (unittest.TestCase):
+
+	def assertListEquals (self, l1, l2):
+		if l1 is l2:
+			return True
+
+		self.assertEquals (len(l1), len(l2))
+
+		for e in l1:
+			if e not in l2:
+				self.fail ("Element missing from list")
+
+		return True
+
 	def setUp (self):
-		pass
+		self.logger = RarslaveLogger ()
 
 	def tearDown (self):
-		pass
+		self.logger = None
 
 	def testGetBasenameNone (self):
 		QNAME = '[something] some names.txt'
@@ -38,14 +52,51 @@ class rarslavetest (unittest.TestCase):
 
 	def testFindLikelyFilesBadDir (self):
 		DIR = '/fake/dir'
-		
-		self.assertRaises (ValueError, find_likely_files, "fake", DIR)
+
+		self.assertRaises (AssertionError, find_likely_files, "fake", DIR)
 
 	def testFindAllPar2Files (self):
 		DIR = '/fake/dir'
 
 		self.assertRaises (ValueError, find_all_par2_files, DIR)
 
+	def testIsNewRar (self):
+		DIR = os.getcwd() + '/test_material/01/'
+
+		self.assertTrue (is_newrar (os.listdir (DIR)))
+
+	def testDeletableFiles1 (self):
+		FILES = ['test.part%d.rar' % n for n in xrange(10)]
+
+		self.assertListEquals (find_deleteable_files (FILES), FILES)
+
+	def testDeletableFiles2 (self):
+		FILESN = ['%d.mp3' % n for n in xrange(20)]
+		FILESY = ['%d.zip' % n for n in xrange(5)]
+
+		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).__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, 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__':
 	unittest.main ()