mirror of
https://git.quad4.io/RNS-Things/MeshChatX.git
synced 2026-05-12 18:04:43 +00:00
Ivan
102 lines
3.0 KiB
Python
102 lines
3.0 KiB
Python
# SPDX-License-Identifier: 0BSD
|
|
|
|
"""Optional integration checks for Argos CLI (skipped when not installed)."""
|
|
|
|
import shutil
|
|
import urllib.error
|
|
import urllib.request
|
|
|
|
import pytest
|
|
|
|
from meshchatx.src.backend.translator_handler import (
|
|
ARGOS_CLI_EXECUTABLE_NAMES,
|
|
TranslatorHandler,
|
|
_find_argos_cli_executable,
|
|
)
|
|
|
|
|
|
def _argos_cli_on_path() -> bool:
|
|
return _find_argos_cli_executable() is not None
|
|
|
|
|
|
_STANZA_RESOURCES_URL = (
|
|
"https://raw.githubusercontent.com/stanfordnlp/stanza-resources/"
|
|
"main/resources_1.10.0.json"
|
|
)
|
|
|
|
|
|
def _argos_cli_needs_network() -> bool:
|
|
"""Argos sentence splitting can trigger Stanza to fetch resources over HTTPS."""
|
|
req = urllib.request.Request(
|
|
_STANZA_RESOURCES_URL,
|
|
method="HEAD",
|
|
headers={"User-Agent": "MeshChatX-Tests/1"},
|
|
)
|
|
try:
|
|
with urllib.request.urlopen(req, timeout=5) as resp:
|
|
code = getattr(resp, "status", resp.getcode())
|
|
return code is not None and int(code) < 500
|
|
except urllib.error.HTTPError as e:
|
|
return e.code < 500
|
|
except OSError:
|
|
return False
|
|
|
|
|
|
@pytest.mark.integration
|
|
def test_find_argos_cli_matches_shutil_which():
|
|
expected = None
|
|
for name in ARGOS_CLI_EXECUTABLE_NAMES:
|
|
expected = shutil.which(name)
|
|
if expected:
|
|
break
|
|
assert _find_argos_cli_executable() == expected
|
|
|
|
|
|
@pytest.mark.integration
|
|
@pytest.mark.skipif(not _argos_cli_on_path(), reason="Argos CLI not on PATH")
|
|
@pytest.mark.skipif(
|
|
not _argos_cli_needs_network(),
|
|
reason="Network unreachable (Argos CLI may download Stanza resources)",
|
|
)
|
|
def test_translate_en_es_via_cli_round_trip():
|
|
handler = TranslatorHandler(
|
|
translator_argos_enabled=True,
|
|
translator_libretranslate_enabled=True,
|
|
)
|
|
assert handler.has_argos_cli
|
|
assert not handler.has_argos_lib
|
|
|
|
try:
|
|
result = handler.translate_text("Hello", "en", "es", use_argos=True)
|
|
except RuntimeError as exc:
|
|
err = str(exc)
|
|
if any(
|
|
needle in err
|
|
for needle in (
|
|
"Network is unreachable",
|
|
"Failed to establish a new connection",
|
|
"Max retries exceeded",
|
|
)
|
|
):
|
|
pytest.skip(
|
|
"Argos CLI could not reach Stanza resources over HTTPS "
|
|
"(offline, sandbox, or subprocess has no route)"
|
|
)
|
|
raise
|
|
assert result["source"] == "argos"
|
|
assert result["source_lang"] == "en"
|
|
assert result["target_lang"] == "es"
|
|
assert len(result["translated_text"].strip()) > 0
|
|
|
|
|
|
@pytest.mark.integration
|
|
@pytest.mark.skipif(not _argos_cli_on_path(), reason="Argos CLI not on PATH")
|
|
def test_get_supported_languages_includes_argos_when_libretranslate_down():
|
|
handler = TranslatorHandler(
|
|
translator_argos_enabled=True,
|
|
translator_libretranslate_enabled=True,
|
|
)
|
|
langs = handler.get_supported_languages()
|
|
argos = [x for x in langs if x.get("source") == "argos"]
|
|
assert len(argos) >= 1
|