def test_handle_all_custom_content_flag(self):
with patch.object(Downloader, "__init__", lambda a, b, c: None):
downloader = Downloader('', '')
downloader.custom_content_temp_dir = CUSTOM_CONTENT_BASE_PATH
downloader.all_custom_content = True
downloader.handle_all_custom_content_flag()
custom_content_names = [cco['name'] for cco in CUSTOM_CONTENT]
assert ordered(custom_content_names) == ordered(downloader.input_files)
def test_verify_flags(self, lf, a, o, i, res, err, capsys):
with patch.object(Downloader, "__init__", lambda x, y, z: None):
downloader = Downloader('', '')
downloader.list_files = lf
downloader.all_custom_content = a
downloader.output_pack_path = o
downloader.input_files = i
answer = downloader.verify_flags()
stdout, _ = capsys.readouterr()
if err:
assert err in stdout
assert answer is res