OSSForks/scdl
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity
master
scdl/tests/test_track.py
Jason Butler 43890f9336
Added '--add-description' option (#459) 
* Added '--add-description' option

This new option will add the track description to a txt file (using the same filename as the music file).
Some music players will display this text whilst the track is playing - great for tracklists on DJ mixes and podcasts.

* Added subfolder switch

* Revert: Added subfolder switch

Accidently pushed to my branch by mistake - changes reverted

* Fix lint errors

* Add test for description file

* Fix tests for pull requests

---------

Co-authored-by: 7x11x13 <x7x11x13@gmail.com>
2024-07-26 15:48:11 -04:00

405 lines
11 KiB
Python
Raw Permalink Blame History

import math
import os
from pathlib import Path
import pytest
from tests.utils import assert_not_track, assert_track, call_scdl_with_auth
@pytest.mark.skipif(not os.getenv("AUTH_TOKEN"), reason="No auth token specified")
def test_original_download(tmp_path: Path) -> None:
os.chdir(tmp_path)
r = call_scdl_with_auth(
"-l",
"https://soundcloud.com/57v/original", # thanks saves for hosting
"--name-format",
"track",
)
assert r.returncode == 0
assert_track(tmp_path, "track.wav", "copy", "saves", None)
def test_original_to_stdout(tmp_path: Path) -> None:
os.chdir(tmp_path)
r = call_scdl_with_auth(
"-l",
"https://soundcloud.com/57v/original",
"--name-format",
"-",
encoding=None,
)
assert r.returncode == 0
with open("track.wav", "wb") as f:
assert isinstance(r.stdout, bytes)
f.write(r.stdout)
assert_track(tmp_path, "track.wav", "copy", "saves", None)
def test_mp3_to_stdout(tmp_path: Path) -> None:
os.chdir(tmp_path)
r = call_scdl_with_auth(
"-l",
"https://soundcloud.com/one-thousand-and-one/test-track",
"--onlymp3",
"--name-format",
"-",
encoding=None,
)
assert r.returncode == 0
with open("track.mp3", "wb") as f:
assert isinstance(r.stdout, bytes)
f.write(r.stdout)
assert_track(tmp_path, "track.mp3")
@pytest.mark.skipif(not os.getenv("AUTH_TOKEN"), reason="No auth token specified")
def test_flac_to_stdout(tmp_path: Path) -> None:
os.chdir(tmp_path)
r = call_scdl_with_auth(
"-l",
"https://soundcloud.com/57v/original",
"--name-format",
"-",
"--flac",
encoding=None,
)
with open("track.flac", "wb") as f:
assert isinstance(r.stdout, bytes)
f.write(r.stdout)
assert r.returncode == 0
assert_track(tmp_path, "track.flac", "copy", "saves", None)
@pytest.mark.skipif(not os.getenv("AUTH_TOKEN"), reason="No auth token specified")
def test_flac(tmp_path: Path) -> None:
os.chdir(tmp_path)
r = call_scdl_with_auth(
"-l",
"https://soundcloud.com/57v/original",
"--name-format",
"track",
"--flac",
)
assert r.returncode == 0
assert_track(tmp_path, "track.flac", "copy", "saves", None)
@pytest.mark.skipif(not os.getenv("AUTH_TOKEN"), reason="No auth token specified")
def test_m4a(tmp_path: Path) -> None:
os.chdir(tmp_path)
r = call_scdl_with_auth(
"-l",
"https://soundcloud.com/7x11x13/wan-bushi-eurodance-vibes-part-123",
"--name-format",
"track",
"--no-original",
"--opus",
)
assert r.returncode == 0
if (tmp_path / "track.opus").exists():
pytest.skip("No go+ subscription")
assert_track(
tmp_path,
"track.m4a",
"Wan Bushi - Eurodance Vibes (part 1+2+3)",
"7x11x13",
"Electronic",
None,
)
def test_opus(tmp_path: Path) -> None:
os.chdir(tmp_path)
r = call_scdl_with_auth(
"-l",
"https://soundcloud.com/one-thousand-and-one/test-track",
"--name-format",
"track",
"--no-original",
"--opus",
)
assert r.returncode == 0
assert_track(tmp_path, "track.opus")
def test_mp3(tmp_path: Path) -> None:
os.chdir(tmp_path)
r = call_scdl_with_auth(
"-l",
"https://soundcloud.com/one-thousand-and-one/test-track",
"--name-format",
"track",
"--onlymp3",
)
assert r.returncode == 0
assert_track(tmp_path, "track.mp3")
def test_unlisted_track(tmp_path: Path) -> None:
os.chdir(tmp_path)
r = call_scdl_with_auth(
"-l",
"https://soundcloud.com/one-thousand-and-one/test-track-2/s-fgLQFAzNIMP",
"--name-format",
"track",
"--onlymp3",
)
assert r.returncode == 0
assert_track(tmp_path, "track.mp3", "test track 2")
def test_original_art(tmp_path: Path) -> None:
os.chdir(tmp_path)
r = call_scdl_with_auth(
"-l",
"https://soundcloud.com/one-thousand-and-one/test-track",
"--name-format",
"track",
"--onlymp3",
"--original-art",
)
assert r.returncode == 0
assert_track(tmp_path, "track.mp3", expected_artwork_len=3409)
@pytest.mark.skipif(not os.getenv("AUTH_TOKEN"), reason="No auth token specified")
def test_original_name(tmp_path: Path) -> None:
os.chdir(tmp_path)
r = call_scdl_with_auth(
"-l",
"https://soundcloud.com/57v/original",
"--name-format",
"track",
"--original-name",
)
assert r.returncode == 0
assert_track(tmp_path, "original.wav", check_metadata=False)
@pytest.mark.skipif(not os.getenv("AUTH_TOKEN"), reason="No auth token specified")
def test_original_metadata(tmp_path: Path) -> None:
os.chdir(tmp_path)
r = call_scdl_with_auth(
"-l",
"https://soundcloud.com/57v/original",
"--name-format",
"track",
"--original-metadata",
)
assert r.returncode == 0
assert_track(tmp_path, "track.wav", "og title", "og artist", "og genre", 0)
@pytest.mark.skipif(not os.getenv("AUTH_TOKEN"), reason="No auth token specified")
def test_force_metadata(tmp_path: Path) -> None:
os.chdir(tmp_path)
r = call_scdl_with_auth(
"-l",
"https://soundcloud.com/57v/original",
"--name-format",
"track",
"--original-metadata",
)
assert r.returncode == 0
assert_track(tmp_path, "track.wav", "og title", "og artist", "og genre", 0)
r = call_scdl_with_auth(
"-l",
"https://soundcloud.com/57v/original",
"--name-format",
"track",
"--force-metadata",
)
assert r.returncode == 0
assert_track(tmp_path, "track.wav", "copy", "saves", None)
def test_addtimestamp(tmp_path: Path) -> None:
os.chdir(tmp_path)
r = call_scdl_with_auth(
"-l",
"https://soundcloud.com/one-thousand-and-one/test-track",
"--onlymp3",
"--addtimestamp",
)
assert r.returncode == 0
assert_track(tmp_path, "1719169486_testing - test track.mp3", check_metadata=False)
def test_addtofile(tmp_path: Path) -> None:
os.chdir(tmp_path)
r = call_scdl_with_auth(
"-l",
"https://soundcloud.com/one-thousand-and-one/test-track-2/s-fgLQFAzNIMP",
"--onlymp3",
"--addtofile",
)
assert r.returncode == 0
assert_track(tmp_path, "7x11x13-testing - test track 2.mp3", check_metadata=False)
def test_extract_artist(tmp_path: Path) -> None:
os.chdir(tmp_path)
r = call_scdl_with_auth(
"-l",
"https://soundcloud.com/one-thousand-and-one/test-track",
"--onlymp3",
"--name-format",
"track",
"--extract-artist",
)
assert r.returncode == 0
assert_track(tmp_path, "track.mp3", "test track", "testing")
def test_maxsize(tmp_path: Path) -> None:
os.chdir(tmp_path)
r = call_scdl_with_auth(
"-l",
"https://soundcloud.com/one-thousand-and-one/test-track",
"--onlymp3",
"--max-size=10kb",
)
assert r.returncode == 1
assert "not within --min-size=0 and --max-size=10240" in r.stderr
def test_minsize(tmp_path: Path) -> None:
os.chdir(tmp_path)
r = call_scdl_with_auth(
"-l",
"https://soundcloud.com/one-thousand-and-one/test-track",
"--onlymp3",
"--min-size=1mb",
)
assert r.returncode == 1
assert f"not within --min-size={1024**2} and --max-size={math.inf}" in r.stderr
def test_only_original(tmp_path: Path) -> None:
os.chdir(tmp_path)
r = call_scdl_with_auth(
"-l",
"https://soundcloud.com/one-thousand-and-one/test-track-2/s-fgLQFAzNIMP",
"--only-original",
)
assert r.returncode == 1
assert "does not have original file available" in r.stderr
def test_overwrite(tmp_path: Path) -> None:
os.chdir(tmp_path)
r = call_scdl_with_auth(
"-l",
"https://soundcloud.com/one-thousand-and-one/test-track",
"--name-format",
"track",
"--onlymp3",
)
assert r.returncode == 0
r = call_scdl_with_auth(
"-l",
"https://soundcloud.com/one-thousand-and-one/test-track",
"--name-format",
"track",
"--onlymp3",
)
assert r.returncode == 1
assert "already exists" in r.stderr
r = call_scdl_with_auth(
"-l",
"https://soundcloud.com/one-thousand-and-one/test-track",
"--name-format",
"track",
"--onlymp3",
"--overwrite",
)
assert r.returncode == 0
def test_path(tmp_path: Path) -> None:
r = call_scdl_with_auth(
"-l",
"https://soundcloud.com/one-thousand-and-one/test-track",
"--name-format",
"track",
"--onlymp3",
"--path",
str(tmp_path),
)
assert r.returncode == 0
assert_track(tmp_path, "track.mp3", check_metadata=False)
def test_remove(tmp_path: Path) -> None:
os.chdir(tmp_path)
r = call_scdl_with_auth(
"-l",
"https://soundcloud.com/one-thousand-and-one/test-track",
"--name-format",
"track",
"--onlymp3",
)
assert r.returncode == 0
assert_track(tmp_path, "track.mp3", check_metadata=False)
r = call_scdl_with_auth(
"-l",
"https://soundcloud.com/one-thousand-and-one/test-track-2/s-fgLQFAzNIMP",
"--name-format",
"track2",
"--remove",
"--onlymp3",
)
assert r.returncode == 0
assert_track(tmp_path, "track2.mp3", check_metadata=False)
assert_not_track(tmp_path, "track.mp3")
def test_download_archive(tmp_path: Path) -> None:
os.chdir(tmp_path)
r = call_scdl_with_auth(
"-l",
"https://soundcloud.com/one-thousand-and-one/test-track",
"--name-format",
"track",
"--onlymp3",
"--download-archive=archive.txt",
)
assert r.returncode == 0
os.remove("track.mp3")
assert not os.path.exists("track.mp3")
r = call_scdl_with_auth(
"-l",
"https://soundcloud.com/one-thousand-and-one/test-track",
"--name-format",
"track",
"--onlymp3",
"--download-archive=archive.txt",
)
assert r.returncode == 1
assert "already exists" in r.stderr
def test_description_file(tmp_path: Path) -> None:
os.chdir(tmp_path)
r = call_scdl_with_auth(
"-l",
"https://soundcloud.com/one-thousand-and-one/test-track",
"--name-format",
"track",
"--onlymp3",
"--add-description",
)
assert r.returncode == 0
desc_file = Path("./track.txt")
assert desc_file.exists()
with open(desc_file, encoding="utf-8") as f:
assert f.read().splitlines() == ["test description:", "9439290883"]
Reference in New Issue View Git Blame Copy Permalink