cmbot/tests/test_ai_outfit_panel.py

"""Smoke tests for AI outfit panel defaults (offscreen Qt)."""
import os
import shutil
import sys
import tempfile
import unittest
from pathlib import Path

os.environ.setdefault("QT_QPA_PLATFORM", "offscreen")
sys.path.insert(0, str(Path(__file__).parent.parent / "src"))

import services.file_service as fs
from services.config_service import DEFAULT_CONFIG


class TestAiOutfitPanelDefaults(unittest.TestCase):
    def setUp(self):
        self.tmp = Path(tempfile.mkdtemp())
        self._env = os.environ.get("CMBOT_DATA_DIR")
        os.environ["CMBOT_DATA_DIR"] = str(self.tmp / "data")
        self.app_dir = self.tmp / "app"
        self.app_dir.mkdir(parents=True)
        self._orig_get_app_dir = fs.get_app_dir
        fs.get_app_dir = lambda: self.app_dir
        self.app = self._app()

    def tearDown(self):
        fs.get_app_dir = self._orig_get_app_dir
        if self._env is None:
            os.environ.pop("CMBOT_DATA_DIR", None)
        else:
            os.environ["CMBOT_DATA_DIR"] = self._env
        shutil.rmtree(str(self.tmp), ignore_errors=True)

    def _app(self):
        from PySide6.QtWidgets import QApplication

        app = QApplication.instance()
        if app is None:
            app = QApplication([])
        return app

    def _panel(self):
        from app.widgets.ai_outfit_panel import AiOutfitPanel

        return AiOutfitPanel()

    def test_empty_config_defaults_to_outfit_output_dir(self):
        panel = self._panel()
        cfg = dict(DEFAULT_CONFIG)
        cfg["outfit_output_dir"] = ""

        panel.apply_config(cfg)

        self.assertEqual(Path(panel._output_edit.text()), self.app_dir / "穿搭图片")

    def test_saved_outfit_output_dir_takes_precedence(self):
        panel = self._panel()
        saved = self.tmp / "custom-output"
        cfg = dict(DEFAULT_CONFIG)
        cfg["outfit_output_dir"] = str(saved)

        panel.apply_config(cfg)

        self.assertEqual(Path(panel._output_edit.text()), saved)

    def test_generation_setting_label_is_image_concurrency(self):
        from PySide6.QtWidgets import QLabel

        panel = self._panel()

        labels = [label.text() for label in panel.findChildren(QLabel)]

        self.assertIn("图片并发数", labels)

    def test_title_group_present_no_dropdown_preview_removed(self):
        """标题生成组存在；无标题模型下拉（配置定名 §17.3）；预览块已移除。"""
        from PySide6.QtWidgets import QLabel

        panel = self._panel()
        panel.apply_config(dict(DEFAULT_CONFIG))

        labels = [label.text() for label in panel.findChildren(QLabel)]
        self.assertIn("标题生成提示词", labels)
        self.assertNotIn("标题模型", labels)        # 下拉已去掉（§17.3 配置定名）
        self.assertNotIn("最终生成要求预览", labels)
        self.assertFalse(hasattr(panel, "_title_model_combo"))
        self.assertFalse(hasattr(panel, "_preview_view"))
        self.assertFalse(hasattr(panel, "_sample_combo"))
        self.assertTrue(hasattr(panel, "_title_btn"))

    def test_title_prompt_defaults_loaded(self):
        from services.config_service import DEFAULT_TITLE_PROMPT

        panel = self._panel()
        panel.apply_config(dict(DEFAULT_CONFIG))

        self.assertEqual(panel._title_prompt_edit.toPlainText(), DEFAULT_TITLE_PROMPT)

    def test_prompt_buttons_save_row_insert_title_and_preview(self):
        """§19.20: 按钮行含「保存」；编辑框下方保留「插入标题」+「预览最终提示词」。"""
        from PySide6.QtWidgets import QPushButton

        panel = self._panel()
        panel.apply_config(dict(DEFAULT_CONFIG))

        texts = [b.text() for b in panel.findChildren(QPushButton)]
        self.assertIn("保存", texts)
        self.assertIn("插入标题", texts)
        self.assertIn("预览最终提示词", texts)
        self.assertNotIn("保存话术", texts)   # 旧按钮已改名上移

    def test_preview_dialog_renders_selected_row(self):
        """§7.3: 预览弹窗按选中数据行替换 {title} 并附加输出要求。"""
        from openpyxl import Workbook
        from app.widgets.ai_outfit_panel import _OutfitPreviewDialog

        xlsx = self.tmp / "rows.xlsx"
        wb = Workbook(); ws = wb.active
        ws.append(["标题", "货号", "原始图片路径", "结果", "状态", "原因"])
        ws.append(["纯棉短袖", "TY001", r"D:\img\a.png", "", "", ""])
        wb.save(str(xlsx)); wb.close()

        panel = self._panel()
        panel.apply_config(dict(DEFAULT_CONFIG))
        panel._excel_edit.setText(str(xlsx))
        panel._prompt_edit.setPlainText("为 {title} 生成")
        panel._set_combo(panel._resolution, "1K")

        dialog = _OutfitPreviewDialog(panel)   # __init__ fills rows + refresh
        # Row dropdown picked up the valid row; pick it.
        self.assertEqual(dialog._combo.count(), 1)
        dialog._combo.setCurrentIndex(0)

        text = dialog._view.toPlainText()
        self.assertTrue(text.startswith("为 纯棉短袖 生成"))
        self.assertIn("批量生成输出要求", text)
        self.assertIn("参考解析度：1K", text)
        dialog.deleteLater()

    def test_find_title_model_resolves_by_config_name(self):
        """title_model 名字命中 ai_models.json → 返回 AiModelConfig。"""
        from services.ai_image_service import AiModelConfig

        panel = self._panel()
        panel._models = [
            {"name": "GPT-5.5 文本", "url": "https://r/v1/chat/completions",
             "model": "gpt-5.5", "api_key": "sk-x", "api_type": "chat"},
            {"name": "图片模型", "url": "https://r/v1/images/edits",
             "model": "img", "api_key": "sk-y", "api_type": "images_edits"},
        ]
        panel._title_model_name = "GPT-5.5 文本"

        config, error = panel._find_title_model()

        self.assertIsNone(error)
        self.assertIsInstance(config, AiModelConfig)
        self.assertEqual(config.model, "gpt-5.5")

    def test_find_title_model_missing_name_errors(self):
        panel = self._panel()
        panel._models = [{"name": "别的模型", "url": "https://r/v1/chat/completions",
                          "model": "m", "api_key": "sk-x", "api_type": "chat"}]
        panel._title_model_name = "GPT-5.5 文本"

        config, error = panel._find_title_model()

        self.assertIsNone(config)
        self.assertIn("GPT-5.5 文本", error)

    def test_find_title_model_no_models_errors(self):
        panel = self._panel()
        panel._models = []
        panel._title_model_name = "GPT-5.5 文本"

        config, error = panel._find_title_model()

        self.assertIsNone(config)
        self.assertTrue(error)

    def test_find_title_model_image_model_rejected_upfront(self):
        """§19.21: title_model 命中图片模型（images_edits）→ 开跑前报错。"""
        panel = self._panel()
        panel._models = [
            {"name": "GPT-5.5 文本", "url": "https://r/v1/images/edits",
             "model": "gpt-image", "api_key": "sk-x", "api_type": "images_edits"},
        ]
        panel._title_model_name = "GPT-5.5 文本"

        config, error = panel._find_title_model()

        self.assertIsNone(config)
        self.assertIn("图片模型", error)

    def test_find_title_model_image_model_by_url_autodetect_rejected(self):
        """api_type=auto 但 URL 是 images 端点 → 仍按图片模型拦截。"""
        panel = self._panel()
        panel._models = [
            {"name": "标题模型", "url": "https://r/v1/images/generations",
             "model": "x", "api_key": "sk-x", "api_type": "auto"},
        ]
        panel._title_model_name = "标题模型"

        config, error = panel._find_title_model()

        self.assertIsNone(config)
        self.assertIn("图片模型", error)


if __name__ == "__main__":
    unittest.main()