def test_list_associated_files(p, create_structure, monkeypatch):
"""Run the test."""
# Given
test_path = create_structure(p['path'], structure=p['structure'])
path = os.path.join(test_path, os.path.normcase(p['path']))
file_path = os.path.join(path, p['structure'][0])
expected_associated_files = p['expected_associated_files']
subtitles_only = p.get('subtitles_only', False)
subfolders = p.get('subfolders', False)
refine = p.get('refine', False)
monkeypatch.setattr(app, 'ALLOWED_EXTENSIONS', p['allowed_extensions'])
monkeypatch.setattr(app, 'MOVE_ASSOCIATED_FILES', 1)
processor = PostProcessor(file_path)
processor._rar_basename = _rar_basename
# When
found_associated_files = processor.list_associated_files(
file_path,
subfolders=subfolders,
subtitles_only=subtitles_only,
refine=refine)
associated_files_basenames = [
os.path.basename(i) for i in found_associated_files
]
# Then
assert set(associated_files_basenames) == set(expected_associated_files)
def test_list_associated_files(p, create_structure, monkeypatch):
"""Run the test."""
# Given
test_path = create_structure(p['path'], structure=p['structure'])
path = os.path.join(test_path, os.path.normcase(p['path']))
file_path = os.path.join(path, p['structure'][0])
expected_associated_files = p['expected_associated_files']
subtitles_only = p.get('subtitles_only', False)
subfolders = p.get('subfolders', False)
refine = p.get('refine', False)
monkeypatch.setattr(app, 'ALLOWED_EXTENSIONS', p['allowed_extensions'])
monkeypatch.setattr(app, 'MOVE_ASSOCIATED_FILES', 1)
processor = PostProcessor(file_path)
processor._rar_basename = _rar_basename
# When
found_associated_files = processor.list_associated_files(file_path, subfolders=subfolders,
subtitles_only=subtitles_only, refine=refine)
associated_files_basenames = [os.path.basename(i) for i in found_associated_files]
# Then
assert set(associated_files_basenames) == set(expected_associated_files)
def test_process(self):
show = TVShow(1, 3)
show.name = test.SHOW_NAME
show.location = test.SHOW_DIR
show.save_to_db()
app.showList = [show]
episode = TVEpisode(show, test.SEASON, test.EPISODE)
episode.name = "some episode name"
episode.save_to_db()
addNameToCache('show name', 3)
app.PROCESS_METHOD = 'move'
post_processor = PostProcessor(test.FILE_PATH)
self.assertTrue(post_processor.process())
def test_parse_info(p, monkeypatch, parse_method):
"""Run the test."""
# Given
monkeypatch.setattr(NameParser, 'parse', parse_method)
sut = PostProcessor(file_path=p['file_path'], nzb_name=p['nzb_name'])
# When
show, season, episodes, quality, version, airdate = sut._parse_info()
# Then
assert show is not None
assert p['expected'] == {
'show': show.name,
'season': season,
'episodes': episodes,
'quality': quality,
'version': version,
}
def test_parse_info(p, monkeypatch, parse_method):
"""Run the test."""
# Given
monkeypatch.setattr(NameParser, 'parse', parse_method)
sut = PostProcessor(file_path=p['file_path'], nzb_name=p['nzb_name'])
# When
show, season, episodes, quality, version, airdate = sut._parse_info()
# Then
assert show is not None
assert p['expected'] == {
'show': show.name,
'season': season,
'episodes': episodes,
'quality': quality,
'version': version,
}
def subtitleMissedPP(self):
t = PageTemplate(rh=self, filename='manage_subtitleMissedPP.mako')
app.RELEASES_IN_PP = []
for root, _, files in os.walk(app.TV_DOWNLOAD_DIR, topdown=False):
# Skip folders that are being used for unpacking
if u'_UNPACK' in root.upper():
continue
for filename in sorted(files):
if not is_media_file(filename):
continue
video_path = os.path.join(root, filename)
video_date = datetime.datetime.fromtimestamp(os.stat(video_path).st_ctime)
video_age = datetime.datetime.today() - video_date
tv_episode = Episode.from_filepath(video_path)
if not tv_episode:
logger.log(u"Filename '{0}' cannot be parsed to an episode".format(filename), logger.DEBUG)
continue
ep_status = tv_episode.status
if ep_status in (SNATCHED, SNATCHED_PROPER, SNATCHED_BEST):
status = 'snatched'
elif ep_status == DOWNLOADED:
status = 'downloaded'
else:
continue
if not tv_episode.series.subtitles:
continue
related_files = PostProcessor(video_path).list_associated_files(video_path, subtitles_only=True)
if related_files:
continue
age_hours = divmod(video_age.seconds, 3600)[0]
age_minutes = divmod(video_age.seconds, 60)[0]
if video_age.days > 0:
age_unit = 'd'
age_value = video_age.days
elif age_hours > 0:
age_unit = 'h'
age_value = age_hours
else:
age_unit = 'm'
age_value = age_minutes
app.RELEASES_IN_PP.append({'release': video_path, 'seriesid': tv_episode.series.indexerid,
'show_name': tv_episode.series.name, 'season': tv_episode.season,
'episode': tv_episode.episode, 'status': status, 'age': age_value,
'age_unit': age_unit, 'date': video_date,
'indexername': tv_episode.series.indexer_name})
return t.render(releases_in_pp=app.RELEASES_IN_PP,
controller='manage', action='subtitleMissedPP')
def test__process_postponed(monkeypatch, p, create_structure):
"""Run the test."""
# Given
test_path = create_structure(p['path'], structure=p['structure'])
path = os.path.join(test_path, os.path.normcase(p['path']))
video_path = os.path.join(path, p['video'])
processor = PostProcessor(path)
sut = ProcessResult(path)
# Overwrite internal method
sut.subtitles_enabled = lambda path, resource_name: p['subtitles_enabled']
# When
result = sut._process_postponed(processor, video_path, p['video'], p['ignore_subs'])
# Then
assert p['expected'] == result
def test_should_process(p):
"""Run the test."""
# Given
current_quality = p['cur_quality']
new_quality = p['new_quality']
allowed_qualities = p['allowed_qualities']
preferred_qualities = p['preferred_qualities']
expected = p['expected']
# When
replace, msg = PostProcessor._should_process(current_quality, new_quality, allowed_qualities, preferred_qualities)
actual = replace
# Then
if expected != actual:
print(msg)
assert expected == actual
def test_rename_associated_file(p, create_dir, monkeypatch):
"""Test rename_associated_file."""
# Given
new_path = p['new_path']
new_basename = p.get('new_basename')
filepath = p['filepath']
monkeypatch.setattr(app, 'NFO_RENAME', p.get('nfo_rename', 1))
if p.get('subtitles'):
# Workaround for absolute subtitles directory
if p['subtitles']['absolute']:
subs_dir = create_dir(p['subtitles']['dir'])
monkeypatch.setattr(app, 'SUBTITLES_DIR', subs_dir)
p['expected'] = os.path.join(subs_dir, p['expected'])
else:
monkeypatch.setattr(app, 'SUBTITLES_DIR', p['subtitles']['dir'])
# When
result = Sut.rename_associated_file(new_path, new_basename, filepath)
# Then
assert os.path.normcase(result) == os.path.normcase(p['expected'])
def test_rename_associated_file(p, create_dir, monkeypatch):
"""Test rename_associated_file."""
# Given
new_path = p['new_path']
new_basename = p.get('new_basename')
filepath = p['filepath']
monkeypatch.setattr(app, 'NFO_RENAME', p.get('nfo_rename', 1))
if p.get('subtitles'):
# Workaround for absolute subtitles directory
if p['subtitles']['absolute']:
subs_dir = create_dir(p['subtitles']['dir'])
monkeypatch.setattr(app, 'SUBTITLES_DIR', subs_dir)
p['expected'] = os.path.join(subs_dir, p['expected'])
else:
monkeypatch.setattr(app, 'SUBTITLES_DIR', p['subtitles']['dir'])
# When
result = Sut.rename_associated_file(new_path, new_basename, filepath)
# Then
assert os.path.normcase(result) == os.path.normcase(p['expected'])
def setUp(self):
"""Set up tests."""
self.post_processor = PostProcessor(test.FILE_PATH)