mirror of
https://github.com/unshackle-dl/unshackle.git
synced 2026-11-12 20:15:10 +00:00
imSp4rky
8bacb247ee
Error messages named the resolved language for a user who only ever defaulted to orig. Audio tracks now carry is_original, resolved through Tracks.by_language so it agrees with the downloader.
44 lines
1.5 KiB
Python
44 lines
1.5 KiB
Python
from __future__ import annotations
|
|
|
|
from typing import Optional
|
|
|
|
import pytest
|
|
|
|
from unshackle.core.api.handlers import original_audio_ids
|
|
|
|
|
|
class FakeTrack:
|
|
def __init__(self, language: str) -> None:
|
|
self.id = language
|
|
self.language = language
|
|
|
|
|
|
class FakeTitle:
|
|
def __init__(self, language: Optional[str]) -> None:
|
|
self.language = language
|
|
|
|
|
|
@pytest.mark.parametrize(
|
|
"title_language,track_languages,expected",
|
|
[
|
|
# paradigm collapse: CLDR calls these the same language, RFC 4647 picks the specific one
|
|
(["en"], ["en", "en-US"], ["en"]),
|
|
(["en"], ["en-US", "en-GB"], ["en-US"]),
|
|
(["pt"], ["pt-BR", "pt-PT"], ["pt-BR"]),
|
|
(["es"], ["es-ES", "es-419"], ["es-ES"]),
|
|
# non-paradigm regionals: exact mode drops them, so the fuzzy fallback catches
|
|
# them and the flag still matches what a default 'orig' download selects
|
|
(["es"], ["es-419"], ["es-419"]),
|
|
(["fr"], ["fr-CA"], ["fr-CA"]),
|
|
# alias
|
|
(["zh"], ["cmn"], ["cmn"]),
|
|
# nothing in the original language, and no language on the title at all
|
|
(["en"], ["ja", "de"], []),
|
|
([None], ["en"], []),
|
|
],
|
|
)
|
|
def test_original_audio_ids(title_language: list, track_languages: list[str], expected: list[str]) -> None:
|
|
tracks = [FakeTrack(t) for t in track_languages]
|
|
ids = original_audio_ids(tracks, FakeTitle(title_language[0]))
|
|
assert [t for t in track_languages if t in ids] == expected
|