def test_enhance(self):
ep = episode.Episode(self.media)
ep.parse()
with mock.patch.object(ep.api, 'get_series_by_name',
return_value=(None, '')):
self.assertRaises(exc.ShowNotFound, ep.enhance)
with mock.patch.object(ep.api, 'get_series_by_name',
return_value=({}, '')):
with mock.patch.object(ep.api, 'get_series_name',
return_value='Revenge'):
with mock.patch.object(ep.api, 'get_episode_name',
return_value=(['Madness'], None)):
ep.enhance()
self.assertEqual(ep.series_name, 'Revenge')
self.assertEqual(ep.episode_names, ['Madness'])
ep = episode.Episode(self.media)
ep.parse()
with mock.patch.object(ep.api, 'get_series_by_name',
return_value=({}, '')):
with mock.patch.object(ep.api, 'get_series_name',
return_value='Revenge'):
with mock.patch.object(ep.api, 'get_episode_name',
return_value=(None, '')):
self.assertRaises(exc.EpisodeNotFound, ep.enhance)
def test_format_name(self):
ep = episode.Episode(self.media)
ep.series_name = 'Revenge'
ep.season_number = 4
ep.episode_numbers = [12]
ep.episode_names = ['Madness']
self.CONF.set_override('move_files_enabled', False)
ep.format_name()
self.assertEqual(ep.formatted_dirname, os.path.dirname(self.media))
self.assertEqual(ep.formatted_filename,
'Revenge - 04x12 - Madness.mp4')
self.CONF.set_override(
'filename_format_ep',
'S%(seasonnumber)02dE%(episode)s-%(episodename)s%(ext)s')
self.CONF.set_override('directory_name_format',
'%(seriesname)s/Season %(seasonnumber)02d')
self.CONF.set_override('move_files_enabled', True)
with mock.patch('tvrenamer.core.formatter.find_library',
return_value='/tmp'):
ep.format_name()
self.assertEqual(ep.formatted_filename, 'S04E12-Madness.mp4')
self.assertEqual(ep.formatted_dirname, '/tmp/Revenge/Season 04')
def test_call(self):
ep = episode.Episode(self.media)
with mock.patch.object(ep, 'parse'):
with mock.patch.object(ep, 'enhance'):
with mock.patch.object(ep, 'format_name'):
with mock.patch.object(ep, 'rename'):
ep()
self.assertEqual(ep.state, episode.const.DONE)
ep = episode.Episode(self.media)
with mock.patch.object(ep, 'parse', side_effect=OSError):
ep()
self.assertEqual(ep.state, episode.const.FAILED)
ep = episode.Episode(self.media)
with mock.patch.object(ep, 'parse',
side_effect=exc.InvalidFilename):
ep()
self.assertEqual(ep.state, episode.const.FAILED)
def _start(processor_mgr):
LOG.debug('tvrenamer starting...')
outputs = []
for afile in watcher.retrieve_files():
next_ep = episode.Episode(afile)
# process the work
outputs.append(next_ep())
processor_mgr.map_method('process', outputs)
LOG.debug('tvrenamer finished')
def test_save(self):
media = self.create_tempfiles(
[('revenge.s04e12.hdtv.x264-2hd', 'dummy data')], '.mp4')[0]
ep = episode.Episode(media)
_saved_ep_id = self.dbconn.save(ep)
self.assertIsInstance(_saved_ep_id, int)
ep.formatted_filename = 'S04E12.mp4'
_saved_ep_id = self.dbconn.save(ep)
self.assertIsInstance(_saved_ep_id, int)
def test_parse(self):
ep = episode.Episode(self.media)
ep.parse()
self.assertEqual(ep.episode_numbers, [12])
self.assertEqual(ep.series_name, 'revenge')
self.assertEqual(ep.season_number, 4)
ep = episode.Episode(self.media)
with mock.patch.object(episode.parser, 'parse_filename',
return_value=None):
self.assertRaises(exc.InvalidFilename, ep.parse)
ep = episode.Episode(self.media)
with mock.patch.object(episode.parser, 'parse_filename',
return_value={'pattern': ''}):
self.assertRaises(exc.ConfigValueError, ep.parse)
ep = episode.Episode(self.media)
with mock.patch.object(episode.parser, 'parse_filename',
return_value={'pattern': '',
'episode_numbers': []}):
self.assertRaises(exc.ConfigValueError, ep.parse)
def test_str_repr(self):
ep = episode.Episode(self.media)
ep_str = ''
ep_str += self.media
ep_str += ' => ['
ep_str += self.dirname
ep_str += ' '
ep_str += self.filename
ep_str += '|None .mp4] '
ep_str += 'meta: ['
ep_str += ' S E[]] '
ep_str += 'formatted: /'
self.assertEqual(str(ep), ep_str)
self.assertEqual(repr(ep), ep_str)
def test_rename(self):
ep = episode.Episode(self.media)
ep.series_name = 'Revenge'
ep.season_number = 4
ep.episode_numbers = [12]
ep.episode_names = ['Madness']
self.CONF.set_override('move_files_enabled', False)
ep.out_location = os.path.join(self.dirname,
'Revenge - 04x12 - Madness.mp4')
with mock.patch.object(episode.renamer, 'execute') as mock_renamer:
ep.rename()
mock_renamer.assert_called_with(
self.media,
os.path.join(self.dirname, 'Revenge - 04x12 - Madness.mp4'))
self.CONF.set_override('move_files_enabled', True)
ep.out_location = os.path.join(
'/tmp', '.', 'Revenge - 04x12 - Madness.mp4')
with mock.patch.object(episode.renamer, 'execute') as mock_renamer:
ep.rename()
mock_renamer.assert_called_with(
self.media,
os.path.join('/tmp', '.', 'Revenge - 04x12 - Madness.mp4'))
def test_status(self):
ep = episode.Episode(self.media)
self.assertTrue(self.media in ep.status)