from core import const from core import handle from core import internals from core import spotify_tools from core import youtube_tools from core import convert from core import metadata import spotdl import loader import os loader.load_defaults() internals.filter_path(const.args.folder) raw_song = "Tony's Videos VERY SHORT VIDEO 28.10.2016" def test_youtube_url(): expect_url = 'http://youtube.com/watch?v=qOOcy2-tmbk' url = youtube_tools.generate_youtube_url(raw_song, meta_tags=None) assert url == expect_url def test_youtube_title(): global content global title expect_title = "Tony's Videos VERY SHORT VIDEO 28.10.2016" content = youtube_tools.go_pafy(raw_song, meta_tags=None) title = youtube_tools.get_youtube_title(content) assert title == expect_title def test_check_exists(): expect_check = False # prerequisites for determining filename global file_name file_name = internals.sanitize_title(title) check = spotdl.check_exists(file_name, raw_song, meta_tags=None) assert check == expect_check def test_download(): expect_download = True download = youtube_tools.download_song(file_name, content) assert download == expect_download def test_convert(): # exit code 0 = success expect_converted = 0 global input_song global output_song input_song = file_name + const.args.input_ext output_song = file_name + const.args.output_ext converted = convert.song(input_song, output_song, const.args.folder) assert converted == expect_converted def test_metadata(): expect_metadata = None meta_tags = spotify_tools.generate_metadata(raw_song) if meta_tags: metadata_output = metadata.embed(os.path.join(const.args.folder, output_song), meta_tags) metadata_input = metadata.embed(os.path.join(const.args.folder, input_song), meta_tags) else: metadata_input = None metadata_output = None assert (metadata_output == expect_metadata) and (metadata_input == expect_metadata) def test_check_exists2(): expect_check = True os.remove(os.path.join(const.args.folder, input_song)) check = spotdl.check_exists(file_name, raw_song, meta_tags=None) os.remove(os.path.join(const.args.folder, output_song)) assert check == expect_check