mirror of
https://github.com/unshackle-dl/unshackle.git
synced 2026-11-11 19:45:06 +00:00
52 lines
2.4 KiB
Python
52 lines
2.4 KiB
Python
from types import SimpleNamespace
|
|
|
|
from langcodes import Language
|
|
|
|
from unshackle.commands.dl import apply_enrich, load_tracks
|
|
from unshackle.core.remote_service import apply_title_fields
|
|
from unshackle.core.titles import Episode, Movie, Song
|
|
from unshackle.core.tracks import Chapters, Tracks
|
|
|
|
SERVICE = type("SVC", (), {})
|
|
|
|
|
|
def refreshing_service(info):
|
|
"""A service that sets its title again in get_tracks, as a remote session does."""
|
|
return SimpleNamespace(
|
|
get_tracks=lambda title: apply_title_fields(title, info) or Tracks(),
|
|
get_chapters=lambda title: Chapters(),
|
|
)
|
|
|
|
|
|
def test_enrich_wins_over_a_movie_title_set_in_get_tracks():
|
|
movie = Movie(id_="movie-0001", service=SERVICE, name="The Gold Seekers", year=2017)
|
|
service = refreshing_service({"name": "Los Buscadores", "year": 2016, "language": "pt"})
|
|
load_tracks(service, movie, name="The Gold Seekers", year=2017, language=Language.get("es"))
|
|
assert (movie.name, movie.year, str(movie.language)) == ("The Gold Seekers", 2017, "es")
|
|
|
|
|
|
def test_enrich_wins_over_a_series_title_set_in_get_tracks():
|
|
episode = Episode(id_="episode-0001", service=SERVICE, title="English", season=1, number=1, name="Pilot")
|
|
service = refreshing_service({"series_title": "Local", "name": "Pilot", "year": 2016})
|
|
load_tracks(service, episode, name="English", year=2017)
|
|
assert (episode.title, episode.name, episode.year) == ("English", "Pilot", 2017)
|
|
|
|
|
|
def test_no_enrich_keeps_the_title_from_get_tracks():
|
|
movie = Movie(id_="movie-0001", service=SERVICE, name="Old", year=2016)
|
|
load_tracks(refreshing_service({"name": "New", "year": 2017}), movie)
|
|
assert (movie.name, movie.year) == ("New", 2017)
|
|
|
|
|
|
def test_a_song_takes_only_the_language():
|
|
song = Song(id_="song-0001", service=SERVICE, name="Track 1", artist="A", album="B", track=1, disc=1, year=1999)
|
|
apply_enrich(song, "The Gold Seekers", 2017, Language.get("es"))
|
|
assert (song.name, song.year, str(song.language)) == ("Track 1", 1999, "es")
|
|
|
|
|
|
def test_a_song_takes_the_lyrics_set_in_get_tracks():
|
|
song = Song(id_="song-0001", service=SERVICE, name="Track 1", artist="A", album="B", track=1, disc=1)
|
|
info = {"name": "Track 1", "lyrics": "Line", "synced_lyrics": "[00:01.00]Line", "lyrics_suffix": ".lrc"}
|
|
load_tracks(refreshing_service(info), song)
|
|
assert (song.lyrics, song.synced_lyrics, song.lyrics_suffix) == ("Line", "[00:01.00]Line", ".lrc")
|