# -*- coding: utf-8 -*-
"""compute_catalog 模型映射（admin 模型 → 桌面端形状）纯函数测试。"""
from __future__ import annotations

from app.services.compute_catalog import _map


def test_map_admin_model():
    row = {
        "model_name": "deepseek-chat", "name": "DeepSeek Chat", "actual_model": "deepseek-chat",
        "group": "default", "billing_unit": "/百万tokens", "input_price": 0.5, "output_price": 1.5,
        "cache_hit_price": 0.1, "vision_support": False, "image_support": False,
        "audio_support": True, "video_support": False, "tags": "对话,通用", "vendor_id": 0, "status": 1,
    }
    m = _map(row)
    assert m["id"] == "deepseek-chat"
    assert m["name"] == "DeepSeek Chat"
    assert m["input_price"] == 0.5 and m["output_price"] == 1.5
    assert m["audio_support"] is True
    assert m["tags"] == ["对话", "通用"]


def test_map_falls_back_to_model_name_when_empty_name():
    m = _map({"model_name": "gpt-4o", "name": ""})
    assert m["id"] == "gpt-4o" and m["name"] == "gpt-4o" and m["actual_model"] == "gpt-4o"