Initial work on extraction.

[rarslave2.git] / rarslave-test.py

diff --git a/rarslave-test.py b/rarslave-test.py
index 21aadb8..64d04f6 100644 (file)
--- 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
 
@@ -46,6 +59,38 @@ class rarslavetest (unittest.TestCase):
 
                self.assertRaises (ValueError, find_all_par2_files, DIR)
 
+       def testHasExtension1 (self):
+               FILE = 'some.file.part01.rar'
+
+               self.assertTrue (has_extension (FILE, 'rar'))
+               self.assertTrue (has_extension (FILE, '.rar'))
+               self.assertTrue (has_extension (FILE, 'part01.rar'))
+               self.assertTrue (has_extension (FILE, '.part01.rar'))
+
+       def testHasExtension2 (self):
+               FILE = 'some.file.part01.rar'
+
+               self.assertFalse (has_extension (FILE, 'zip'))
+               self.assertFalse (has_extension (FILE, '.zip'))
+               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 ()