Add type detection code.
[rarslave2.git] / rarslave-test.py
1 #!/usr/bin/env python
2 # vim: set ts=4 sts=4 sw=4 textwidth=92 :
3
4 import os, sys, unittest
5 from rarslave import *
6
7 class rarslavetest (unittest.TestCase):
8         def setUp (self):
9                 pass
10
11         def tearDown (self):
12                 pass
13
14         def testGetBasenameNone (self):
15                 QNAME = '[something] some names.txt'
16                 ANAME = '[something] some names.txt'
17
18                 # Tests for an exension that should not be stripped
19                 self.assertEquals (ANAME, get_basename (QNAME))
20
21         def testGetBasenameSingle (self):
22                 QNAME = '[something] some names.par2'
23                 ANAME = '[something] some names'
24
25                 self.assertEquals (ANAME, get_basename (QNAME))
26
27         def testGetBasenameMulti (self):
28                 QNAME = '[a.f.k.] The Melancholy of Haruhi Suzumiya - 13.avi.001'
29                 ANAME = '[a.f.k.] The Melancholy of Haruhi Suzumiya - 13'
30
31                 self.assertEquals (ANAME, get_basename (QNAME))
32
33         def testGetBasenameMulti2 (self):
34                 QNAME = '[AonE-AnY]_Ah_My_Goddess_-_Sorezore_no_Tsubasa_-_13_[WS][E6380C3F].avi.vol00+01.PAR2'
35                 ANAME = '[AonE-AnY]_Ah_My_Goddess_-_Sorezore_no_Tsubasa_-_13_[WS][E6380C3F]'
36
37                 self.assertEquals (ANAME, get_basename (QNAME))
38
39         def testFindLikelyFilesBadDir (self):
40                 DIR = '/fake/dir'
41                 
42                 self.assertRaises (ValueError, find_likely_files, "fake", DIR)
43
44         def testFindAllPar2Files (self):
45                 DIR = '/fake/dir'
46
47                 self.assertRaises (ValueError, find_all_par2_files, DIR)
48
49         def testHasExtension1 (self):
50                 FILE = 'some.file.part01.rar'
51
52                 self.assertTrue (has_extension (FILE, 'rar'))
53                 self.assertTrue (has_extension (FILE, '.rar'))
54                 self.assertTrue (has_extension (FILE, 'part01.rar'))
55                 self.assertTrue (has_extension (FILE, '.part01.rar'))
56
57         def testHasExtension2 (self):
58                 FILE = 'some.file.part01.rar'
59
60                 self.assertFalse (has_extension (FILE, 'zip'))
61                 self.assertFalse (has_extension (FILE, '.zip'))
62                 self.assertFalse (has_extension (FILE, '.part01'))
63                 self.assertFalse (has_extension (FILE, 'part01'))
64
65
66 if __name__ == '__main__':
67         unittest.main ()
68