dockeruser
/
SickGear
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity
Browse Source

Change test code to follow quotation standards

pull/483/head
Adam 10 years ago
parent
8eee9d8699
commit
03defe0219
10 changed files with 63 additions and 63 deletions
Whitespace
Show all changes Ignore whitespace when comparing lines Ignore changes in amount of whitespace Ignore changes in whitespace at EOL
Unified View
Diff Options
Show Stats Download Patch File Download Diff File
  1. 2
      tests/all_tests.py
  2. 2
      tests/db_tests.py
  3. 4
      tests/migration_tests.py
  4. 4
      tests/name_parser_tests.py
  5. 2
      tests/pp_tests.py
  6. 20
      tests/snatch_tests.py
  7. 30
      tests/test_lib.py
  8. 6
      tests/torrent_tests.py
  9. 42
      tests/tv_tests.py
  10. 14
      tests/xem_tests.py

2
tests/all_tests.py
View File

@ -19,7 +19,7 @@
# along with SickGear. If not, see <http://www.gnu.org/licenses/>. # along with SickGear. If not, see <http://www.gnu.org/licenses/>.
from __future__ import print_function from __future__ import print_function
if __name__ == "__main__": if __name__ == '__main__':
import glob import glob
import unittest import unittest
import sys import sys

2
tests/db_tests.py
View File

@ -29,7 +29,7 @@ class DBBasicTests(test.SickbeardTestDBCase):
self.db = test.db.DBConnection() self.db = test.db.DBConnection()
def test_select(self): def test_select(self):
self.db.select("SELECT * FROM tv_episodes WHERE showid = ? AND location != ''", [0000]) self.db.select('SELECT * FROM tv_episodes WHERE showid = ? AND location != ""', [0000])
self.db.close() self.db.close()
if __name__ == '__main__': if __name__ == '__main__':

4
tests/migration_tests.py
View File

@ -86,7 +86,7 @@ class OldInitialSchema(db.SchemaUpgrade):
logger.log_error_and_exit(u'Your database version (' logger.log_error_and_exit(u'Your database version ('
+ str(cur_db_version) + str(cur_db_version)
+ ') is too old to migrate from what this version of SickGear supports (' + ') is too old to migrate from what this version of SickGear supports ('
+ str(MIN_DB_VERSION) + ').' + "\n" + str(MIN_DB_VERSION) + ').' + '\n'
+ 'Upgrade using a previous version (tag) build 496 to build 501 of SickGear first or remove database file to begin fresh.' + 'Upgrade using a previous version (tag) build 496 to build 501 of SickGear first or remove database file to begin fresh.'
) )
@ -94,7 +94,7 @@ class OldInitialSchema(db.SchemaUpgrade):
logger.log_error_and_exit(u'Your database version (' logger.log_error_and_exit(u'Your database version ('
+ str(cur_db_version) + str(cur_db_version)
+ ') has been incremented past what this version of SickGear supports (' + ') has been incremented past what this version of SickGear supports ('
+ str(MAX_DB_VERSION) + ').' + "\n" + str(MAX_DB_VERSION) + ').' + '\n'
+ 'If you have used other forks of SickGear, your database may be unusable due to their modifications.' + 'If you have used other forks of SickGear, your database may be unusable due to their modifications.'
) )

4
tests/name_parser_tests.py
View File

@ -146,7 +146,7 @@ simple_test_cases = {
'[ACX]Black Lagoon - 02v2 - Mangrove Heaven [SaintDeath] [7481F875]': parser.ParseResult(None, 'Black Lagoon', None, [], 'SaintDeath', 'ACX', None, [2]), '[ACX]Black Lagoon - 02v2 - Mangrove Heaven [SaintDeath] [7481F875]': parser.ParseResult(None, 'Black Lagoon', None, [], 'SaintDeath', 'ACX', None, [2]),
}, },
"anime_standard_round": { 'anime_standard_round': {
'[SGKK] Bleach - 312v2 (1280x720 h264 AAC) [F501C9BE]': parser.ParseResult(None, 'Bleach', None, [], '1280x720 h264 AAC', 'SGKK', None, [312]), '[SGKK] Bleach - 312v2 (1280x720 h264 AAC) [F501C9BE]': parser.ParseResult(None, 'Bleach', None, [], '1280x720 h264 AAC', 'SGKK', None, [312]),
}, },
@ -160,7 +160,7 @@ simple_test_cases = {
'[Ayako] Infinite Stratos - IS - 07v2 [H264][720p][44419534]': parser.ParseResult(None, 'Infinite Stratos', None, [], '720p', 'Ayako', None, [7]), '[Ayako] Infinite Stratos - IS - 07v2 [H264][720p][44419534]': parser.ParseResult(None, 'Infinite Stratos', None, [], '720p', 'Ayako', None, [7]),
'[Ayako-Shikkaku] Oniichan no Koto Nanka Zenzen Suki Janain Dakara ne - 10 [LQ][h264][720p] [8853B21C]': parser.ParseResult(None, 'Oniichan no Koto Nanka Zenzen Suki Janain Dakara ne', None, [], '720p', 'Ayako-Shikkaku', None, [10]), '[Ayako-Shikkaku] Oniichan no Koto Nanka Zenzen Suki Janain Dakara ne - 10 [LQ][h264][720p] [8853B21C]': parser.ParseResult(None, 'Oniichan no Koto Nanka Zenzen Suki Janain Dakara ne', None, [], '720p', 'Ayako-Shikkaku', None, [10]),
'[Tsuki] Fairy Tail - 72 [XviD][C4807111]': parser.ParseResult(None, 'Fairy Tail', None, [], 'C4807111', 'Tsuki', None, [72]), '[Tsuki] Fairy Tail - 72 [XviD][C4807111]': parser.ParseResult(None, 'Fairy Tail', None, [], 'C4807111', 'Tsuki', None, [72]),
'Bubblegum Crisis Tokyo 2040 - 25 [aX] [F4E2E558]': parser.ParseResult(None, 'Bubblegum Crisis Tokyo 2040', None, [], "aX", None, None, [25]), 'Bubblegum Crisis Tokyo 2040 - 25 [aX] [F4E2E558]': parser.ParseResult(None, 'Bubblegum Crisis Tokyo 2040', None, [], 'aX', None, None, [25]),
}, },

2
tests/pp_tests.py
View File

@ -52,7 +52,7 @@ class PPBasicTests(test.SickbeardTestDBCase):
sickbeard.showList = [show] sickbeard.showList = [show]
ep = TVEpisode(show, test.SEASON, test.EPISODE) ep = TVEpisode(show, test.SEASON, test.EPISODE)
ep.name = "some ep name" ep.name = 'some ep name'
ep.saveToDB() ep.saveToDB()
addNameToCache('show name', 3) addNameToCache('show name', 3)

20
tests/snatch_tests.py
View File

@ -30,9 +30,9 @@ import sickbeard
from sickbeard.tv import TVEpisode, TVShow from sickbeard.tv import TVEpisode, TVShow
import sickbeard.common as c import sickbeard.common as c
tests = {"Dexter": {"a": 1, "q": c.HD, "s": 5, "e": [7], "b": 'Dexter.S05E07.720p.BluRay.X264-REWARD', "i": ['Dexter.S05E07.720p.BluRay.X264-REWARD', 'Dexter.S05E07.720p.X264-REWARD']}, tests = {'Dexter': {'a': 1, 'q': c.HD, 's': 5, 'e': [7], 'b': 'Dexter.S05E07.720p.BluRay.X264-REWARD', 'i': ['Dexter.S05E07.720p.BluRay.X264-REWARD', 'Dexter.S05E07.720p.X264-REWARD']},
"House": {"a": 1, "q": c.HD, "s": 4, "e": [5], "b": 'House.4x5.720p.BluRay.X264-REWARD', "i": ['Dexter.S05E04.720p.X264-REWARD', 'House.4x5.720p.BluRay.X264-REWARD']}, 'House': {'a': 1, 'q': c.HD, 's': 4, 'e': [5], 'b': 'House.4x5.720p.BluRay.X264-REWARD', 'i': ['Dexter.S05E04.720p.X264-REWARD', 'House.4x5.720p.BluRay.X264-REWARD']},
"Hells Kitchen": {"a": 1, "q": c.SD, "s": 6, "e": [14, 15], "b": 'Hells.Kitchen.s6e14e15.HDTV.XviD-ASAP', "i": ['Hells.Kitchen.S06E14.HDTV.XviD-ASAP', 'Hells.Kitchen.6x14.HDTV.XviD-ASAP', 'Hells.Kitchen.s6e14e15.HDTV.XviD-ASAP']} 'Hells Kitchen': {'a': 1, 'q': c.SD, 's': 6, 'e': [14, 15], 'b': 'Hells.Kitchen.s6e14e15.HDTV.XviD-ASAP', 'i': ['Hells.Kitchen.S06E14.HDTV.XviD-ASAP', 'Hells.Kitchen.6x14.HDTV.XviD-ASAP', 'Hells.Kitchen.s6e14e15.HDTV.XviD-ASAP']}
} }
@ -69,23 +69,23 @@ def test_generator(tvdbdid, show_name, curData, forceSearch):
def test(self): def test(self):
global searchItems global searchItems
searchItems = curData["i"] searchItems = curData['i']
show = TVShow(1, tvdbdid) show = TVShow(1, tvdbdid)
show.name = show_name show.name = show_name
show.quality = curData["q"] show.quality = curData['q']
show.saveToDB() show.saveToDB()
sickbeard.showList.append(show) sickbeard.showList.append(show)
episode = None episode = None
for epNumber in curData["e"]: for epNumber in curData['e']:
episode = TVEpisode(show, curData["s"], epNumber) episode = TVEpisode(show, curData['s'], epNumber)
episode.status = c.WANTED episode.status = c.WANTED
episode.saveToDB() episode.saveToDB()
bestResult = search.searchProviders(show, episode.season, episode.episode, forceSearch) bestResult = search.searchProviders(show, episode.season, episode.episode, forceSearch)
if not bestResult: if not bestResult:
self.assertEqual(curData["b"], bestResult) self.assertEqual(curData['b'], bestResult)
self.assertEqual(curData["b"], bestResult.name) #first is expected, second is choosen one self.assertEqual(curData['b'], bestResult.name) #first is expected, second is choosen one
return test return test
if __name__ == '__main__': if __name__ == '__main__':
@ -97,7 +97,7 @@ if __name__ == '__main__':
tvdbdid = 1 tvdbdid = 1
for forceSearch in (True, False): for forceSearch in (True, False):
for name, curData in tests.items(): for name, curData in tests.items():
if not curData["a"]: if not curData['a']:
continue continue
fname = name.replace(' ', '_') fname = name.replace(' ', '_')
if forceSearch: if forceSearch:

30
tests/test_lib.py
View File

@ -41,18 +41,18 @@ from sickbeard.databases import cache_db, failed_db
# test globals # test globals
#================= #=================
TESTDIR = os.path.abspath('.') TESTDIR = os.path.abspath('.')
TESTDBNAME = "sickbeard.db" TESTDBNAME = 'sickbeard.db'
TESTCACHEDBNAME = "cache.db" TESTCACHEDBNAME = 'cache.db'
TESTFAILEDDBNAME = "failed.db" TESTFAILEDDBNAME = 'failed.db'
SHOWNAME = u"show name" SHOWNAME = u'show name'
SEASON = 4 SEASON = 4
EPISODE = 2 EPISODE = 2
FILENAME = u"show name - s0" + str(SEASON) + "e0" + str(EPISODE) + ".mkv" FILENAME = u'show name - s0' + str(SEASON) + 'e0' + str(EPISODE) + '.mkv'
FILEDIR = os.path.join(TESTDIR, SHOWNAME) FILEDIR = os.path.join(TESTDIR, SHOWNAME)
FILEPATH = os.path.join(FILEDIR, FILENAME) FILEPATH = os.path.join(FILEDIR, FILENAME)
SHOWDIR = os.path.join(TESTDIR, SHOWNAME + " final") SHOWDIR = os.path.join(TESTDIR, SHOWNAME + ' final')
#sickbeard.logger.sb_log_instance = sickbeard.logger.SBRotatingLogHandler(os.path.join(TESTDIR, 'sickbeard.log'), sickbeard.logger.NUM_LOGS, sickbeard.logger.LOG_SIZE) #sickbeard.logger.sb_log_instance = sickbeard.logger.SBRotatingLogHandler(os.path.join(TESTDIR, 'sickbeard.log'), sickbeard.logger.NUM_LOGS, sickbeard.logger.LOG_SIZE)
sickbeard.logger.SBRotatingLogHandler.log_file = os.path.join(os.path.join(TESTDIR, 'Logs'), 'test_sickbeard.log') sickbeard.logger.SBRotatingLogHandler.log_file = os.path.join(os.path.join(TESTDIR, 'Logs'), 'test_sickbeard.log')
@ -85,7 +85,7 @@ sickbeard.NAMING_SPORTS_PATTERN = ''
sickbeard.NAMING_MULTI_EP = 1 sickbeard.NAMING_MULTI_EP = 1
sickbeard.PROVIDER_ORDER = ["sick_beard_index"] sickbeard.PROVIDER_ORDER = ['sick_beard_index']
sickbeard.newznabProviderList = providers.getNewznabProviderList("'Sick Beard Index|http://lolo.sickbeard.com/|0|5030,5040,5060|0|eponly|0!!!NZBs.org|https://nzbs.org/||5030,5040,5060,5070,5090|0|eponly|0!!!Usenet-Crawler|https://www.usenet-crawler.com/||5030,5040,5060|0|eponly|0'") sickbeard.newznabProviderList = providers.getNewznabProviderList("'Sick Beard Index|http://lolo.sickbeard.com/|0|5030,5040,5060|0|eponly|0!!!NZBs.org|https://nzbs.org/||5030,5040,5060,5070,5090|0|eponly|0!!!Usenet-Crawler|https://www.usenet-crawler.com/||5030,5040,5060|0|eponly|0'")
sickbeard.providerList = providers.makeProviderList() sickbeard.providerList = providers.makeProviderList()
@ -147,20 +147,20 @@ class TestCacheDBConnection(TestDBConnection, object):
# Create the table if it's not already there # Create the table if it's not already there
try: try:
sql = "CREATE TABLE " + providerName + " (name TEXT, season NUMERIC, episodes TEXT, indexerid NUMERIC, url TEXT, time NUMERIC, quality TEXT);" sql = 'CREATE TABLE ' + providerName + ' (name TEXT, season NUMERIC, episodes TEXT, indexerid NUMERIC, url TEXT, time NUMERIC, quality TEXT);'
self.connection.execute(sql) self.connection.execute(sql)
self.connection.commit() self.connection.commit()
except sqlite3.OperationalError as e: except sqlite3.OperationalError as e:
if str(e) != "table " + providerName + " already exists": if str(e) != 'table ' + providerName + ' already exists':
raise raise
# Create the table if it's not already there # Create the table if it's not already there
try: try:
sql = "CREATE TABLE lastUpdate (provider TEXT, time NUMERIC);" sql = 'CREATE TABLE lastUpdate (provider TEXT, time NUMERIC);'
self.connection.execute(sql) self.connection.execute(sql)
self.connection.commit() self.connection.commit()
except sqlite3.OperationalError as e: except sqlite3.OperationalError as e:
if str(e) != "table lastUpdate already exists": if str(e) != 'table lastUpdate already exists':
raise raise
# this will override the normal db connection # this will override the normal db connection
@ -180,10 +180,10 @@ def setUp_test_db():
db.sanityCheckDatabase(db.DBConnection(), mainDB.MainSanityCheck) db.sanityCheckDatabase(db.DBConnection(), mainDB.MainSanityCheck)
# and for cachedb too # and for cachedb too
db.upgradeDatabase(db.DBConnection("cache.db"), cache_db.InitialSchema) db.upgradeDatabase(db.DBConnection('cache.db'), cache_db.InitialSchema)
# and for faileddb too # and for faileddb too
db.upgradeDatabase(db.DBConnection("failed.db"), failed_db.InitialSchema) db.upgradeDatabase(db.DBConnection('failed.db'), failed_db.InitialSchema)
def tearDown_test_db(): def tearDown_test_db():
@ -207,7 +207,7 @@ def setUp_test_episode_file():
try: try:
with open(FILEPATH, 'w') as f: with open(FILEPATH, 'w') as f:
f.write("foo bar") f.write('foo bar')
except EnvironmentError: except EnvironmentError:
print('Unable to set up test episode') print('Unable to set up test episode')
raise raise
@ -237,7 +237,7 @@ if __name__ == '__main__':
dirList = os.listdir(TESTDIR) dirList = os.listdir(TESTDIR)
for fname in dirList: for fname in dirList:
if (fname.find("_test") > 0) and (fname.find("pyc") < 0): if (fname.find('_test') > 0) and (fname.find('pyc') < 0):
print('- ' + fname) print('- ' + fname)
print('==================') print('==================')

6
tests/torrent_tests.py
View File

@ -41,7 +41,7 @@ class TorrentBasicTests(test.SickbeardTestDBCase):
if not html: if not html:
return return
soup = BeautifulSoup(html, features=["html5lib", "permissive"]) soup = BeautifulSoup(html, features=['html5lib', 'permissive'])
torrent_table = soup.find('table', attrs={'class': 'data'}) torrent_table = soup.find('table', attrs={'class': 'data'})
torrent_rows = torrent_table.find_all('tr') if torrent_table else [] torrent_rows = torrent_table.find_all('tr') if torrent_table else []
@ -51,7 +51,7 @@ class TorrentBasicTests(test.SickbeardTestDBCase):
#Continue only if one Release is found #Continue only if one Release is found
if len(torrent_rows) < 2: if len(torrent_rows) < 2:
print(u"The data returned does not contain any torrents") print(u'The data returned does not contain any torrents')
return return
for tr in torrent_rows[1:]: for tr in torrent_rows[1:]:
@ -71,7 +71,7 @@ class TorrentBasicTests(test.SickbeardTestDBCase):
print(title) print(title)
if __name__ == "__main__": if __name__ == '__main__':
print('==================') print('==================')
print('STARTING - XEM Scene Numbering TESTS') print('STARTING - XEM Scene Numbering TESTS')
print('==================') print('==================')

42
tests/tv_tests.py
View File

@ -31,18 +31,18 @@ class TVShowTests(test.SickbeardTestDBCase):
sickbeard.showList = [] sickbeard.showList = []
def test_init_indexerid(self): def test_init_indexerid(self):
show = TVShow(1, 1, "en") show = TVShow(1, 1, 'en')
self.assertEqual(show.indexerid, 1) self.assertEqual(show.indexerid, 1)
def test_change_indexerid(self): def test_change_indexerid(self):
show = TVShow(1, 1, "en") show = TVShow(1, 1, 'en')
show.name = "show name" show.name = 'show name'
show.tvrname = "show name" show.tvrname = 'show name'
show.network = "cbs" show.network = 'cbs'
show.genre = "crime" show.genre = 'crime'
show.runtime = 40 show.runtime = 40
show.status = "5" show.status = '5'
show.airs = "monday" show.airs = 'monday'
show.startyear = 1987 show.startyear = 1987
show.saveToDB() show.saveToDB()
@ -55,11 +55,11 @@ class TVShowTests(test.SickbeardTestDBCase):
self.assertEqual(show.indexerid, 2) self.assertEqual(show.indexerid, 2)
def test_set_name(self): def test_set_name(self):
show = TVShow(1, 1, "en") show = TVShow(1, 1, 'en')
show.name = "newName" show.name = 'newName'
show.saveToDB() show.saveToDB()
show.loadFromDB(skipNFO=True) show.loadFromDB(skipNFO=True)
self.assertEqual(show.name, "newName") self.assertEqual(show.name, 'newName')
class TVEpisodeTests(test.SickbeardTestDBCase): class TVEpisodeTests(test.SickbeardTestDBCase):
@ -69,12 +69,12 @@ class TVEpisodeTests(test.SickbeardTestDBCase):
sickbeard.showList = [] sickbeard.showList = []
def test_init_empty_db(self): def test_init_empty_db(self):
show = TVShow(1, 1, "en") show = TVShow(1, 1, 'en')
ep = TVEpisode(show, 1, 1) ep = TVEpisode(show, 1, 1)
ep.name = "asdasdasdajkaj" ep.name = 'asdasdasdajkaj'
ep.saveToDB() ep.saveToDB()
ep.loadFromDB(1, 1) ep.loadFromDB(1, 1)
self.assertEqual(ep.name, "asdasdasdajkaj") self.assertEqual(ep.name, 'asdasdasdajkaj')
class TVTests(test.SickbeardTestDBCase): class TVTests(test.SickbeardTestDBCase):
@ -84,14 +84,14 @@ class TVTests(test.SickbeardTestDBCase):
sickbeard.showList = [] sickbeard.showList = []
def test_getEpisode(self): def test_getEpisode(self):
show = TVShow(1, 1, "en") show = TVShow(1, 1, 'en')
show.name = "show name" show.name = 'show name'
show.tvrname = "show name" show.tvrname = 'show name'
show.network = "cbs" show.network = 'cbs'
show.genre = "crime" show.genre = 'crime'
show.runtime = 40 show.runtime = 40
show.status = "5" show.status = '5'
show.airs = "monday" show.airs = 'monday'
show.startyear = 1987 show.startyear = 1987
show.saveToDB() show.saveToDB()
sickbeard.showList = [show] sickbeard.showList = [show]

14
tests/xem_tests.py
View File

@ -41,11 +41,11 @@ class XEMBasicTests(test.SickbeardTestDBCase):
""" """
myDB = test.db.DBConnection() myDB = test.db.DBConnection()
sqlResults = myDB.select("SELECT * FROM tv_shows") sqlResults = myDB.select('SELECT * FROM tv_shows')
for sqlShow in sqlResults: for sqlShow in sqlResults:
try: try:
curShow = TVShow(int(sqlShow["indexer"]), int(sqlShow["indexer_id"])) curShow = TVShow(int(sqlShow['indexer']), int(sqlShow['indexer_id']))
sickbeard.showList.append(curShow) sickbeard.showList.append(curShow)
except Exception: except Exception:
pass pass
@ -56,18 +56,18 @@ class XEMBasicTests(test.SickbeardTestDBCase):
""" """
myDB = test.db.DBConnection() myDB = test.db.DBConnection()
sqlResults = myDB.select("SELECT * FROM tv_shows") sqlResults = myDB.select('SELECT * FROM tv_shows')
for sqlShow in sqlResults: for sqlShow in sqlResults:
try: try:
curShow = TVShow(int(sqlShow["indexer"]), int(sqlShow["indexer_id"])) curShow = TVShow(int(sqlShow['indexer']), int(sqlShow['indexer_id']))
sickbeard.showList.append(curShow) sickbeard.showList.append(curShow)
except Exception as e: except Exception as e:
print('There was an error creating the show') print('There was an error creating the show')
def test_formating(self): def test_formating(self):
name = "Game.of.Thrones.S03.720p.HDTV.x264-CtrlHD" name = 'Game.of.Thrones.S03.720p.HDTV.x264-CtrlHD'
release = "Game of Thrones" release = 'Game of Thrones'
# m = re.match('(?P<ep_ab_num>(?>\d{1,3})(?![ip])).+', name) # m = re.match('(?P<ep_ab_num>(?>\d{1,3})(?![ip])).+', name)
@ -80,7 +80,7 @@ class XEMBasicTests(test.SickbeardTestDBCase):
# print(u"Matched " + curRegex + " to " + name) # print(u"Matched " + curRegex + " to " + name)
if __name__ == "__main__": if __name__ == '__main__':
print('==================') print('==================')
print('STARTING - XEM Scene Numbering TESTS') print('STARTING - XEM Scene Numbering TESTS')
print('==================') print('==================')

Write Preview
Loading…
Cancel
Save