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

diff --git a/rarslave-test.py b/rarslave-test.py
index c57075a..64d04f6 100644
--- a/rarslave-test.py
+++ b/rarslave-test.py
@@ -5,6 +5,19 @@ import os, sys, unittest
 from rarslave import *
 
 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
 
@@ -62,6 +75,22 @@ class rarslavetest (unittest.TestCase):
 		self.assertFalse (has_extension (FILE, '.part01'))
 		self.assertFalse (has_extension (FILE, 'part01'))
 
+	def testIsNewRar (self):
+		DIR = '/home/irasnyd/downloads/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)
+
 
 if __name__ == '__main__':
 	unittest.main ()