mirror of
https://github.com/moparisthebest/SickRage
synced 2024-08-13 16:53:54 -04:00
0d9fbc1ad7
This version of SickBeard uses both TVDB and TVRage to search and gather it's series data from allowing you to now have access to and download shows that you couldn't before because of being locked into only what TheTVDB had to offer. Also this edition is based off the code we used in our XEM editon so it does come with scene numbering support as well as all the other features our XEM edition has to offer. Please before using this with your existing database (sickbeard.db) please make a backup copy of it and delete any other database files such as cache.db and failed.db if present, we HIGHLY recommend starting out with no database files at all to make this a fresh start but the choice is at your own risk! Enjoy!
139 lines
6.6 KiB
Python
139 lines
6.6 KiB
Python
import unittest
|
|
import test_lib as test
|
|
|
|
import sys, os.path
|
|
sys.path.append(os.path.abspath('..'))
|
|
|
|
from sickbeard import show_name_helpers, scene_exceptions, common, name_cache
|
|
|
|
import sickbeard
|
|
from sickbeard import db
|
|
from sickbeard.databases import cache_db
|
|
from sickbeard.tv import TVShow as Show
|
|
|
|
|
|
class SceneTests(test.SickbeardTestDBCase):
|
|
|
|
def _test_sceneToNormalShowNames(self, name, expected):
|
|
result = show_name_helpers.sceneToNormalShowNames(name)
|
|
self.assertTrue(len(set(expected).intersection(set(result))) == len(expected))
|
|
|
|
dot_result = show_name_helpers.sceneToNormalShowNames(name.replace(' ', '.'))
|
|
dot_expected = [x.replace(' ', '.') for x in expected]
|
|
self.assertTrue(len(set(dot_expected).intersection(set(dot_result))) == len(dot_expected))
|
|
|
|
def _test_allPossibleShowNames(self, name, tvdbid=0, expected=[]):
|
|
s = Show(tvdbid)
|
|
s.name = name
|
|
|
|
result = show_name_helpers.allPossibleShowNames(s)
|
|
self.assertTrue(len(set(expected).intersection(set(result))) == len(expected))
|
|
|
|
def _test_filterBadReleases(self, name, expected):
|
|
result = show_name_helpers.filterBadReleases(name)
|
|
self.assertEqual(result, expected)
|
|
|
|
def _test_isGoodName(self, name, show):
|
|
self.assertTrue(show_name_helpers.isGoodResult(name, show))
|
|
|
|
def test_isGoodName(self):
|
|
listOfcases = [('Show.Name.S01E02.Test-Test', 'Show/Name'),
|
|
('Show.Name.S01E02.Test-Test', 'Show. Name'),
|
|
('Show.Name.S01E02.Test-Test', 'Show- Name'),
|
|
('Show.Name.Part.IV.Test-Test', 'Show Name'),
|
|
('Show.Name.S01.Test-Test', 'Show Name'),
|
|
('Show.Name.E02.Test-Test', 'Show: Name'),
|
|
('Show Name Season 2 Test', 'Show: Name'),
|
|
]
|
|
|
|
for testCase in listOfcases:
|
|
scene_name, show_name = testCase
|
|
s = Show(0)
|
|
s.name = show_name
|
|
self._test_isGoodName(scene_name, s)
|
|
|
|
def test_sceneToNormalShowNames(self):
|
|
self._test_sceneToNormalShowNames('Show Name 2010', ['Show Name 2010', 'Show Name (2010)'])
|
|
self._test_sceneToNormalShowNames('Show Name US', ['Show Name US', 'Show Name (US)'])
|
|
self._test_sceneToNormalShowNames('Show Name AU', ['Show Name AU', 'Show Name (AU)'])
|
|
self._test_sceneToNormalShowNames('Show Name CA', ['Show Name CA', 'Show Name (CA)'])
|
|
self._test_sceneToNormalShowNames('Show and Name', ['Show and Name', 'Show & Name'])
|
|
self._test_sceneToNormalShowNames('Show and Name 2010', ['Show and Name 2010', 'Show & Name 2010', 'Show and Name (2010)', 'Show & Name (2010)'])
|
|
self._test_sceneToNormalShowNames('show name us', ['show name us', 'show name (us)'])
|
|
self._test_sceneToNormalShowNames('Show And Name', ['Show And Name', 'Show & Name'])
|
|
|
|
# failure cases
|
|
self._test_sceneToNormalShowNames('Show Name 90210', ['Show Name 90210'])
|
|
self._test_sceneToNormalShowNames('Show Name YA', ['Show Name YA'])
|
|
|
|
def test_allPossibleShowNames(self):
|
|
#common.sceneExceptions[-1] = ['Exception Test']
|
|
myDB = db.DBConnection("cache.db")
|
|
myDB.action("INSERT INTO scene_exceptions (indexer_id, show_name) VALUES (?,?)", [-1, 'Exception Test'])
|
|
common.countryList['Full Country Name'] = 'FCN'
|
|
|
|
self._test_allPossibleShowNames('Show Name', expected=['Show Name'])
|
|
self._test_allPossibleShowNames('Show Name', -1, expected=['Show Name', 'Exception Test'])
|
|
self._test_allPossibleShowNames('Show Name FCN', expected=['Show Name FCN', 'Show Name (Full Country Name)'])
|
|
self._test_allPossibleShowNames('Show Name (FCN)', expected=['Show Name (FCN)', 'Show Name (Full Country Name)'])
|
|
self._test_allPossibleShowNames('Show Name Full Country Name', expected=['Show Name Full Country Name', 'Show Name (FCN)'])
|
|
self._test_allPossibleShowNames('Show Name (Full Country Name)', expected=['Show Name (Full Country Name)', 'Show Name (FCN)'])
|
|
|
|
def test_filterBadReleases(self):
|
|
self._test_filterBadReleases('Show.S02.German.Stuff-Grp', False)
|
|
self._test_filterBadReleases('Show.S02.Some.Stuff-Core2HD', False)
|
|
self._test_filterBadReleases('Show.S02.Some.German.Stuff-Grp', False)
|
|
self._test_filterBadReleases('German.Show.S02.Some.Stuff-Grp', True)
|
|
self._test_filterBadReleases('Show.S02.This.Is.German', False)
|
|
|
|
|
|
class SceneExceptionTestCase(test.SickbeardTestDBCase):
|
|
|
|
def setUp(self):
|
|
super(SceneExceptionTestCase, self).setUp()
|
|
scene_exceptions.retrieve_exceptions()
|
|
|
|
def test_sceneExceptionsEmpty(self):
|
|
self.assertEqual(scene_exceptions.get_scene_exceptions(0), [])
|
|
|
|
def test_sceneExceptionsBabylon5(self):
|
|
self.assertEqual(sorted(scene_exceptions.get_scene_exceptions(70726)), ['Babylon 5', 'Babylon5'])
|
|
|
|
def test_sceneExceptionByName(self):
|
|
self.assertEqual(scene_exceptions.get_scene_exception_by_name('Babylon5'), 70726)
|
|
self.assertEqual(scene_exceptions.get_scene_exception_by_name('babylon 5'), 70726)
|
|
self.assertEqual(scene_exceptions.get_scene_exception_by_name('Carlos 2010'), 164451)
|
|
|
|
def test_sceneExceptionByNameEmpty(self):
|
|
self.assertEqual(scene_exceptions.get_scene_exception_by_name('nothing useful'), None)
|
|
|
|
def test_sceneExceptionsResetNameCache(self):
|
|
# clear the exceptions
|
|
myDB = db.DBConnection("cache.db")
|
|
myDB.action("DELETE FROM scene_exceptions")
|
|
|
|
# put something in the cache
|
|
name_cache.addNameToCache('Cached Name', 0)
|
|
|
|
# updating should clear the cache so our previously "Cached Name" won't be in there
|
|
scene_exceptions.retrieve_exceptions()
|
|
self.assertEqual(name_cache.retrieveNameFromCache('Cached Name'), None)
|
|
|
|
# put something in the cache
|
|
name_cache.addNameToCache('Cached Name', 0)
|
|
|
|
# updating should not clear the cache this time since our exceptions didn't change
|
|
scene_exceptions.retrieve_exceptions()
|
|
self.assertEqual(name_cache.retrieveNameFromCache('Cached Name'), 0)
|
|
|
|
|
|
if __name__ == '__main__':
|
|
if len(sys.argv) > 1:
|
|
suite = unittest.TestLoader().loadTestsFromName('scene_helpers_tests.SceneExceptionTestCase.test_' + sys.argv[1])
|
|
unittest.TextTestRunner(verbosity=2).run(suite)
|
|
else:
|
|
suite = unittest.TestLoader().loadTestsFromTestCase(SceneTests)
|
|
unittest.TextTestRunner(verbosity=2).run(suite)
|
|
suite = unittest.TestLoader().loadTestsFromTestCase(SceneExceptionTestCase)
|
|
unittest.TextTestRunner(verbosity=2).run(suite)
|