mirror of
https://github.com/SickGear/SickGear.git
synced 2024-12-11 21:53:37 +00:00
43778d8edd
Add BitMeTV torrent provider. Add TVChaosUK torrent provider. Add HD-Space torrent provider. Add Shazbat torrent provider. Remove Animenzb provider. Change use tvdbid for searching usenet providers. Change consolidate global and per show ignore and require words functions. Change "Require word" title and notes on Config Search page to properly describe its functional logic. Add "Reject Blu-ray M2TS releases" to BTN provider. Add regular expression capability to ignore and require words by starting wordlist with "regex:". Add list shows with custom ignore and require words under the global counterparts on the Search Settings page. Fix failure to search for more than one selected wanted episode.
111 lines
5.6 KiB
Python
111 lines
5.6 KiB
Python
import os.path
|
|
import sys
|
|
import unittest
|
|
|
|
import sickbeard
|
|
from sickbeard import show_name_helpers
|
|
|
|
sys.path.insert(1, os.path.abspath('..'))
|
|
|
|
|
|
class TestCase(unittest.TestCase):
|
|
|
|
cases_pass_wordlist_checks = [
|
|
('[GroupName].Show.Name.-.%02d.[null]', '', '', True),
|
|
|
|
('[GroupName].Show.Name.-.%02d.[ignore]', '', 'required', False),
|
|
('[GroupName].Show.Name.-.%02d.[required]', '', 'required', True),
|
|
('[GroupName].Show.Name.-.%02d.[blahblah]', 'not_ignored', 'GroupName', True),
|
|
('[GroupName].Show.Name.-.%02d.[blahblah]', 'not_ignored', '[GroupName]', True),
|
|
('[GroupName].Show.Name.-.%02d.[blahblah]', 'not_ignored', 'Show.Name', True),
|
|
('[GroupName].Show.Name.-.%02d.[required]', 'not_ignored', 'required', True),
|
|
('[GroupName].Show.Name.-.%02d.[required]', '[not_ignored]', '[required]', True),
|
|
|
|
('[GroupName].Show.Name.-.%02d.[ignore]', '[ignore]', '', False),
|
|
('[GroupName].Show.Name.-.%02d.[required]', '[GroupName]', 'required', False),
|
|
('[GroupName].Show.Name.-.%02d.[required]', 'GroupName', 'required', False),
|
|
('[GroupName].Show.Name.-.%02d.[ignore]', 'ignore', 'GroupName', False),
|
|
('[GroupName].Show.Name.-.%02d.[required]', 'Show.Name', 'required', False),
|
|
|
|
('[GroupName].Show.Name.-.%02d.[ignore]', 'regex: no_ignore', '', True),
|
|
('[GroupName].Show.Name.-.%02d.[480p]', 'ignore', 'regex: \d?\d80p', True),
|
|
('[GroupName].Show.Name.-.%02d.[480p]', 'ignore', 'regex: \[\d?\d80p\]', True),
|
|
('[GroupName].Show.Name.-.%02d.[ignore]', 'regex: ignore', '', False),
|
|
('[GroupName].Show.Name.-.%02d.[ignore]', 'regex: \[ignore\]', '', False),
|
|
('[GroupName].Show.Name.-.%02d.[ignore]', 'regex: ignore', 'required', False),
|
|
|
|
# The following test is True because a boundary is added to each regex not overridden with the prefix param
|
|
('[GroupName].Show.ONEONE.-.%02d.[required]', 'regex: (one(two)?)', '', True),
|
|
('[GroupName].Show.ONETWO.-.%02d.[required]', 'regex: ((one)?two)', 'required', False),
|
|
('[GroupName].Show.TWO.-.%02d.[required]', 'regex: ((one)?two)', 'required', False),
|
|
]
|
|
|
|
cases_contains = [
|
|
('[GroupName].Show.Name.-.%02d.[illegal_regex]', 'regex:??illegal_regex', False),
|
|
('[GroupName].Show.Name.-.%02d.[480p]', 'regex:(480|1080)p', True),
|
|
('[GroupName].Show.Name.-.%02d.[contains]', 'regex:\[contains\]', True),
|
|
('[GroupName].Show.Name.-.%02d.[contains]', '[contains]', True),
|
|
('[GroupName].Show.Name.-.%02d.[contains]', 'contains', True),
|
|
('[GroupName].Show.Name.-.%02d.[contains]', '[not_contains]', False),
|
|
('[GroupName].Show.Name.-.%02d.[null]', '', False)
|
|
]
|
|
|
|
cases_not_contains = [
|
|
('[GroupName].Show.Name.-.%02d.[480p]', 'regex:(480|1080)p', False),
|
|
('[GroupName].Show.Name.-.%02d.[contains]', 'regex:\[contains\]', False),
|
|
('[GroupName].Show.Name.-.%02d.[contains]', '[contains]', False),
|
|
('[GroupName].Show.Name.-.%02d.[contains]', 'contains', False),
|
|
('[GroupName].Show.Name.-.%02d.[not_contains]', '[blah_blah]', True),
|
|
('[GroupName].Show.Name.-.%02d.[null]', '', False)
|
|
]
|
|
|
|
def test_pass_wordlist_checks(self):
|
|
# default:[] or copy in a test case tuple to debug in isolation
|
|
isolated = []
|
|
|
|
test_cases = (self.cases_pass_wordlist_checks, isolated)[len(isolated)]
|
|
for case_num, (name, ignore_list, require_list, expected_result) in enumerate(test_cases):
|
|
name = (name, name % case_num)['%02d' in name]
|
|
sickbeard.IGNORE_WORDS = ignore_list
|
|
sickbeard.REQUIRE_WORDS = require_list
|
|
self.assertEqual(expected_result, show_name_helpers.pass_wordlist_checks(name, False),
|
|
'Expected %s with test: "%s" with ignore: "%s", require: "%s"' %
|
|
(expected_result, name, ignore_list, require_list))
|
|
|
|
def test_contains_any(self):
|
|
# default:[] or copy in a test case tuple to debug in isolation
|
|
isolated = []
|
|
|
|
test_cases = (self.cases_contains, isolated)[len(isolated)]
|
|
for case_num, (name, csv_words, expected_result) in enumerate(test_cases):
|
|
name = (name, name % case_num)['%02d' in name]
|
|
self.assertEqual(expected_result, self.call_contains_any(name, csv_words),
|
|
'Expected %s test: "%s" with csv_words: "%s"' %
|
|
(expected_result, name, csv_words))
|
|
|
|
@staticmethod
|
|
def call_contains_any(name, csv_words):
|
|
re_extras = dict(re_prefix='.*', re_suffix='.*')
|
|
match = show_name_helpers.contains_any(name, csv_words, **re_extras)
|
|
return None is not match and match
|
|
|
|
def test_not_contains_any(self):
|
|
# default:[] or copy in a test case tuple to debug in isolation
|
|
isolated = []
|
|
|
|
test_cases = (self.cases_not_contains, isolated)[len(isolated)]
|
|
for case_num, (name, csv_words, expected_result) in enumerate(test_cases):
|
|
name = (name, name % case_num)['%02d' in name]
|
|
self.assertEqual(expected_result, self.call_not_contains_any(name, csv_words),
|
|
'Expected %s test: "%s" with csv_words:"%s"' %
|
|
(expected_result, name, csv_words))
|
|
|
|
@staticmethod
|
|
def call_not_contains_any(name, csv_words):
|
|
re_extras = dict(re_prefix='.*', re_suffix='.*')
|
|
match = show_name_helpers.not_contains_any(name, csv_words, **re_extras)
|
|
return None is not match and match
|
|
|
|
|
|
if __name__ == '__main__':
|
|
unittest.main()
|