From 6401eca454b01133e9bf5f92bab39fef4387b314 Mon Sep 17 00:00:00 2001 From: ritiek Date: Tue, 25 Jul 2017 01:13:22 +0530 Subject: [PATCH] Optimize and add non-spotify tests --- test/test_simple.py | 72 ++++++++++++++++++++++++ test/{test_single.py => test_spotify.py} | 17 +----- 2 files changed, 74 insertions(+), 15 deletions(-) create mode 100644 test/test_simple.py rename test/{test_single.py => test_spotify.py} (76%) diff --git a/test/test_simple.py b/test/test_simple.py new file mode 100644 index 0000000..44e8dbd --- /dev/null +++ b/test/test_simple.py @@ -0,0 +1,72 @@ +# -*- coding: UTF-8 -*- + +import spotdl +import os + +raw_song = "Tony's Videos VERY SHORT VIDEO 28.10.2016" + +for x in os.listdir(spotdl.args.folder): + os.remove(os.path.join(spotdl.args.folder, x)) + +def test_youtube_url(): + expect_url = 'youtube.com/watch?v=qOOcy2-tmbk' + url = spotdl.generate_youtube_url(raw_song) + assert url == expect_url + + +def test_youtube_title(): + expect_title = "Tony's Videos VERY SHORT VIDEO 28.10.2016" + global content + content = spotdl.go_pafy(raw_song) + global title + title = spotdl.get_youtube_title(content) + assert title == expect_title + +def test_check_exists(): + expect_check = False + # prerequisites for determining filename + file_name = spotdl.misc.sanitize_title(title) + check = spotdl.check_exists(file_name, raw_song, islist=True) + assert check == expect_check + + +def test_download(): + expect_download = True + # prerequisites for determining filename + file_name = spotdl.misc.sanitize_title(title) + download = spotdl.download_song(file_name, content) + assert download == expect_download + + +def test_convert(): + # exit code 0 = success + expect_convert = 0 + # prerequisites for determining filename + file_name = spotdl.misc.sanitize_title(title) + input_song = file_name + spotdl.args.input_ext + output_song = file_name + spotdl.args.output_ext + convert = spotdl.convert.song(input_song, output_song, spotdl.args.folder) + assert convert == expect_convert + + +def test_metadata(): + expect_metadata = None + # prerequisites for determining filename + meta_tags = spotdl.generate_metadata(raw_song) + meta_tags = spotdl.generate_metadata(raw_song) + file_name = spotdl.misc.sanitize_title(title) + output_song = file_name + spotdl.args.output_ext + metadata_output = spotdl.metadata.embed(os.path.join(spotdl.args.folder, output_song), meta_tags) + input_song = file_name + spotdl.args.input_ext + metadata_input = spotdl.metadata.embed(os.path.join(spotdl.args.folder, input_song), meta_tags) + assert (metadata_output == expect_metadata) and (metadata_input == expect_metadata) + + +def test_check_exists2(): + expect_check = True + # prerequisites for determining filename + file_name = spotdl.misc.sanitize_title(title) + input_song = file_name + spotdl.args.input_ext + os.remove(os.path.join(spotdl.args.folder, input_song)) + check = spotdl.check_exists(file_name, raw_song, islist=True) + assert check == expect_check diff --git a/test/test_single.py b/test/test_spotify.py similarity index 76% rename from test/test_single.py rename to test/test_spotify.py index 54e38e0..37119a8 100644 --- a/test/test_single.py +++ b/test/test_spotify.py @@ -11,6 +11,7 @@ for x in os.listdir(spotdl.args.folder): def test_spotify_title(): expect_title = 'David André Østby - Intro' + global meta_tags meta_tags = spotdl.generate_metadata(raw_song) title = spotdl.generate_songname(meta_tags) assert title == expect_title @@ -26,15 +27,14 @@ def youtube_title(): expect_title = 'Intro - David André Østby' content = spotdl.go_pafy(raw_song) title = spotdl.get_youtube_title(content) - print(title) assert title == expect_title def test_check_exists(): expect_check = False # prerequisites for determining filename - meta_tags = spotdl.generate_metadata(raw_song) songname = spotdl.generate_songname(meta_tags) + global file_name file_name = spotdl.misc.sanitize_title(songname) check = spotdl.check_exists(file_name, raw_song, islist=True) assert check == expect_check @@ -44,9 +44,6 @@ def test_download(): expect_download = True # prerequisites for determining filename content = spotdl.go_pafy(raw_song) - meta_tags = spotdl.generate_metadata(raw_song) - songname = spotdl.generate_songname(meta_tags) - file_name = spotdl.misc.sanitize_title(songname) download = spotdl.download_song(file_name, content) assert download == expect_download @@ -55,9 +52,6 @@ def test_convert(): # exit code 0 = success expect_convert = 0 # prerequisites for determining filename - meta_tags = spotdl.generate_metadata(raw_song) - songname = spotdl.generate_songname(meta_tags) - file_name = spotdl.misc.sanitize_title(songname) input_song = file_name + spotdl.args.input_ext output_song = file_name + spotdl.args.output_ext convert = spotdl.convert.song(input_song, output_song, spotdl.args.folder) @@ -67,10 +61,6 @@ def test_convert(): def test_metadata(): expect_metadata = True # prerequisites for determining filename - meta_tags = spotdl.generate_metadata(raw_song) - songname = spotdl.generate_songname(meta_tags) - meta_tags = spotdl.generate_metadata(raw_song) - file_name = spotdl.misc.sanitize_title(songname) output_song = file_name + spotdl.args.output_ext metadata_output = spotdl.metadata.embed(os.path.join(spotdl.args.folder, output_song), meta_tags) input_song = file_name + spotdl.args.input_ext @@ -81,9 +71,6 @@ def test_metadata(): def test_check_exists2(): expect_check = True # prerequisites for determining filename - meta_tags = spotdl.generate_metadata(raw_song) - songname = spotdl.generate_songname(meta_tags) - file_name = spotdl.misc.sanitize_title(songname) input_song = file_name + spotdl.args.input_ext os.remove(os.path.join(spotdl.args.folder, input_song)) check = spotdl.check_exists(file_name, raw_song, islist=True)