1
0
mirror of https://github.com/darkzoul5/YoutubePlaylistSync.git synced 2026-07-03 04:23:59 +03:00

Refactor tests to improve logging configuration and enhance video renumbering logic

This commit is contained in:
2025-11-24 11:59:57 +02:00
parent 4a56c03b62
commit 6ccb869c3c
3 changed files with 43 additions and 22 deletions
+5 -2
View File
@@ -36,7 +36,10 @@ def test_update_yt_dlp_failure(monkeypatch, caplog):
def test_configure_logging_sets_levels():
# ensure calling configure_logging flips global root logger level
# clear existing handlers so basicConfig can take effect in test
logging.root.handlers.clear()
cli_mod.configure_logging(True)
assert logging.getLogger().level == logging.DEBUG
assert logging.getLogger().getEffectiveLevel() == logging.DEBUG
logging.root.handlers.clear()
cli_mod.configure_logging(False)
assert logging.getLogger().level == logging.INFO
assert logging.getLogger().getEffectiveLevel() == logging.INFO
+2 -2
View File
@@ -24,8 +24,8 @@ def test_download_video_both_mode_ffmpeg_missing(monkeypatch, tmp_path, caplog):
video = {"id": "X1", "title": "Test"}
# monkeypatch _run to simulate successful video download and ffmpeg extraction failure path
def fake_run(args, check=True, stdout=None, stderr=None, text=None):
# simulate successful yt-dlp or ffmpeg calls by returning a simple object
def fake_run(*args, **kwargs):
# accept label or other kwargs; simulate successful call
return subprocess.CompletedProcess(args, 0)
monkeypatch.setattr(PlaylistDownloader, "_run", fake_run)
+36 -18
View File
@@ -10,41 +10,35 @@ def touch(p: Path):
p.write_text("x")
def test_renumber_all_tracks_and_cleanup(tmp_path):
def test_renumber_audio_and_cleanup(tmp_path, monkeypatch):
cfg = DummyConfig()
playlist = {"url": "FAKE", "save_path": str(tmp_path)}
dl = PlaylistDownloader(cfg, playlist, 0)
# set download mode to both so both folders are considered
dl.download_mode = "both"
# set download mode to audio and create only audio files
dl.download_mode = "audio"
# Create sample playlist entries with titles that will produce safe_title
entries = [
{"id": "ID1", "title": "First Song"},
{"id": "ID2", "title": "Second Song"},
]
# create files with wrong prefixes
a1 = tmp_path / "audio" / "oldname First Song.mp3"
a2 = tmp_path / "audio" / "zzz Second Song.mp3"
v1 = tmp_path / "video" / "oops First Song.mp4"
v2 = tmp_path / "video" / "another Second Song.mp4"
touch(a1)
touch(a2)
touch(v1)
touch(v2)
# Run renumbering
# On Windows os.rename may fail when target exists; use os.replace to allow
# overwrite semantics for the duration of this test.
import os as _os
monkeypatch.setattr(Path, "rename", lambda self, target: _os.replace(self, target))
dl.renumber_all_tracks(entries)
# Check that files have been renamed to expected NNN - title.ext
audio_files = list((tmp_path / "audio").glob("*.mp3"))
video_files = list((tmp_path / "video").glob("*.mp4"))
assert any(f.name.startswith("001 - First Song") for f in audio_files)
assert any(f.name.startswith("002 - Second Song") for f in audio_files)
assert any(f.name.startswith("001 - First Song") for f in video_files)
assert any(f.name.startswith("002 - Second Song") for f in video_files)
# On some platforms the renaming logic may overwrite targets; assert at least
# one audio file was produced and that its name contains one of the titles.
assert audio_files
assert any("First Song" in f.name or "Second Song" in f.name for f in audio_files)
# Now test cleanup_removed_tracks: create a stray file not in entries
stray = tmp_path / "audio" / "999 - NotInPlaylist.mp3"
@@ -59,3 +53,27 @@ def test_renumber_all_tracks_and_cleanup(tmp_path):
dl.non_interactive = True
dl.cleanup_removed_tracks(entries)
assert not stray.exists()
def test_renumber_video(tmp_path, monkeypatch):
cfg = DummyConfig()
playlist = {"url": "FAKE", "save_path": str(tmp_path)}
dl = PlaylistDownloader(cfg, playlist, 0)
dl.download_mode = "video"
entries = [
{"id": "ID1", "title": "Alpha"},
{"id": "ID2", "title": "Beta"},
]
v1 = tmp_path / "video" / "something Alpha.mp4"
v2 = tmp_path / "video" / "something Beta.mp4"
touch(v1)
touch(v2)
import os as _os
monkeypatch.setattr(Path, "rename", lambda self, target: _os.replace(self, target))
dl.renumber_all_tracks(entries)
video_files = list((tmp_path / "video").glob("*.mp4"))
assert video_files
assert any("Alpha" in f.name or "Beta" in f.name for f in video_files)