cmshoppe/tests/test_product_suite_gui.py

import os
import sys
import unittest

os.environ.setdefault("QT_QPA_PLATFORM", "offscreen")
sys.path.insert(0, os.path.dirname(__file__))

from _helpers import TempDirMixin

from app import accounts, appconfig, image_studio, image_studio_images
from app import gui

if gui.QT_IMPORT_ERROR is not None:
    raise unittest.SkipTest("PySide6 未安装")

from PySide6.QtGui import QImage
from PySide6.QtWidgets import QApplication, QLabel, QPushButton

from app.gui.tabs.product_suite import ProductSuiteTab, SuiteResultCard


class ProductSuiteGuiTests(TempDirMixin, unittest.TestCase):
    @classmethod
    def setUpClass(cls):
        cls.app = QApplication.instance() or QApplication([])

    def tearDown(self):
        for widget in QApplication.topLevelWidgets():
            widget.close()
            widget.deleteLater()
        self.app.processEvents()

    def _config(self, temp_dir):
        return {
            "chrome_path": "chrome.exe",
            "user_data_root": os.path.join(temp_dir, "chrome_user_data_dir"),
            "image_dir": os.path.join(temp_dir, "images"),
            "db_path": os.path.join(temp_dir, "cmshopee.db"),
            "debug_port_range": [9222, 9260],
            "config_path": os.path.join(temp_dir, "config.json"),
            "cmhub_config_path": os.path.join(temp_dir, "cmhub.json"),
        }

    def _write_image(self, path):
        image = QImage(40, 30, QImage.Format_RGB32)
        image.fill(0xFF336699)
        self.assertTrue(image.save(path))

    def test_tab_builds_suite_controls_without_old_detail_workspace(self):
        with self.make_temp_dir() as temp_dir:
            config = self._config(temp_dir)
            accounts.create_account("主店", "alias-a", debug_port=9222, config=config)
            tab = ProductSuiteTab(config=config, db_path=config["db_path"])
            self.addCleanup(tab.close)
            tab.resize(1180, 760)
            tab.show()
            self.app.processEvents()

            self.assertEqual("productSuiteTab", tab.objectName())
            self.assertEqual(1, tab.task_tabs.count())
            self.assertEqual("套图任务 1", tab.task_tabs.tabText(0))
            self.assertEqual("alias-a", tab.account_combo.currentData())
            self.assertEqual("Shopee", tab.platform_combo.currentData())
            self.assertEqual("中国台湾", tab.country_combo.currentData())
            self.assertEqual("繁体中文", tab.language_combo.currentData())
            self.assertEqual("1:1", tab.ratio_combo.currentData())
            self.assertEqual("Shopee", tab.platform_combo.currentText())
            self.assertEqual("中国台湾", tab.country_combo.currentText())
            self.assertEqual("繁体中文", tab.language_combo.currentText())
            self.assertEqual("1:1", tab.ratio_combo.currentText())
            self.assertEqual("平台", tab.platform_label.text())
            self.assertEqual("站点", tab.country_label.text())
            self.assertEqual("语言", tab.language_label.text())
            self.assertEqual("比例", tab.ratio_label.text())
            for column, combo in enumerate(
                (
                    tab.platform_combo,
                    tab.country_combo,
                    tab.language_combo,
                    tab.ratio_combo,
                )
            ):
                self.assertIs(combo, tab.settings_grid.itemAtPosition(1, column).widget())
            self.assertEqual(
                1,
                len(
                    {
                        tab.platform_combo.width(),
                        tab.country_combo.width(),
                        tab.language_combo.width(),
                        tab.ratio_combo.width(),
                    }
                ),
            )
            self.assertLessEqual(tab.task_tabs.maximumHeight(), 36)
            self.assertEqual(120, tab.account_combo.minimumWidth())
            self.assertEqual(160, tab.account_combo.maximumWidth())
            self.assertGreaterEqual(tab.item_id_edit.minimumWidth(), 120)
            self.assertLessEqual(tab.item_id_edit.maximumWidth(), 140)
            self.assertLess(
                tab.context_bar_layout.indexOf(tab.history_button),
                tab.context_bar_layout.indexOf(tab.account_combo),
            )
            self.assertLess(
                tab.context_bar_layout.indexOf(tab.open_folder_button),
                tab.context_bar_layout.indexOf(tab.account_combo),
            )
            self.assertLess(
                tab.context_bar_layout.indexOf(tab.add_images_button),
                tab.context_bar_layout.indexOf(tab.account_combo),
            )
            self.assertEqual(-1, tab.results_toolbar_layout.indexOf(tab.history_button))
            self.assertEqual(-1, tab.results_toolbar_layout.indexOf(tab.open_folder_button))
            self.assertEqual("打开结果文件夹", tab.open_folder_button.text())
            self.assertEqual("合计 5 张", tab.category_total_label.text())
            self.assertEqual("生成套图（5）", tab.generate_button.text())

            visible_text = " ".join(
                [widget.text() for widget in tab.findChildren(QLabel)]
                + [widget.text() for widget in tab.findChildren(QPushButton)]
            )
            self.assertNotIn("详情图", visible_text)
            self.assertNotIn("AI工场", visible_text)
            self.assertIn("白底图", visible_text)
            self.assertIn("场景图", visible_text)
            self.assertIn("卖点图", visible_text)

            tab.add_custom_category()
            self.assertFalse(tab.custom_category_edit.isHidden())
            tab.custom_category_edit.setText("尺寸图")
            tab._commit_custom_category()
            self.assertTrue(tab.custom_category_edit.isHidden())
            self.assertIn("尺寸图", tab._displayed_state.settings["categories"])
            self.assertEqual("尺寸图", tab._displayed_state.active_category)

        self.assert_removed(temp_dir)

    def test_recent_dropdown_settings_restore_after_restart_and_project_wins(self):
        with self.make_temp_dir() as temp_dir:
            config_path = os.path.join(temp_dir, "config.json")
            config = appconfig.save_config(self._config(temp_dir), path=config_path)
            account = accounts.create_account(
                "主店",
                "alias-a",
                debug_port=9222,
                config=config,
            )
            project = image_studio.create_or_get_project(
                account,
                item_id="51100639510",
                path=config["db_path"],
            )
            project_settings = image_studio.project_suite_settings(project)
            project_settings["ratio"] = "4:3"
            image_studio.update_project_suite_settings(
                project.id,
                project_settings,
                path=config["db_path"],
            )

            first = ProductSuiteTab(
                config=config,
                config_path=config_path,
                db_path=config["db_path"],
            )
            first.ratio_combo.setCurrentIndex(first.ratio_combo.findData("16:9"))
            self.assertEqual(
                "16:9",
                appconfig.product_suite_last_settings(
                    appconfig.load_config(config_path)
                )["ratio"],
            )
            first.close()
            first.deleteLater()
            self.app.processEvents()

            reloaded = appconfig.load_config(config_path)
            second = ProductSuiteTab(
                config=reloaded,
                config_path=config_path,
                db_path=reloaded["db_path"],
            )
            self.addCleanup(second.close)
            self.assertEqual("16:9", second.ratio_combo.currentData())

            state = second._displayed_state
            state.account_alias = "alias-a"
            state.item_id = "51100639510"
            second._bind_project(state, load_existing=True)

            self.assertEqual("4:3", second.ratio_combo.currentData())
            self.assertEqual("4:3", state.settings["ratio"])

        self.assert_removed(temp_dir)

    def test_task_tabs_keep_independent_prompt_and_settings(self):
        with self.make_temp_dir() as temp_dir:
            config = self._config(temp_dir)
            accounts.create_account("主店", "alias-a", debug_port=9222, config=config)
            tab = ProductSuiteTab(config=config, db_path=config["db_path"])
            self.addCleanup(tab.close)

            tab.prompt_edit.setPlainText("任务一卖点")
            first_state = tab._displayed_state
            tab.ratio_combo.setCurrentIndex(tab.ratio_combo.findData("3:4"))
            second_state = tab.add_task(inherit=True)

            self.assertEqual(2, tab.task_tabs.count())
            self.assertEqual("任务一卖点", second_state.prompt)
            self.assertEqual("3:4", second_state.settings["ratio"])
            tab.prompt_edit.setPlainText("任务二卖点")
            tab.ratio_combo.setCurrentIndex(tab.ratio_combo.findData("16:9"))
            tab.task_tabs.setCurrentIndex(0)

            self.assertIs(first_state, tab._displayed_state)
            self.assertEqual("任务一卖点", tab.prompt_edit.toPlainText())
            self.assertEqual("3:4", tab.ratio_combo.currentData())

        self.assert_removed(temp_dir)

    def test_project_settings_and_result_history_use_existing_backend(self):
        with self.make_temp_dir() as temp_dir:
            config = self._config(temp_dir)
            account = accounts.create_account("主店", "alias-a", debug_port=9222, config=config)
            project = image_studio.create_or_get_project(
                account,
                item_id="51100639510",
                path=config["db_path"],
            )
            source_path = os.path.join(temp_dir, "source.png")
            self._write_image(source_path)
            source = image_studio_images.import_original_files(
                project.id,
                [source_path],
                path=config["db_path"],
                config=config,
            )["assets"][0]
            job = image_studio.create_job(
                project.id,
                source_asset_id=source.id,
                job_type="场景图",
                prompt="场景卖点",
                path=config["db_path"],
            )
            image_studio.update_job_status(
                job.id,
                "failed",
                error="上游超时 https://example.invalid/private",
                path=config["db_path"],
            )

            tab = ProductSuiteTab(config=config, db_path=config["db_path"])
            self.addCleanup(tab.close)
            state = tab._displayed_state
            state.account_alias = "alias-a"
            state.item_id = "51100639510"
            state.project_id = project.id
            state.current_job_ids = [job.id]
            tab._load_state(state)

            self.assertEqual([source.id], tab.original_list.asset_ids())
            self.assertEqual("共 1 张 · 成功 0 张", tab.result_summary_label.text())
            cards = tab.findChildren(SuiteResultCard)
            self.assertEqual(1, len(cards))
            self.assertNotIn(
                "https://",
                " ".join(label.text() for label in cards[0].findChildren(QLabel)),
            )

            state.settings["ratio"] = "4:3"
            state.prompt = "持久化卖点"
            tab._persist_state(state)
            stored = image_studio.get_project(project.id, path=config["db_path"])
            self.assertEqual("持久化卖点", stored.draft_prompt)
            self.assertEqual("4:3", image_studio.project_suite_settings(stored)["ratio"])

        self.assert_removed(temp_dir)


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