xiaoxia-saas/tests/unit/test_viral_video_models.py

"""tests for GET /api/v1/viral-video/models route function (#2159)."""

from __future__ import annotations

from unittest.mock import MagicMock, patch


class TestModelsRoute:
    def _call(self):
        from apps.api.app.api.routes import viral_video as routes

        return routes.list_available_models()

    def test_without_dashscope_hides_wan(self):
        with patch("apps.api.app.api.routes.viral_video.get_dashscope_client", return_value=None):
            result = self._call()
        assert "models" in result
        keys = {m["key"] for m in result["models"]}
        assert "seedance-2.5" in keys
        assert "wan-3.0" not in keys
        for m in result["models"]:
            if m["key"].startswith("seedance"):
                assert m["supports_audio"] is True

    def test_with_dashscope_includes_wan(self):
        with patch("apps.api.app.api.routes.viral_video.get_dashscope_client", return_value=MagicMock()):
            result = self._call()
        keys = {m["key"] for m in result["models"]}
        assert "wan-3.0" in keys
        wan = next(m for m in result["models"] if m["key"] == "wan-3.0")
        assert wan["billing_mode"] == "per_second"

    def test_exactly_one_default(self):
        with patch("apps.api.app.api.routes.viral_video.get_dashscope_client", return_value=None):
            result = self._call()
        defaults = [m for m in result["models"] if m["is_default"]]
        assert len(defaults) == 1
        assert defaults[0]["key"] == "seedance-2.5"