mirror of
https://github.com/SickGear/SickGear.git
synced 2024-12-29 14:13:38 +00:00
8e8568adb3
Change refactor scene_exceptions.
138 lines
6.1 KiB
Python
138 lines
6.1 KiB
Python
# coding=utf-8
|
|
import warnings
|
|
warnings.filterwarnings('ignore', module=r'.*fuz.*', message='.*Sequence.*')
|
|
warnings.filterwarnings('ignore', module=r'.*connectionpool.*', message='.*certificate verification.*')
|
|
|
|
import unittest
|
|
import test_lib as test
|
|
|
|
import sys
|
|
import os.path
|
|
sys.path.insert(1, os.path.abspath('..'))
|
|
|
|
from sickgear import show_name_helpers, scene_exceptions, common, name_cache
|
|
from sickgear.indexers.indexer_config import TVINFO_TVDB
|
|
|
|
import sickgear
|
|
from sickgear import db
|
|
from sickgear.tv import TVShow
|
|
|
|
|
|
class SceneTests(test.SickbeardTestDBCase):
|
|
|
|
def _test_allPossibleShowNames(self, name, prodid=0, expected=None, season=-1):
|
|
expected = (expected, [])[None is expected]
|
|
s = TVShow(TVINFO_TVDB, prodid)
|
|
s.tvid = TVINFO_TVDB
|
|
s.name = name
|
|
|
|
result = show_name_helpers.all_possible_show_names(s, season=season)
|
|
self.assertTrue(len(set(expected).intersection(set(result))) == len(expected))
|
|
|
|
def _test_pass_wordlist_checks(self, name, expected):
|
|
result = show_name_helpers.pass_wordlist_checks(name)
|
|
self.assertEqual(result, expected)
|
|
|
|
def test_allPossibleShowNames(self):
|
|
# common.sceneExceptions[-1] = ['Exception Test']
|
|
my_db = db.DBConnection()
|
|
my_db.mass_action([
|
|
['INSERT INTO scene_exceptions (indexer, indexer_id, show_name, season) VALUES (?,?,?,?)',
|
|
[TVINFO_TVDB, -1, 'Exception Test', -1]],
|
|
['INSERT INTO scene_exceptions (indexer, indexer_id, show_name, season) VALUES (?,?,?,?)',
|
|
[TVINFO_TVDB, -1, 'Season Test', 19]]
|
|
])
|
|
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)'])
|
|
self._test_allPossibleShowNames('Show Name', -1, expected=['Season Test'], season=19)
|
|
|
|
def test_pass_wordlist_checks(self):
|
|
self._test_pass_wordlist_checks('Show.S02.German.Stuff-Grp', False)
|
|
self._test_pass_wordlist_checks('Show.S02.Some.Stuff-Core2HD', False)
|
|
self._test_pass_wordlist_checks('Show.S02.Some.German.Stuff-Grp', False)
|
|
# self._test_pass_wordlist_checks('German.Show.S02.Some.Stuff-Grp', True)
|
|
self._test_pass_wordlist_checks('Show.S02.This.Is.German', False)
|
|
|
|
|
|
class SceneExceptionTestCase(test.SickbeardTestDBCase):
|
|
|
|
def setUp(self):
|
|
super(SceneExceptionTestCase, self).setUp()
|
|
|
|
sickgear.showList = []
|
|
sickgear.showDict = {}
|
|
for s in [TVShow(TVINFO_TVDB, 79604), TVShow(TVINFO_TVDB, 251085), TVShow(TVINFO_TVDB, 78744)]:
|
|
sickgear.showList.append(s)
|
|
sickgear.showDict[s.sid_int] = s
|
|
sickgear.webserve.Home.make_showlist_unique_names()
|
|
scene_exceptions.ReleaseMap().fetch_exceptions()
|
|
name_cache.build_name_cache()
|
|
|
|
def test_sceneExceptionsEmpty(self):
|
|
self.assertEqual(scene_exceptions.ReleaseMap().get_alt_names(0, 0), [])
|
|
|
|
def test_sceneExceptionsBlack_Lagoon(self):
|
|
self.assertEqual(sorted(
|
|
scene_exceptions.ReleaseMap().get_alt_names(1, 79604)), ['Black-Lagoon'])
|
|
|
|
def test_sceneExceptionByName(self):
|
|
self.assertEqual(scene_exceptions.get_scene_exception_by_name(
|
|
'Black-Lagoon'), [1, 79604, -1])
|
|
self.assertEqual(scene_exceptions.get_scene_exception_by_name(
|
|
'Black Lagoon: The Second Barrage'), [1, 79604, 2])
|
|
self.assertEqual(scene_exceptions.get_scene_exception_by_name(
|
|
'Rokka no Yuusha'), [None, None, None])
|
|
|
|
def test_sceneExceptionByNameAnime(self):
|
|
sickgear.showList = []
|
|
sickgear.showDict = {}
|
|
for s in [TVShow(TVINFO_TVDB, 79604), TVShow(TVINFO_TVDB, 295243)]:
|
|
s.anime = 1
|
|
sickgear.showList.append(s)
|
|
sickgear.showDict[s.sid_int] = s
|
|
scene_exceptions.ReleaseMap().fetch_exceptions()
|
|
name_cache.build_name_cache()
|
|
self.assertEqual(scene_exceptions.get_scene_exception_by_name(
|
|
'ブラック・ラグーン'), [1, 79604, -1])
|
|
self.assertEqual(scene_exceptions.get_scene_exception_by_name(
|
|
'Burakku Ragūn'), [1, 79604, -1])
|
|
self.assertEqual(scene_exceptions.get_scene_exception_by_name(
|
|
'Rokka no Yuusha'), [1, 295243, -1])
|
|
|
|
def test_sceneExceptionByNameEmpty(self):
|
|
self.assertEqual(scene_exceptions.get_scene_exception_by_name(
|
|
'nothing useful'), [None, None, None])
|
|
|
|
def test_sceneExceptionsResetNameCache(self):
|
|
# clear the exceptions
|
|
my_db = db.DBConnection()
|
|
# noinspection SqlConstantCondition
|
|
my_db.action('DELETE FROM scene_exceptions WHERE 1=1')
|
|
|
|
# put something in the cache
|
|
name_cache.add_name_to_cache('Cached Name', prodid=0)
|
|
|
|
# updating should not clear the cache this time since our exceptions didn't change
|
|
scene_exceptions.ReleaseMap().fetch_exceptions()
|
|
self.assertEqual(name_cache.retrieve_name_from_cache('Cached Name'), (0, 0))
|
|
|
|
|
|
if '__main__' == __name__:
|
|
if 1 < len(sys.argv):
|
|
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)
|