def test_update_0_to_10(self):
tmpfile = unicode(tempfile.mkstemp('.json')[1])
os.remove(tmpfile)
activilylog = ActivityLog.ActivityLog(tmpfile,
verbosity=DEBUG_ACTIVITYLOG)
tvShow = tvShowSchedule.tvShowSchedule(seriesid=321,
autoComplete=False,
verbosity=DEBUG_TVSHOWSCHEDULE)
tvShow.set(season=1,
episode=2,
nextUpdate=datetime.datetime.now(),
info={'seriesname': 'TvShow 2'},
status=0)
self.assertEqual(tvShow['status'], 0)
tvShow.update(downloader=self.downloader,
transferer=self.transferer,
searcher=self.ts,
activitylog=activilylog,
force=True)
self.assertEqual(tvShow['status'], 10)
log = activilylog.get_entry(seriesid=321)
del (log[0]['datetime'])
self.assertEqual(log, [{
'seriesid': 321,
'episode': 2,
'season': 1,
'oldStatus': 0,
'newStatus': 10,
'type': u'info',
'id': 1
}])
os.remove(tmpfile)
def test_update_20_to_30(
self): # Waiting for torrent availability to Download in progress
httpretty.register_uri(
httpretty.POST,
"http://localhost:9091/transmission/rpc",
responses=[
httpretty.Response(
body=open('tests/httpretty_transmission_get_session.json',
'r').read()),
httpretty.Response(
body=open('tests/httpretty_transmission_torrent_get.json',
'r').read()),
httpretty.Response(
body=open('tests/httpretty_transmission_add_torrent.json',
'r').read()),
httpretty.Response(body=open(
'tests/httpretty_transmission_torrent_get_downloading.json',
'r').read()),
])
self.downloader.loadConfig(self.configFileTransmission)
tvShow = tvShowSchedule.tvShowSchedule(seriesid=321,
autoComplete=False,
verbosity=DEBUG_TVSHOWSCHEDULE)
tvShow.set(season=1,
episode=1,
nextUpdate=datetime.datetime.now(),
info={'seriesname': 'TvShow 2'},
status=20)
tvShow.update(downloader=self.downloader,
transferer=self.transferer,
searcher=self.ts,
force=True)
self.assertEqual(tvShow['status'], 30)
def test_update_10_to_30_synology(self): # not broadcasted to Download in progress
httpretty.register_uri(httpretty.GET, "https://localhost:5001/webapi/auth.cgi",responses=[
httpretty.Response(body='{"data":{"sid":"ZexNvOGV.xh7kA4GEN01857"},"success":true}')
])
httpretty.register_uri(httpretty.GET, "https://localhost:5001/webapi/DownloadStation/task.cgi",responses=[
httpretty.Response(body='{"data":{"offeset":0,"tasks":[],"total":0},"success":true}'), # list
httpretty.Response(body='{"data":{"offeset":0,"tasks":[],"total":0},"success":true}'), # list
httpretty.Response(body='''{"data":{"offeset":0,"tasks":[{"id":"dbid_160","size":"0","status":"waiting","status_extra":null,
"title":"tmpvgwQmq.torrent","type":"bt","username":"******"}],"total":1},"success":true}'''), # list
])
httpretty.register_uri(httpretty.POST, "https://localhost:5001/webapi/DownloadStation/task.cgi",responses=[
httpretty.Response(body='{"success":true}'), # create
])
self.downloader.loadConfig(self.configFileSynology)
tvShow = tvShowSchedule.tvShowSchedule(seriesid=321,autoComplete=False,verbosity=DEBUG_TVSHOWSCHEDULE)
tvShow.set(
season=1,
episode=1,
nextUpdate=datetime.datetime.now(),
info={'seriesname':'TvShow 2'},
status=10
)
self.assertEqual(tvShow['status'],10)
tvShow.update(downloader=self.downloader,transferer=self.transferer,searcher=self.ts,force=True)
self.assertEqual(tvShow['status'],30)
def test_get_progression(self):
httpretty.register_uri(
httpretty.POST,
"http://localhost:9091/transmission/rpc",
responses=[
httpretty.Response(
body=open('tests/httpretty_transmission_get_session.json',
'r').read()),
httpretty.Response(body=open(
'tests/httpretty_transmission_torrent_get_downloading.json',
'r').read()),
httpretty.Response(body=open(
'tests/httpretty_transmission_torrent_get_downloading.json',
'r').read()),
])
self.downloader.loadConfig(self.configFileTransmission)
self.transferer.addData(self.configFileTvShowSchedule)
self.transferer.setValue(self.transfererData)
tvShow = tvShowSchedule.tvShowSchedule(seriesid=321,
autoComplete=False,
verbosity=DEBUG_TVSHOWSCHEDULE)
tvShow.set(season=1,
episode=1,
nextUpdate=datetime.datetime.now(),
info={'seriesname': 'TvShow 2'},
status=30,
downloader_id=3)
self.assertEqual(tvShow.get_progression(downloader=self.downloader),
75)
def test_update_10_to_22(
self): # not broadcasted to waiting for torrent availability
for mock_url in httpretty_urls:
httpretty.register_uri(httpretty.GET,
mock_url[0],
body=open(mock_url[1], 'r').read())
httpretty.register_uri(httpretty.POST,
mock_url[0],
body=open(mock_url[1], 'r').read())
httpretty.register_uri(
httpretty.POST,
T411_URL + "/torrents/search/TvShow%201%20S01E02%20720p",
body=open('tests/httpretty_kat_search_not_found.json', 'r').read())
httpretty.register_uri(httpretty.POST,
"https://kat.cr/json.php",
body=open(
'tests/httpretty_kat_search_not_found.json',
'r').read())
tvShow = tvShowSchedule.tvShowSchedule(seriesid=321,
autoComplete=False,
verbosity=DEBUG_TVSHOWSCHEDULE)
tvShow.set(season=1,
episode=1,
nextUpdate=datetime.datetime.now(),
info={'seriesname': 'TvShow 2'},
status=10)
self.assertEqual(tvShow['status'], 10)
tvShow.update(downloader=self.downloader,
transferer=self.transferer,
searcher=self.ts,
force=True)
self.assertEqual(tvShow['status'], 22)
def test_update_10_to_10_none(
self): # not broadcasted to Download in progress
self.downloader.loadConfig(self.configFileNone)
self.downloader['torrentFolder'] = unicode(tempfile.mkdtemp())
tvShow = tvShowSchedule.tvShowSchedule(seriesid=321,
autoComplete=False,
verbosity=DEBUG_TVSHOWSCHEDULE)
tvShow.set(season=1,
episode=1,
nextUpdate=datetime.datetime.now(),
info={'seriesname': 'TvShow 2'},
status=10)
self.assertEqual(tvShow['status'], 10)
tvShow.update(downloader=self.downloader,
transferer=self.transferer,
searcher=self.ts,
force=True)
self.assertEqual(tvShow['status'], 10)
logging.debug("{0}/{1}".format(
self.downloader['torrentFolder'],
'F261769DEEF448D86B23A8A0F2CFDEF0F64113C9.torrent'))
self.assertTrue(
os.path.isfile("{0}/{1}".format(
self.downloader['torrentFolder'],
'F261769DEEF448D86B23A8A0F2CFDEF0F64113C9.torrent')))
def test_update_0_to_90(self): # Added to TvShow Achieved
tvShow = tvShowSchedule.tvShowSchedule(seriesid=123,autoComplete=False,verbosity=DEBUG_TVSHOWSCHEDULE)
tvShow.set(
info={'seriesname':'TvShow 1'}
)
self.assertEqual(tvShow['status'],0)
tvShow.update(downloader=self.downloader,transferer=self.transferer,searcher=self.ts,force=True)
self.assertEqual(tvShow['status'],90)
def test_update_90_to_90(self):
tvShow = tvShowSchedule.tvShowSchedule(seriesid=456,autoComplete=False,verbosity=DEBUG_TVSHOWSCHEDULE)
tvShow.set(
nextUpdate=datetime.datetime.now(),
info={'seriesname':'TvShow 1'},
status=90
)
tvShow.update(downloader=self.downloader,transferer=self.transferer,searcher=self.ts,force=True)
self.assertEqual(tvShow['status'],90)
def test_update_0_to_22(self): # Added to waiting for torrent availability
for mock_url in httpretty_urls:
httpretty.register_uri(httpretty.GET, mock_url[0],body=open(mock_url[1],'r').read())
httpretty.register_uri(httpretty.POST, mock_url[0],body=open(mock_url[1],'r').read())
httpretty.register_uri(httpretty.POST, "https://kat.cr/json.php", body=open('tests/httpretty_kat_search_not_found.json','r').read())
httpretty.register_uri(httpretty.POST, "http://localhost:9091/transmission/rpc",responses=[
httpretty.Response(body=open('tests/httpretty_transmission_get_session.json','r').read()),
httpretty.Response(body=open('tests/httpretty_transmission_torrent_get.json','r').read()),
])
tmpfile = unicode(tempfile.mkstemp('.json')[1])
os.remove(tmpfile)
activilylog = ActivityLog.ActivityLog(tmpfile,verbosity=DEBUG_ACTIVITYLOG)
self.downloader.loadConfig(self.configFileTransmission)
tvShow = tvShowSchedule.tvShowSchedule(seriesid=123,autoComplete=False,verbosity=DEBUG_TVSHOWSCHEDULE)
tvShow.set(
season=1,
episode=1,
nextUpdate=datetime.datetime.now(),
info={'seriesname':'TvShow 1'},
status=0
)
self.assertEqual(tvShow['status'],0)
tvShow.update(
downloader=self.downloader,
transferer=self.transferer,
searcher=self.ts,
activitylog=activilylog,
force=True
)
self.assertEqual(tvShow['status'],22)
log = activilylog.get_entry(seriesid=123)
del(log[0]['datetime'])
del(log[1]['datetime'])
self.assertEqual(
log,
[{
'seriesid': 123,
'episode': 1,
'season': 1,
'oldStatus': 0,
'newStatus': 20,
'type': u'info',
'id': 1
},{
'seriesid': 123,
'episode': 1,
'season': 1,
'oldStatus': 20,
'newStatus': 22,
'type': u'info',
'id': 2
}]
)
os.remove(tmpfile)
def test_delete_banner(self):
tmpdir = unicode(tempfile.mkdtemp())
tmpfile = '{0}/banner_123.jpg'.format(tmpdir)
shutil.copyfile('tests/image1.jpg',tmpfile)
tvShow = tvShowSchedule.tvShowSchedule(seriesid=123,bannerDir=tmpdir,autoComplete=True,verbosity=DEBUG_TVSHOWSCHEDULE)
tvShow.set(season=1,episode=1,status=0,nextUpdate=datetime.datetime.now(),info={'seriesname':'Lost'})
tvShow.deleteBanner()
self.assertFalse(tvShow['info']['banner'])
self.assertFalse(os.path.isfile(tmpfile))
shutil.rmtree(tmpdir)
def test_update_30_to_90(self): # Download in progress to Achieved
files = ['file1.txt', 'file2.tgz', 'foo/file4.txt']
httpretty.register_uri(
httpretty.POST,
"http://localhost:9091/transmission/rpc",
responses=[
httpretty.Response(
body=open('tests/httpretty_transmission_get_session.json',
'r').read()),
httpretty.Response(
body=open('tests/httpretty_transmission_torrent_get.json',
'r').read()),
httpretty.Response(
body=open('tests/httpretty_transmission_torrent_get.json',
'r').read()),
])
if not self.testTransmission:
print "No configuration for Transmission in file {0}, skipping specific tests".format(
self.configFileTransmission)
os.makedirs(self.tmpdir1 + "/foo")
for myFile in files:
with open(self.tmpdir1 + "/" + myFile, 'a'):
os.utime(self.tmpdir1 + "/" + myFile, None)
self.downloader.loadConfig(self.configFileTransmission)
self.transferer.addData(self.configFileTvShowSchedule)
self.transferer.setValue(self.transfererData)
tvShow = tvShowSchedule.tvShowSchedule(seriesid=321,
autoComplete=False,
verbosity=DEBUG_TVSHOWSCHEDULE)
firstaired = datetime.datetime.now() - datetime.timedelta(days=45)
tvShow.set(season=1,
episode=2,
nextUpdate=datetime.datetime.now(),
info={
'seriesname': 'TvShow 2',
'firstaired': firstaired
},
status=30,
downloader_id=3)
for myFile in files:
self.assertTrue(os.path.isfile(self.tmpdir1 + "/" + myFile))
self.assertFalse(os.path.isfile(self.tmpdir2 + "/" + myFile))
tvShow.update(downloader=self.downloader,
transferer=self.transferer,
searcher=self.ts,
force=True)
for myFile in files:
self.assertTrue(os.path.isfile(self.tmpdir1 + "/" + myFile))
os.remove(self.tmpdir1 + "/" + myFile)
self.assertTrue(
os.path.isfile(self.tmpdir2 + "/TvShow 2/season 1/" + myFile))
self.assertFalse('firstaired' in tvShow['info'].keys())
self.assertEqual(tvShow['status'], 90)
def test_creation_without_overview_and_banner(self):
tvShow = tvShowSchedule.tvShowSchedule(seriesid=111,
autoComplete=False,
verbosity=DEBUG_TVSHOWSCHEDULE)
tvShow.set(season=1,
episode=1,
status=0,
nextUpdate=datetime.datetime.now(),
info={'seriesname': 'About Face'})
self.assertIsInstance(tvShow, tvShowSchedule.tvShowSchedule)
self.assertEqual(tvShow['info']['overview'], "")
def test_update_0_to_90(self): # Added to TvShow Achieved
tvShow = tvShowSchedule.tvShowSchedule(seriesid=123,
autoComplete=False,
verbosity=DEBUG_TVSHOWSCHEDULE)
tvShow.set(info={'seriesname': 'TvShow 1'})
self.assertEqual(tvShow['status'], 0)
tvShow.update(downloader=self.downloader,
transferer=self.transferer,
searcher=self.ts,
force=True)
self.assertEqual(tvShow['status'], 90)
def test_update_10_to_30_synology(
self): # not broadcasted to Download in progress
httpretty.register_uri(
httpretty.GET,
"https://localhost:5001/webapi/auth.cgi",
responses=[
httpretty.Response(
body=
'{"data":{"sid":"ZexNvOGV.xh7kA4GEN01857"},"success":true}'
)
])
httpretty.register_uri(
httpretty.GET,
"https://localhost:5001/webapi/DownloadStation/task.cgi",
responses=[
httpretty.Response(
body=
'{"data":{"offeset":0,"tasks":[],"total":0},"success":true}'
), # list
httpretty.Response(
body=
'{"data":{"offeset":0,"tasks":[],"total":0},"success":true}'
), # list
httpretty.Response(
body=
'''{"data":{"offeset":0,"tasks":[{"id":"dbid_160","size":"0","status":"waiting","status_extra":null,
"title":"tmpvgwQmq.torrent","type":"bt","username":"******"}],"total":1},"success":true}'''
), # list
])
httpretty.register_uri(
httpretty.POST,
"https://localhost:5001/webapi/DownloadStation/task.cgi",
responses=[
httpretty.Response(body='{"success":true}'), # create
])
self.downloader.loadConfig(self.configFileSynology)
tvShow = tvShowSchedule.tvShowSchedule(seriesid=321,
autoComplete=False,
verbosity=DEBUG_TVSHOWSCHEDULE)
tvShow.set(season=1,
episode=1,
nextUpdate=datetime.datetime.now(),
info={'seriesname': 'TvShow 2'},
status=10)
self.assertEqual(tvShow['status'], 10)
tvShow.update(downloader=self.downloader,
transferer=self.transferer,
searcher=self.ts,
force=True)
self.assertEqual(tvShow['status'], 30)
def test_creation(self):
tvShow = tvShowSchedule.tvShowSchedule(seriesid=123,
autoComplete=False,
verbosity=DEBUG_TVSHOWSCHEDULE)
tvShow.set(season=1,
episode=1,
status=0,
nextUpdate=datetime.datetime.now(),
info={'seriesname': 'Lost'},
autoComplete=False)
self.assertIsInstance(tvShow, tvShowSchedule.tvShowSchedule)
self.assertEqual(tvShow['status'], 0)
def test_update_90_to_90(self):
tvShow = tvShowSchedule.tvShowSchedule(seriesid=456,
autoComplete=False,
verbosity=DEBUG_TVSHOWSCHEDULE)
tvShow.set(nextUpdate=datetime.datetime.now(),
info={'seriesname': 'TvShow 1'},
status=90)
tvShow.update(downloader=self.downloader,
transferer=self.transferer,
searcher=self.ts,
force=True)
self.assertEqual(tvShow['status'], 90)
def test_creation_with_fresh_banner(self):
tmpdir = unicode(tempfile.mkdtemp())
tmpfile = '{0}/banner_123.jpg'.format(tmpdir)
shutil.copyfile('tests/image1.jpg',tmpfile)
tvShow = tvShowSchedule.tvShowSchedule(seriesid=123,bannerDir=tmpdir,autoComplete=True,verbosity=DEBUG_TVSHOWSCHEDULE)
tvShow.set(season=1,episode=1,status=0,nextUpdate=datetime.datetime.now(),info={'seriesname':'Lost'})
self.assertIsInstance(tvShow,tvShowSchedule.tvShowSchedule)
self.assertTrue(tvShow['info']['banner'])
self.assertTrue(os.path.isfile(tmpfile))
self.assertEqual(LogTestCase.md5(tmpfile),LogTestCase.md5('tests/image1.jpg'))
self.assertFalse(httpretty.has_request())
os.remove(tmpfile)
shutil.rmtree(tmpdir)
def test_update_21_to_22(self): # pushing torrent required -to- wathing torrent on torrent providers
tvShow = tvShowSchedule.tvShowSchedule(seriesid=321,autoComplete=False,verbosity=DEBUG_TVSHOWSCHEDULE)
tvShow.set(
season=1,
episode=1,
nextUpdate=datetime.datetime.now(),
info={'seriesname':'TvShow 2'},
status=21
)
self.assertEqual(tvShow['status'],21)
tvShow.update(downloader=self.downloader,transferer=self.transferer,searcher=self.ts,force=True)
self.assertEqual(tvShow['status'],22)
def test_creation_with_new_banner_dl(self,mock_getmtime):
mock_getmtime.return_value=time.mktime((datetime.datetime.now() - datetime.timedelta(days=45)).timetuple())
tmpdir = unicode(tempfile.mkdtemp())
tmpfile = '{0}/banner_123.jpg'.format(tmpdir)
tvShow = tvShowSchedule.tvShowSchedule(seriesid=123,bannerDir=tmpdir,autoComplete=True,verbosity=DEBUG_TVSHOWSCHEDULE)
tvShow.set(season=1,episode=1,status=0,nextUpdate=datetime.datetime.now(),info={'seriesname':'Lost'})
self.assertIsInstance(tvShow,tvShowSchedule.tvShowSchedule)
self.assertTrue(tvShow['info']['banner'])
self.assertTrue(os.path.isfile(tmpfile))
self.assertEqual(LogTestCase.md5(tmpfile),LogTestCase.md5('tests/image.jpg'))
self.assertTrue(httpretty.has_request())
os.remove(tmpfile)
shutil.rmtree(tmpdir)
def test_update_20_to_22_downloader_connection_error(self): # Waiting for torrent availability to itself caused by connection error
httpretty.register_uri(httpretty.POST, "http://localhost:9091/transmission/rpc",status=500)
self.downloader.loadConfig(self.configFileTransmission)
tvShow = tvShowSchedule.tvShowSchedule(seriesid=321,autoComplete=False,verbosity=DEBUG_TVSHOWSCHEDULE)
tvShow.set(
season=1,
episode=1,
nextUpdate=datetime.datetime.now(),
info={'seriesname':'TvShow 2'},
status=20
)
with self.assertLogs(logger=tvShow.logger,level='ERROR'):
with self.assertLogs(logger=self.downloader.logger,level='ERROR'):
tvShow.update(downloader=self.downloader,transferer=self.transferer,searcher=self.ts,force=True)
self.assertEqual(tvShow['status'],22)
def test_delete_banner(self):
tmpdir = unicode(tempfile.mkdtemp())
tmpfile = '{0}/banner_123.jpg'.format(tmpdir)
shutil.copyfile('tests/image1.jpg', tmpfile)
tvShow = tvShowSchedule.tvShowSchedule(seriesid=123,
bannerDir=tmpdir,
autoComplete=True,
verbosity=DEBUG_TVSHOWSCHEDULE)
tvShow.set(season=1,
episode=1,
status=0,
nextUpdate=datetime.datetime.now(),
info={'seriesname': 'Lost'})
tvShow.deleteBanner()
self.assertFalse(tvShow['info']['banner'])
self.assertFalse(os.path.isfile(tmpfile))
shutil.rmtree(tmpdir)
def test_update_10_to_10_none(self): # not broadcasted to Download in progress
self.downloader.loadConfig(self.configFileNone)
self.downloader['torrentFolder'] = unicode(tempfile.mkdtemp())
tvShow = tvShowSchedule.tvShowSchedule(seriesid=321,autoComplete=False,verbosity=DEBUG_TVSHOWSCHEDULE)
tvShow.set(
season=1,
episode=1,
nextUpdate=datetime.datetime.now(),
info={'seriesname':'TvShow 2'},
status=10
)
self.assertEqual(tvShow['status'],10)
tvShow.update(downloader=self.downloader,transferer=self.transferer,searcher=self.ts,force=True)
self.assertEqual(tvShow['status'],10)
logging.debug("{0}/{1}".format(self.downloader['torrentFolder'],'F261769DEEF448D86B23A8A0F2CFDEF0F64113C9.torrent'))
self.assertTrue(os.path.isfile("{0}/{1}".format(self.downloader['torrentFolder'],'F261769DEEF448D86B23A8A0F2CFDEF0F64113C9.torrent')))
def test_update_21_to_22(
self
): # pushing torrent required -to- wathing torrent on torrent providers
tvShow = tvShowSchedule.tvShowSchedule(seriesid=321,
autoComplete=False,
verbosity=DEBUG_TVSHOWSCHEDULE)
tvShow.set(season=1,
episode=1,
nextUpdate=datetime.datetime.now(),
info={'seriesname': 'TvShow 2'},
status=21)
self.assertEqual(tvShow['status'], 21)
tvShow.update(downloader=self.downloader,
transferer=self.transferer,
searcher=self.ts,
force=True)
self.assertEqual(tvShow['status'], 22)
def test_update_10_to_22(self): # not broadcasted to waiting for torrent availability
for mock_url in httpretty_urls:
httpretty.register_uri(httpretty.GET, mock_url[0],body=open(mock_url[1],'r').read())
httpretty.register_uri(httpretty.POST, mock_url[0],body=open(mock_url[1],'r').read())
httpretty.register_uri(httpretty.POST, T411_URL + "/torrents/search/TvShow%201%20S01E02%20720p", body=open('tests/httpretty_kat_search_not_found.json','r').read())
httpretty.register_uri(httpretty.POST, "https://kat.cr/json.php", body=open('tests/httpretty_kat_search_not_found.json','r').read())
tvShow = tvShowSchedule.tvShowSchedule(seriesid=321,autoComplete=False,verbosity=DEBUG_TVSHOWSCHEDULE)
tvShow.set(
season=1,
episode=1,
nextUpdate=datetime.datetime.now(),
info={'seriesname':'TvShow 2'},
status=10
)
self.assertEqual(tvShow['status'],10)
tvShow.update(downloader=self.downloader,transferer=self.transferer,searcher=self.ts,force=True)
self.assertEqual(tvShow['status'],22)
def test_update_20_to_30(self): # Waiting for torrent availability to Download in progress
httpretty.register_uri(httpretty.POST, "http://localhost:9091/transmission/rpc",responses=[
httpretty.Response(body=open('tests/httpretty_transmission_get_session.json','r').read()),
httpretty.Response(body=open('tests/httpretty_transmission_torrent_get.json','r').read()),
httpretty.Response(body=open('tests/httpretty_transmission_add_torrent.json','r').read()),
httpretty.Response(body=open('tests/httpretty_transmission_torrent_get_downloading.json','r').read()),
])
self.downloader.loadConfig(self.configFileTransmission)
tvShow = tvShowSchedule.tvShowSchedule(seriesid=321,autoComplete=False,verbosity=DEBUG_TVSHOWSCHEDULE)
tvShow.set(
season=1,
episode=1,
nextUpdate=datetime.datetime.now(),
info={'seriesname':'TvShow 2'},
status=20
)
tvShow.update(downloader=self.downloader,transferer=self.transferer,searcher=self.ts,force=True)
self.assertEqual(tvShow['status'],30)
def test_get_progression(self):
httpretty.register_uri(httpretty.POST, "http://localhost:9091/transmission/rpc",responses=[
httpretty.Response(body=open('tests/httpretty_transmission_get_session.json','r').read()),
httpretty.Response(body=open('tests/httpretty_transmission_torrent_get_downloading.json','r').read()),
httpretty.Response(body=open('tests/httpretty_transmission_torrent_get_downloading.json','r').read()),
])
self.downloader.loadConfig(self.configFileTransmission)
self.transferer.addData(self.configFileTvShowSchedule)
self.transferer.setValue(self.transfererData)
tvShow = tvShowSchedule.tvShowSchedule(seriesid=321,autoComplete=False,verbosity=DEBUG_TVSHOWSCHEDULE)
tvShow.set(
season=1,
episode=1,
nextUpdate=datetime.datetime.now(),
info={'seriesname':'TvShow 2'},
status=30,
downloader_id = 3
)
self.assertEqual(tvShow.get_progression(downloader=self.downloader),75)
def test_creation_with_fresh_banner(self):
tmpdir = unicode(tempfile.mkdtemp())
tmpfile = '{0}/banner_123.jpg'.format(tmpdir)
shutil.copyfile('tests/image1.jpg', tmpfile)
tvShow = tvShowSchedule.tvShowSchedule(seriesid=123,
bannerDir=tmpdir,
autoComplete=True,
verbosity=DEBUG_TVSHOWSCHEDULE)
tvShow.set(season=1,
episode=1,
status=0,
nextUpdate=datetime.datetime.now(),
info={'seriesname': 'Lost'})
self.assertIsInstance(tvShow, tvShowSchedule.tvShowSchedule)
self.assertTrue(tvShow['info']['banner'])
self.assertTrue(os.path.isfile(tmpfile))
self.assertEqual(LogTestCase.md5(tmpfile),
LogTestCase.md5('tests/image1.jpg'))
self.assertFalse(httpretty.has_request())
os.remove(tmpfile)
shutil.rmtree(tmpdir)
def addData(self, dataFile):
logging.debug("[tvShowList] Add data {0}".format(unicode(dataFile)))
self.filename = unicode(dataFile)
if self.isDataInitialized(self.filename):
with open(self.filename) as data_file:
fileContent = json.load(data_file)[self.id]
for item in fileContent:
if self.inList(item["seriesid"]):
tvShow = self.getTvShow(item["seriesid"])
tvShow.setValue(item)
else:
tvShow = tvShowSchedule.tvShowSchedule(
item["seriesid"], bannerDir=self.banner_dir, autoComplete=False, verbosity=self.verbosity
)
tvShow.setValue(item)
self.add(tvShow)
for key in [item["seriesid"] for item in self if not self.inList(item["seriesid"])]:
self.delete(key)
else:
self.initDataFile()
setattr(self.root.data, self.id.encode("utf8"), staticJsonFile(self.filename, self.id))
def test_update_10_to_21(self): # not broadcasted to waiting for torrent push
tmpfile = unicode(tempfile.mkstemp('.json')[1])
os.remove(tmpfile)
emptyTS = torrentSearch.torrentSearch(
id="torrentSearch",
dataFile=tmpfile,
verbosity=DEBUG_TORRENT_SEARCH
)
tvShow = tvShowSchedule.tvShowSchedule(seriesid=321,autoComplete=False,verbosity=DEBUG_TVSHOWSCHEDULE)
tvShow.set(
season=1,
episode=1,
nextUpdate=datetime.datetime.now(),
info={'seriesname':'TvShow 2'},
status=10
)
self.assertEqual(tvShow['status'],10)
tvShow.update(downloader=self.downloader,transferer=self.transferer,searcher=emptyTS,force=True)
self.assertEqual(tvShow['status'],21)
os.remove(tmpfile)
def test_update_30_to_10_with_delete_after(self): # Download in progress to Added
files = ['file1.txt','file2.tgz','foo/file4.txt']
httpretty.register_uri(httpretty.POST, "http://localhost:9091/transmission/rpc",responses=[
httpretty.Response(body=open('tests/httpretty_transmission_get_session.json','r').read()),
httpretty.Response(body=open('tests/httpretty_transmission_torrent_get.json','r').read()),
httpretty.Response(body=open('tests/httpretty_transmission_torrent_get.json','r').read()),
])
if not self.testTransmission:
print "No configuration for Transmission in file {0}, skipping specific tests".format(self.configFileTransmission)
os.makedirs(self.tmpdir1+"/foo")
for myFile in files:
with open(self.tmpdir1+"/"+myFile, 'a'):
os.utime(self.tmpdir1+"/"+myFile, None)
self.downloader.loadConfig(self.configFileTransmission)
self.transferer.addData(self.configFileTvShowSchedule)
transfererData = dict(self.transfererData)
transfererData['delete_after'] = True
self.transferer.setValue(transfererData)
tvShow = tvShowSchedule.tvShowSchedule(seriesid=321,autoComplete=False,verbosity=DEBUG_TVSHOWSCHEDULE)
tvShow.set(
season=1,
episode=1,
nextUpdate=datetime.datetime.now(),
info={'seriesname':'TvShow 2'},
status=30,
downloader_id=3
)
for myFile in files:
self.assertTrue(os.path.isfile(self.tmpdir1+"/"+myFile))
self.assertFalse(os.path.isfile(self.tmpdir2+"/"+myFile))
tvShow.update(downloader=self.downloader,transferer=self.transferer,searcher=self.ts,force=True)
for myFile in files:
self.assertTrue(os.path.isfile(self.tmpdir1+"/"+myFile))
os.remove(self.tmpdir1+"/"+myFile)
self.assertTrue(os.path.isfile(self.tmpdir2+"/TvShow 2/season 1/"+myFile))
self.assertEqual(tvShow['status'],10)
def test_update_20_to_22_downloader_connection_error(
self
): # Waiting for torrent availability to itself caused by connection error
httpretty.register_uri(httpretty.POST,
"http://localhost:9091/transmission/rpc",
status=500)
self.downloader.loadConfig(self.configFileTransmission)
tvShow = tvShowSchedule.tvShowSchedule(seriesid=321,
autoComplete=False,
verbosity=DEBUG_TVSHOWSCHEDULE)
tvShow.set(season=1,
episode=1,
nextUpdate=datetime.datetime.now(),
info={'seriesname': 'TvShow 2'},
status=20)
with self.assertLogs(logger=tvShow.logger, level='ERROR'):
with self.assertLogs(logger=self.downloader.logger, level='ERROR'):
tvShow.update(downloader=self.downloader,
transferer=self.transferer,
searcher=self.ts,
force=True)
self.assertEqual(tvShow['status'], 22)
def test_creation_with_new_banner_dl(self, mock_getmtime):
mock_getmtime.return_value = time.mktime(
(datetime.datetime.now() -
datetime.timedelta(days=45)).timetuple())
tmpdir = unicode(tempfile.mkdtemp())
tmpfile = '{0}/banner_123.jpg'.format(tmpdir)
tvShow = tvShowSchedule.tvShowSchedule(seriesid=123,
bannerDir=tmpdir,
autoComplete=True,
verbosity=DEBUG_TVSHOWSCHEDULE)
tvShow.set(season=1,
episode=1,
status=0,
nextUpdate=datetime.datetime.now(),
info={'seriesname': 'Lost'})
self.assertIsInstance(tvShow, tvShowSchedule.tvShowSchedule)
self.assertTrue(tvShow['info']['banner'])
self.assertTrue(os.path.isfile(tmpfile))
self.assertEqual(LogTestCase.md5(tmpfile),
LogTestCase.md5('tests/image.jpg'))
self.assertTrue(httpretty.has_request())
os.remove(tmpfile)
shutil.rmtree(tmpdir)
def test_update_22_to_21(self): # Watching torrent from TP but no TP setup
tmpfile = unicode(tempfile.mkstemp('.json')[1])
os.remove(tmpfile)
emptyTS = torrentSearch.torrentSearch(
id="torrentSearch",
dataFile=tmpfile,
verbosity=DEBUG_TORRENT_SEARCH
)
tvShow = tvShowSchedule.tvShowSchedule(seriesid=321,autoComplete=False,verbosity=DEBUG_TVSHOWSCHEDULE)
tvShow.set(
season=1,
episode=1,
nextUpdate=datetime.datetime.now(),
info={'seriesname':'TvShow 2'},
status=22
)
self.assertEqual(tvShow['status'],22)
with self.assertLogs(logger=emptyTS.logger,level='ERROR'):
tvShow.update(downloader=self.downloader,transferer=self.transferer,searcher=emptyTS,force=True)
self.assertEqual(tvShow['status'],21)
os.remove(tmpfile)
def test_update_0_to_10(self):
tmpfile = unicode(tempfile.mkstemp('.json')[1])
os.remove(tmpfile)
activilylog = ActivityLog.ActivityLog(tmpfile,verbosity=DEBUG_ACTIVITYLOG)
tvShow = tvShowSchedule.tvShowSchedule(seriesid=321,autoComplete=False,verbosity=DEBUG_TVSHOWSCHEDULE)
tvShow.set(
season=1,
episode=2,
nextUpdate=datetime.datetime.now(),
info={'seriesname':'TvShow 2'},
status=0
)
self.assertEqual(tvShow['status'],0)
tvShow.update(
downloader=self.downloader,
transferer=self.transferer,
searcher=self.ts,
activitylog=activilylog,
force=True
)
self.assertEqual(tvShow['status'],10)
log = activilylog.get_entry(seriesid=321)
del(log[0]['datetime'])
self.assertEqual(
log,
[{
'seriesid': 321,
'episode': 2,
'season': 1,
'oldStatus': 0,
'newStatus': 10,
'type': u'info',
'id': 1
}]
)
os.remove(tmpfile)
def test_update_10_to_21(
self): # not broadcasted to waiting for torrent push
tmpfile = unicode(tempfile.mkstemp('.json')[1])
os.remove(tmpfile)
emptyTS = torrentSearch.torrentSearch(id="torrentSearch",
dataFile=tmpfile,
verbosity=DEBUG_TORRENT_SEARCH)
tvShow = tvShowSchedule.tvShowSchedule(seriesid=321,
autoComplete=False,
verbosity=DEBUG_TVSHOWSCHEDULE)
tvShow.set(season=1,
episode=1,
nextUpdate=datetime.datetime.now(),
info={'seriesname': 'TvShow 2'},
status=10)
self.assertEqual(tvShow['status'], 10)
tvShow.update(downloader=self.downloader,
transferer=self.transferer,
searcher=emptyTS,
force=True)
self.assertEqual(tvShow['status'], 21)
os.remove(tmpfile)
def addData(self,dataFile):
logging.debug("[tvShowList] Add data {0}".format(unicode(dataFile)))
self.filename = unicode(dataFile)
if self.isDataInitialized(self.filename):
with open(self.filename) as data_file:
fileContent = json.load(data_file)[self.id]
for item in fileContent:
if self.inList(item['seriesid']):
tvShow = self.getTvShow(item['seriesid'])
tvShow.setValue(item)
else:
tvShow = tvShowSchedule.tvShowSchedule(
item['seriesid'],
bannerDir=self.banner_dir,
autoComplete=False,
verbosity=self.verbosity
)
tvShow.setValue(item)
self.add(tvShow)
for key in [item['seriesid'] for item in self if not self.inList(item['seriesid'])]:
self.delete(key)
else:
self.initDataFile()
setattr(self.root.data,self.id.encode('utf8'),staticJsonFile(self.filename,self.id))
def test_update_22_to_21(self): # Watching torrent from TP but no TP setup
tmpfile = unicode(tempfile.mkstemp('.json')[1])
os.remove(tmpfile)
emptyTS = torrentSearch.torrentSearch(id="torrentSearch",
dataFile=tmpfile,
verbosity=DEBUG_TORRENT_SEARCH)
tvShow = tvShowSchedule.tvShowSchedule(seriesid=321,
autoComplete=False,
verbosity=DEBUG_TVSHOWSCHEDULE)
tvShow.set(season=1,
episode=1,
nextUpdate=datetime.datetime.now(),
info={'seriesname': 'TvShow 2'},
status=22)
self.assertEqual(tvShow['status'], 22)
with self.assertLogs(logger=emptyTS.logger, level='ERROR'):
tvShow.update(downloader=self.downloader,
transferer=self.transferer,
searcher=emptyTS,
force=True)
self.assertEqual(tvShow['status'], 21)
os.remove(tmpfile)
def creation(self): self.l1 = tvShowList.tvShowList(banner_dir=self.tmpdirBanner,verbosity=DEBUG) self.tvShowSchedule1 = tvShowSchedule.tvShowSchedule(seriesid=self.id1, verbosity=DEBUG)
def test_update_30_to_10(self): # Download in progress to Added
files = ['file1.txt','file2.tgz','foo/file4.txt']
httpretty.register_uri(httpretty.POST, "http://localhost:9091/transmission/rpc",responses=[
httpretty.Response(body=open('tests/httpretty_transmission_get_session.json','r').read()),
httpretty.Response(body=open('tests/httpretty_transmission_torrent_get.json','r').read()),
httpretty.Response(body=open('tests/httpretty_transmission_torrent_get.json','r').read()),
])
tmpfile = unicode(tempfile.mkstemp('.json')[1])
os.remove(tmpfile)
activitylog = ActivityLog.ActivityLog(tmpfile,verbosity=DEBUG_ACTIVITYLOG)
os.makedirs(self.tmpdir1+"/foo")
for myFile in files:
with open(self.tmpdir1+"/"+myFile, 'a'):
os.utime(self.tmpdir1+"/"+myFile, None)
self.downloader.loadConfig(self.configFileTransmission)
self.transferer.addData(self.configFileTvShowSchedule)
self.transferer.setValue(self.transfererData)
self.transferer.data['pathPattern'] = "{seriesname}/season {seasonnumber}/episode {episodenumber}"
tvShow = tvShowSchedule.tvShowSchedule(seriesid=321,autoComplete=False,verbosity=DEBUG_TVSHOWSCHEDULE)
tvShow.set(
season=1,
episode=1,
nextUpdate=datetime.datetime.now(),
info={'seriesname':'TvShow 2'},
status=30,
downloader_id=3
)
for myFile in files:
self.assertTrue(os.path.isfile(self.tmpdir1+"/"+myFile))
self.assertFalse(os.path.isfile(self.tmpdir2+"/"+myFile))
tvShow.update(
downloader=self.downloader,
transferer=self.transferer,
searcher=self.ts,
activitylog=activitylog,
force=True
)
for myFile in files:
self.assertTrue(os.path.isfile(self.tmpdir1+"/"+myFile))
os.remove(self.tmpdir1+"/"+myFile)
self.assertTrue(os.path.isfile(self.tmpdir2+"/TvShow 2/season 1/episode 1/"+myFile))
self.assertEqual(tvShow['status'],10)
log = activitylog.get_entry(seriesid=321)
del(log[0]['datetime'])
del(log[1]['datetime'])
del(log[2]['datetime'])
del(log[3]['datetime'])
self.assertEqual(
log,
[{
'seriesid': 321,
'episode': 1,
'season': 1,
'oldStatus': 30,
'newStatus': 35,
'type': u'info',
'id': 1
},{
'seriesid': 321,
'episode': 1,
'season': 1,
'oldStatus': 35,
'newStatus': 39,
'type': u'info',
'id': 2
},{
'seriesid': 321,
'episode': 2,
'season': 1,
'oldStatus': 39,
'newStatus': 0,
'type': u'info',
'id': 3
},{
'seriesid': 321,
'episode': 2,
'season': 1,
'oldStatus': 0,
'newStatus': 10,
'type': u'info',
'id': 4
}]
)
log = activitylog.get_last_downloads()
del(log[0]['datetime'])
self.assertEqual(
log,
[{
'seriesid': 321,
'episode': 1,
'season': 1,
'oldStatus': 35,
'newStatus': 39,
'type': u'info',
'id': 2
}]
)
os.remove(tmpfile)
def test_update_30_to_10(self): # Download in progress to Added
with mock_dt.mock_datetime_now(target1230, datetime):
files = ['file1.txt', 'file2.tgz', 'foo/file4.txt']
httpretty.register_uri(
httpretty.POST,
"http://localhost:9091/transmission/rpc",
responses=[
httpretty.Response(body=open(
'tests/httpretty_transmission_get_session.json',
'r').read()),
httpretty.Response(body=open(
'tests/httpretty_transmission_torrent_get.json',
'r').read()),
httpretty.Response(body=open(
'tests/httpretty_transmission_torrent_get.json',
'r').read()),
])
tmpfile = unicode(tempfile.mkstemp('.json')[1])
os.remove(tmpfile)
activitylog = ActivityLog.ActivityLog(tmpfile,
verbosity=DEBUG_ACTIVITYLOG)
os.makedirs(self.tmpdir1 + "/foo")
for myFile in files:
with open(self.tmpdir1 + "/" + myFile, 'a'):
os.utime(self.tmpdir1 + "/" + myFile, None)
self.downloader.loadConfig(self.configFileTransmission)
self.transferer.addData(self.configFileTvShowSchedule)
self.transferer.setValue(self.transfererData)
self.transferer.data[
'pathPattern'] = "{seriesname}/season {seasonnumber}/episode {episodenumber}"
self.transferer.data['time_restriction'] = True
self.transferer.data['time_restriction_conf'] = {
"start": "10:00:00",
"end": "15:00:00"
}
tvShow = tvShowSchedule.tvShowSchedule(
seriesid=321,
autoComplete=False,
verbosity=DEBUG_TVSHOWSCHEDULE)
tvShow.set(season=1,
episode=1,
nextUpdate=datetime.datetime.now(),
info={'seriesname': 'TvShow 2'},
status=30,
downloader_id=3)
for myFile in files:
self.assertTrue(os.path.isfile(self.tmpdir1 + "/" + myFile))
self.assertFalse(os.path.isfile(self.tmpdir2 + "/" + myFile))
tvShow.update(downloader=self.downloader,
transferer=self.transferer,
searcher=self.ts,
activitylog=activitylog,
force=True)
for myFile in files:
self.assertTrue(os.path.isfile(self.tmpdir1 + "/" + myFile))
os.remove(self.tmpdir1 + "/" + myFile)
self.assertTrue(
os.path.isfile(self.tmpdir2 +
"/TvShow 2/season 1/episode 1/" + myFile))
self.assertEqual(tvShow['status'], 10)
log = activitylog.get_entry(seriesid=321)
del (log[0]['datetime'])
del (log[1]['datetime'])
del (log[2]['datetime'])
del (log[3]['datetime'])
self.assertEqual(log, [{
'seriesid': 321,
'episode': 1,
'season': 1,
'oldStatus': 30,
'newStatus': 35,
'type': u'info',
'id': 1
}, {
'seriesid': 321,
'episode': 1,
'season': 1,
'oldStatus': 35,
'newStatus': 39,
'type': u'info',
'id': 2
}, {
'seriesid': 321,
'episode': 2,
'season': 1,
'oldStatus': 39,
'newStatus': 0,
'type': u'info',
'id': 3
}, {
'seriesid': 321,
'episode': 2,
'season': 1,
'oldStatus': 0,
'newStatus': 10,
'type': u'info',
'id': 4
}])
log = activitylog.get_last_downloads()
del (log[0]['datetime'])
self.assertEqual(log, [{
'seriesid': 321,
'episode': 1,
'season': 1,
'oldStatus': 35,
'newStatus': 39,
'type': u'info',
'id': 2
}])
os.remove(tmpfile)
def test_creation_defaultKeywords(self):
tvShow = tvShowSchedule.tvShowSchedule(seriesid=123,autoComplete=False,keywords=['niouf','niorf'],verbosity=DEBUG_TVSHOWSCHEDULE)
tvShow.set(season=1,episode=1,status=0,nextUpdate=datetime.datetime.now(),info={'seriesname':'Lost'},autoComplete=False)
self.assertIsInstance(tvShow,tvShowSchedule.tvShowSchedule)
self.assertEqual(tvShow['status'],0)
self.assertEqual(tvShow['keywords'],['niouf','niorf'])
def test_creation_without_overview_and_banner(self):
tvShow = tvShowSchedule.tvShowSchedule(seriesid=111,autoComplete=False,verbosity=DEBUG_TVSHOWSCHEDULE)
tvShow.set(season=1,episode=1,status=0,nextUpdate=datetime.datetime.now(),info={'seriesname':'About Face'})
self.assertIsInstance(tvShow,tvShowSchedule.tvShowSchedule)
self.assertEqual(tvShow['info']['overview'],"")
def test_update_0_to_30(self): # Added to Download in progress
httpretty.register_uri(
httpretty.POST,
"http://localhost:9091/transmission/rpc",
responses=[
httpretty.Response(
body=open('tests/httpretty_transmission_get_session.json',
'r').read()),
httpretty.Response(
body=open('tests/httpretty_transmission_torrent_get.json',
'r').read()),
httpretty.Response(
body=open('tests/httpretty_transmission_add_torrent.json',
'r').read()),
httpretty.Response(body=open(
'tests/httpretty_transmission_torrent_get_downloading.json',
'r').read()),
])
tmpfile = unicode(tempfile.mkstemp('.json')[1])
os.remove(tmpfile)
activilylog = ActivityLog.ActivityLog(tmpfile,
verbosity=DEBUG_ACTIVITYLOG)
self.downloader.loadConfig(self.configFileTransmission)
tvShow = tvShowSchedule.tvShowSchedule(seriesid=123,
autoComplete=False,
verbosity=DEBUG_TVSHOWSCHEDULE)
tvShow.set(season=1,
episode=1,
nextUpdate=datetime.datetime.now(),
info={'seriesname': 'TvShow 1'},
status=0)
self.assertEqual(tvShow['status'], 0)
tvShow.update(downloader=self.downloader,
transferer=self.transferer,
searcher=self.ts,
activitylog=activilylog,
force=True)
self.assertEqual(tvShow['status'], 30)
log = activilylog.get_entry(seriesid=123)
del (log[0]['datetime'])
del (log[1]['datetime'])
del (log[2]['datetime'])
self.assertEqual(log, [{
'seriesid': 123,
'episode': 1,
'season': 1,
'oldStatus': 0,
'newStatus': 20,
'type': u'info',
'id': 1
}, {
'seriesid': 123,
'episode': 1,
'season': 1,
'oldStatus': 20,
'newStatus': 22,
'type': u'info',
'id': 2
}, {
'seriesid': 123,
'episode': 1,
'season': 1,
'oldStatus': 22,
'newStatus': 30,
'type': u'info',
'id': 3
}])
os.remove(tmpfile)
