Import AITURK IDE 1.0.0-beta.1 from Hermes 63279301; preserve MIT license

tests/tools/test_tts_minimax_region.py

@@ -0,0 +1,190 @@
+"""MiniMax TTS region, endpoint, and credential selection tests."""
+
+from unittest.mock import MagicMock, patch
+
+import pytest
+
+from tools.tts_tool import (
+    DEFAULT_MINIMAX_BASE_URL,
+    DEFAULT_MINIMAX_CN_BASE_URL,
+    _generate_minimax_tts,
+    _resolve_minimax_tts_runtime,
+    check_tts_requirements,
+)
+
+
+GLOBAL_CREDENTIAL_SENTINEL = "FAKE_GLOBAL_CREDENTIAL"
+CN_CREDENTIAL_SENTINEL = "FAKE_CN_CREDENTIAL"
+
+
+@pytest.fixture(autouse=True)
+def _fake_minimax_credentials(monkeypatch):
+    values = {}
+    monkeypatch.setattr(
+        "tools.tts_tool.get_env_value",
+        lambda name, default=None: values.get(name, default),
+    )
+    return values
+
+
+@pytest.mark.parametrize(
+    ("config", "credentials", "expected"),
+    [
+        pytest.param(
+            {},
+            {"MINIMAX_API_KEY": GLOBAL_CREDENTIAL_SENTINEL},
+            (
+                "global",
+                DEFAULT_MINIMAX_BASE_URL,
+                "MINIMAX_API_KEY",
+                GLOBAL_CREDENTIAL_SENTINEL,
+            ),
+            id="global-only",
+        ),
+        pytest.param(
+            {},
+            {"MINIMAX_CN_API_KEY": CN_CREDENTIAL_SENTINEL},
+            (
+                "cn",
+                DEFAULT_MINIMAX_CN_BASE_URL,
+                "MINIMAX_CN_API_KEY",
+                CN_CREDENTIAL_SENTINEL,
+            ),
+            id="china-only",
+        ),
+        pytest.param(
+            {},
+            {
+                "MINIMAX_API_KEY": GLOBAL_CREDENTIAL_SENTINEL,
+                "MINIMAX_CN_API_KEY": CN_CREDENTIAL_SENTINEL,
+            },
+            (
+                "global",
+                DEFAULT_MINIMAX_BASE_URL,
+                "MINIMAX_API_KEY",
+                GLOBAL_CREDENTIAL_SENTINEL,
+            ),
+            id="both-default-to-global",
+        ),
+        pytest.param(
+            {"minimax": {"region": "global"}},
+            {
+                "MINIMAX_API_KEY": GLOBAL_CREDENTIAL_SENTINEL,
+                "MINIMAX_CN_API_KEY": CN_CREDENTIAL_SENTINEL,
+            },
+            (
+                "global",
+                DEFAULT_MINIMAX_BASE_URL,
+                "MINIMAX_API_KEY",
+                GLOBAL_CREDENTIAL_SENTINEL,
+            ),
+            id="explicit-global",
+        ),
+        pytest.param(
+            {"minimax": {"region": "cn"}},
+            {
+                "MINIMAX_API_KEY": GLOBAL_CREDENTIAL_SENTINEL,
+                "MINIMAX_CN_API_KEY": CN_CREDENTIAL_SENTINEL,
+            },
+            (
+                "cn",
+                DEFAULT_MINIMAX_CN_BASE_URL,
+                "MINIMAX_CN_API_KEY",
+                CN_CREDENTIAL_SENTINEL,
+            ),
+            id="explicit-china",
+        ),
+    ],
+)
+def test_runtime_selection_matrix(
+    _fake_minimax_credentials,
+    config,
+    credentials,
+    expected,
+):
+    _fake_minimax_credentials.update(credentials)
+
+    runtime = _resolve_minimax_tts_runtime(config)
+
+    assert (
+        runtime.region,
+        runtime.endpoint,
+        runtime.credential_source,
+        runtime.api_key,
+    ) == expected
+
+
+@pytest.mark.parametrize(
+    ("region", "credentials", "missing_source"),
+    [
+        pytest.param(
+            "global",
+            {"MINIMAX_CN_API_KEY": CN_CREDENTIAL_SENTINEL},
+            "MINIMAX_API_KEY",
+            id="global-does-not-borrow-china-key",
+        ),
+        pytest.param(
+            "cn",
+            {"MINIMAX_API_KEY": GLOBAL_CREDENTIAL_SENTINEL},
+            "MINIMAX_CN_API_KEY",
+            id="china-does-not-borrow-global-key",
+        ),
+    ],
+)
+def test_explicit_region_requires_matching_credential(
+    _fake_minimax_credentials,
+    region,
+    credentials,
+    missing_source,
+):
+    _fake_minimax_credentials.update(credentials)
+
+    with pytest.raises(ValueError, match=missing_source):
+        _resolve_minimax_tts_runtime({"minimax": {"region": region}})
+
+
+@pytest.mark.parametrize(
+    ("config", "credentials", "expected"),
+    [
+        pytest.param(
+            {"provider": "minimax"},
+            {"MINIMAX_CN_API_KEY": CN_CREDENTIAL_SENTINEL},
+            True,
+            id="china-only-available",
+        ),
+        pytest.param(
+            {"provider": "minimax", "minimax": {"region": "cn"}},
+            {"MINIMAX_API_KEY": GLOBAL_CREDENTIAL_SENTINEL},
+            False,
+            id="selected-region-missing",
+        ),
+        pytest.param(
+            {"provider": "minimax", "minimax": {"region": "invalid"}},
+            {
+                "MINIMAX_API_KEY": GLOBAL_CREDENTIAL_SENTINEL,
+                "MINIMAX_CN_API_KEY": CN_CREDENTIAL_SENTINEL,
+            },
+            False,
+            id="invalid-region",
+        ),
+    ],
+)
+def test_availability_uses_atomic_runtime(
+    monkeypatch,
+    _fake_minimax_credentials,
+    config,
+    credentials,
+    expected,
+):
+    _fake_minimax_credentials.update(credentials)
+    monkeypatch.setattr("tools.tts_tool._load_tts_config", lambda: config)
+
+    assert check_tts_requirements() is expected
+
+
+def test_runtime_repr_excludes_raw_credential(_fake_minimax_credentials):
+    _fake_minimax_credentials["MINIMAX_API_KEY"] = GLOBAL_CREDENTIAL_SENTINEL
+
+    runtime = _resolve_minimax_tts_runtime({})
+
+    assert GLOBAL_CREDENTIAL_SENTINEL not in repr(runtime)