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