Add more basic tests

spotdl.py

@@ -105,7 +105,7 @@ def go_pafy(raw_song):
         return pafy.new(track_url)
 
 # title of the YouTube video
-def get_YouTube_title(content, number=None):
+def get_youtube_title(content, number=None):
     title = misc.fix_encoding(content.title)
     if number is None:
         return title
@@ -160,7 +160,7 @@ def download_song(content):
         return True
 
 # check if input song already exists in Music folder
-def check_exists(music_file, raw_song, islist):
+def check_exists(music_file, raw_song, islist=True):
     files = os.listdir("Music")
     for file in files:
         if file.endswith(".temp"):
@@ -187,6 +187,7 @@ def check_exists(music_file, raw_song, islist):
                     return False
                 else:
                     return True
+    return False
 
 # download songs from list
 def grab_list(file):
@@ -240,7 +241,7 @@ def grab_single(raw_song, number=None):
         return
     # print "[number]. [artist] - [song]" if downloading from list
     # otherwise print "[artist] - [song]"
-    print(get_YouTube_title(content, number))
+    print(get_youtube_title(content, number))
     # generate file name of the song to download
     music_file = misc.generate_filename(content.title)
     music_file = misc.fix_decoding(music_file)

test/test_basic.py

@@ -2,14 +2,34 @@
 
 import spotdl
 
-raw_song = "http://open.spotify.com/track/2e0jnySVkYF1pvBlpoNX1Y"
+raw_song = 'http://open.spotify.com/track/2e0jnySVkYF1pvBlpoNX1Y'
 
-def test_spotify():
-    expected_title = "David André Østby - Tilbake (SAEVIK Remix)"
+def test_spotify_title():
+    expect_title = 'David André Østby - Tilbake (SAEVIK Remix)'
     title = spotdl.generate_songname(raw_song)
-    assert title == expected_title
+    assert title == expect_title
+
+def test_youtube_url():
+    expect_url = 'youtube.com/watch?v=zkD4smbefbc'
+    url = spotdl.generate_youtube_url(raw_song)
+    assert url == expect_url
 
 def test_youtube_title():
-    expected_url = "youtube.com/watch?v=zkD4smbefbc"
-    url = spotdl.generate_youtube_url(raw_song)
-    assert url == expected_url
+    expect_title = 'Tilbake (SAEVIK Remix) [Feat. Marie Hognestad] - David André Østby'
+    content = spotdl.go_pafy(raw_song)
+    title = spotdl.get_youtube_title(content)
+    assert title == expect_title
+
+def test_check_exists():
+    expect_check = False
+    content = spotdl.go_pafy(raw_song)
+    music_file = spotdl.misc.generate_filename(content.title)
+    music_file = spotdl.misc.fix_decoding(music_file)
+    check = spotdl.check_exists(music_file, raw_song)
+    assert check == expect_check
+
+def test_download():
+    expect_download = True
+    content = spotdl.go_pafy(raw_song)
+    download = download_song(content)
+    assert download == expect_download