mirror of https://github.com/morpheus65535/bazarr
131 lines
3.9 KiB
Python
131 lines
3.9 KiB
Python
import inspect
|
|
|
|
import pytest
|
|
from subliminal_patch.core import Language
|
|
|
|
from bazarr.app import get_providers
|
|
|
|
|
|
def test_get_providers_auth():
|
|
for val in get_providers.get_providers_auth().values():
|
|
assert isinstance(val, dict)
|
|
|
|
|
|
def test_get_providers_auth_with_provider_registry():
|
|
"""Make sure all providers will be properly initialized with bazarr
|
|
configs"""
|
|
from subliminal_patch.extensions import provider_registry
|
|
|
|
auths = get_providers.get_providers_auth()
|
|
for key, val in auths.items():
|
|
provider = provider_registry[key]
|
|
sign = inspect.signature(provider.__init__)
|
|
for sub_key in val.keys():
|
|
if sub_key not in sign.parameters:
|
|
raise ValueError(f"'{sub_key}' parameter not present in {provider}")
|
|
|
|
assert sign.parameters[sub_key] is not None
|
|
|
|
|
|
def test_get_providers_auth_embeddedsubtitles():
|
|
item = get_providers.get_providers_auth()["embeddedsubtitles"]
|
|
assert isinstance(item["included_codecs"], list)
|
|
assert isinstance(item["hi_fallback"], bool)
|
|
assert isinstance(item["cache_dir"], str)
|
|
assert isinstance(item["ffprobe_path"], str)
|
|
assert isinstance(item["ffmpeg_path"], str)
|
|
assert isinstance(item["timeout"], str)
|
|
assert isinstance(item["unknown_as_english"], bool)
|
|
|
|
|
|
def test_get_providers_auth_karagarga():
|
|
item = get_providers.get_providers_auth()["karagarga"]
|
|
assert item["username"] is not None
|
|
assert item["password"] is not None
|
|
assert item["f_username"] is not None
|
|
assert item["f_password"] is not None
|
|
|
|
|
|
def test_get_language_equals_default_settings():
|
|
assert isinstance(get_providers.get_language_equals(), list)
|
|
|
|
|
|
def test_get_language_equals_injected_settings_invalid():
|
|
config = get_providers.settings
|
|
config.set("general", "language_equals", '["invalid"]')
|
|
assert not get_providers.get_language_equals(config)
|
|
|
|
|
|
def test_get_language_equals_injected_settings_valid():
|
|
config = get_providers.settings
|
|
config.set("general", "language_equals", '["spa:spa-MX"]')
|
|
|
|
result = get_providers.get_language_equals(config)
|
|
assert result == [(Language("spa"), Language("spa", "MX"))]
|
|
|
|
|
|
@pytest.mark.parametrize(
|
|
"config_value,expected",
|
|
[
|
|
('["spa:spl"]', (Language("spa"), Language("spa", "MX"))),
|
|
('["por:pob"]', (Language("por"), Language("por", "BR"))),
|
|
('["zho:zht"]', (Language("zho"), Language("zho", "TW"))),
|
|
],
|
|
)
|
|
def test_get_language_equals_injected_settings_custom_lang_alpha3(
|
|
config_value, expected
|
|
):
|
|
config = get_providers.settings
|
|
|
|
config.set("general", "language_equals", config_value)
|
|
|
|
result = get_providers.get_language_equals(config)
|
|
assert result == [expected]
|
|
|
|
|
|
def test_get_language_equals_injected_settings_multiple():
|
|
config = get_providers.settings
|
|
|
|
config.set(
|
|
"general",
|
|
"language_equals",
|
|
"['eng@hi:eng', 'spa:spl', 'spa@hi:spl', 'spl@hi:spl']",
|
|
)
|
|
|
|
result = get_providers.get_language_equals(config)
|
|
assert len(result) == 4
|
|
|
|
|
|
def test_get_language_equals_injected_settings_valid_multiple():
|
|
config = get_providers.settings
|
|
config.set("general", "language_equals", '["spa:spa-MX", "spa-MX:spa"]')
|
|
|
|
result = get_providers.get_language_equals(config)
|
|
assert result == [
|
|
(Language("spa"), Language("spa", "MX")),
|
|
(Language("spa", "MX"), Language("spa")),
|
|
]
|
|
|
|
|
|
def test_get_language_equals_injected_settings_hi():
|
|
config = get_providers.settings
|
|
config.set("general", "language_equals", '["eng@hi:eng"]')
|
|
|
|
result = get_providers.get_language_equals(config)
|
|
assert result == [(Language("eng", hi=True), Language("eng"))]
|
|
|
|
|
|
def _get_error():
|
|
try:
|
|
raise ValueError("Some error" * 100)
|
|
except ValueError as error:
|
|
return error
|
|
|
|
|
|
def test_get_traceback_info():
|
|
error_ = _get_error()
|
|
|
|
if error_ is not None:
|
|
msg = get_providers._get_traceback_info(error_)
|
|
assert len(msg) == 100
|