mirror of
https://github.com/KevinMidboe/spotify-downloader.git
synced 2025-12-29 05:11:21 +00:00
Update tests (#384)
* Update tests * Move comment regarding changing YT URLs to the appropriate assert
This commit is contained in:
committed by
Ritiek Malhotra
parent
6cb12722d0
commit
2725402ab3
@@ -1,11 +1,38 @@
|
||||
from spotdl import internals
|
||||
|
||||
import sys
|
||||
import os
|
||||
import subprocess
|
||||
|
||||
from spotdl import internals
|
||||
|
||||
import pytest
|
||||
|
||||
|
||||
DUPLICATE_TRACKS_TEST_TABLE = [
|
||||
(('https://open.spotify.com/track/2DGa7iaidT5s0qnINlwMjJ',
|
||||
'https://open.spotify.com/track/2DGa7iaidT5s0qnINlwMjJ'),
|
||||
('https://open.spotify.com/track/2DGa7iaidT5s0qnINlwMjJ',)),
|
||||
|
||||
(('https://open.spotify.com/track/2DGa7iaidT5s0qnINlwMjJ',
|
||||
'',
|
||||
'https://open.spotify.com/track/3SipFlNddvL0XNZRLXvdZD'),
|
||||
('https://open.spotify.com/track/2DGa7iaidT5s0qnINlwMjJ',
|
||||
'https://open.spotify.com/track/3SipFlNddvL0XNZRLXvdZD')),
|
||||
|
||||
(('ncs fade',
|
||||
'https://open.spotify.com/track/2DGa7iaidT5s0qnINlwMjJ',
|
||||
'',
|
||||
'ncs fade'),
|
||||
('ncs fade',
|
||||
'https://open.spotify.com/track/2DGa7iaidT5s0qnINlwMjJ')),
|
||||
|
||||
(('ncs spectre ',
|
||||
' https://open.spotify.com/track/2DGa7iaidT5s0qnINlwMjJ',
|
||||
''),
|
||||
('ncs spectre',
|
||||
'https://open.spotify.com/track/2DGa7iaidT5s0qnINlwMjJ'))
|
||||
]
|
||||
|
||||
|
||||
def test_default_music_directory():
|
||||
if sys.platform.startswith('linux'):
|
||||
output = subprocess.check_output(['xdg-user-dir', 'MUSIC'])
|
||||
@@ -82,37 +109,11 @@ class TestGetSeconds:
|
||||
internals.get_sec('02,28,46')
|
||||
|
||||
|
||||
duplicate_tracks_test_table = [
|
||||
(('https://open.spotify.com/track/2DGa7iaidT5s0qnINlwMjJ',
|
||||
'https://open.spotify.com/track/2DGa7iaidT5s0qnINlwMjJ'),
|
||||
('https://open.spotify.com/track/2DGa7iaidT5s0qnINlwMjJ',)),
|
||||
|
||||
(('https://open.spotify.com/track/2DGa7iaidT5s0qnINlwMjJ',
|
||||
'',
|
||||
'https://open.spotify.com/track/3SipFlNddvL0XNZRLXvdZD'),
|
||||
('https://open.spotify.com/track/2DGa7iaidT5s0qnINlwMjJ',
|
||||
'https://open.spotify.com/track/3SipFlNddvL0XNZRLXvdZD')),
|
||||
|
||||
(('ncs fade',
|
||||
'https://open.spotify.com/track/2DGa7iaidT5s0qnINlwMjJ',
|
||||
'',
|
||||
'ncs fade'),
|
||||
('ncs fade',
|
||||
'https://open.spotify.com/track/2DGa7iaidT5s0qnINlwMjJ')),
|
||||
|
||||
(('ncs spectre ',
|
||||
' https://open.spotify.com/track/2DGa7iaidT5s0qnINlwMjJ',
|
||||
''),
|
||||
('ncs spectre',
|
||||
'https://open.spotify.com/track/2DGa7iaidT5s0qnINlwMjJ'))
|
||||
]
|
||||
|
||||
|
||||
@pytest.mark.parametrize("duplicates, expected", duplicate_tracks_test_table)
|
||||
@pytest.mark.parametrize("duplicates, expected", DUPLICATE_TRACKS_TEST_TABLE)
|
||||
def test_get_unique_tracks(tmpdir, duplicates, expected):
|
||||
file_path = os.path.join(str(tmpdir), 'test_duplicates.txt')
|
||||
with open(file_path, 'w') as tin:
|
||||
tin.write('\n'.join(duplicates))
|
||||
with open(file_path, 'w') as f:
|
||||
f.write('\n'.join(duplicates))
|
||||
|
||||
unique_tracks = internals.get_unique_tracks(file_path)
|
||||
assert tuple(unique_tracks) == expected
|
||||
|
||||
Reference in New Issue
Block a user