516 lines
22 KiB
Python
516 lines
22 KiB
Python
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.QtCore import Qt
|
||
from PySide6.QtGui import QIcon, QImage, QPixmap
|
||
from PySide6.QtTest import QTest
|
||
from PySide6.QtWidgets import QApplication, QLabel, QListWidgetItem, QPushButton
|
||
|
||
from app.gui.tabs.product_suite import (
|
||
ORIGINAL_CHECK_STATE_ROLE,
|
||
ProductOriginalDelegate,
|
||
ProductOriginalList,
|
||
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 _create_project_with_assets(self, temp_dir, config, count, item_id="51100639510"):
|
||
account = accounts.create_account(
|
||
"主店",
|
||
"alias-a",
|
||
debug_port=9222,
|
||
config=config,
|
||
)
|
||
project = image_studio.create_or_get_project(
|
||
account,
|
||
item_id=item_id,
|
||
path=config["db_path"],
|
||
)
|
||
assets = []
|
||
for index in range(count):
|
||
source_path = os.path.join(temp_dir, "%s-%02d.png" % (item_id, index + 1))
|
||
self._write_image(source_path)
|
||
assets.append(
|
||
image_studio.add_asset(
|
||
project.id,
|
||
image_studio.ASSET_KIND_ORIGINAL,
|
||
local_path=source_path,
|
||
source_order=index + 1,
|
||
path=config["db_path"],
|
||
)
|
||
)
|
||
return project, assets
|
||
|
||
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)
|
||
|
||
def test_original_list_expands_without_internal_scrollbars(self):
|
||
with self.make_temp_dir() as temp_dir:
|
||
config = self._config(temp_dir)
|
||
project, assets = self._create_project_with_assets(temp_dir, config, 0)
|
||
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
|
||
tab.resize(1180, 760)
|
||
tab.show()
|
||
tab._load_state(state)
|
||
self.app.processEvents()
|
||
|
||
self.assertEqual(Qt.ScrollBarAlwaysOff, tab.original_list.horizontalScrollBarPolicy())
|
||
self.assertEqual(Qt.ScrollBarAlwaysOff, tab.original_list.verticalScrollBarPolicy())
|
||
self.assertEqual(6, tab.original_list.count())
|
||
self.assertEqual(2, tab.original_list.content_row_count())
|
||
empty_height = tab.original_list.height()
|
||
|
||
for target_count in (6, 7, 16):
|
||
for index in range(len(assets), target_count):
|
||
source_path = os.path.join(temp_dir, "added-%02d.png" % (index + 1))
|
||
self._write_image(source_path)
|
||
assets.append(
|
||
image_studio.add_asset(
|
||
project.id,
|
||
image_studio.ASSET_KIND_ORIGINAL,
|
||
local_path=source_path,
|
||
source_order=index + 1,
|
||
path=config["db_path"],
|
||
)
|
||
)
|
||
tab._refresh_originals(state)
|
||
self.app.processEvents()
|
||
self.assertEqual(target_count, len(tab.original_list.asset_ids()))
|
||
expected_rows = (max(6, target_count) + 2) // 3
|
||
self.assertEqual(expected_rows, tab.original_list.content_row_count())
|
||
if target_count == 6:
|
||
self.assertEqual(empty_height, tab.original_list.height())
|
||
|
||
wide_height = tab.original_list.height()
|
||
tab.original_list.setFixedWidth(250)
|
||
self.app.processEvents()
|
||
self.assertEqual(2, tab.original_list.content_column_count())
|
||
self.assertEqual(8, tab.original_list.content_row_count())
|
||
self.assertGreater(tab.original_list.height(), wide_height)
|
||
|
||
self.assert_removed(temp_dir)
|
||
|
||
def test_original_checks_preserve_on_refresh_and_clear_on_task_switch(self):
|
||
with self.make_temp_dir() as temp_dir:
|
||
config = self._config(temp_dir)
|
||
project, assets = self._create_project_with_assets(temp_dir, config, 2)
|
||
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
|
||
tab._load_state(state)
|
||
|
||
self.assertEqual(6, tab.original_list.count())
|
||
self.assertTrue(tab.select_all_originals_button.isEnabled())
|
||
self.assertTrue(tab.invert_originals_button.isEnabled())
|
||
for row in range(2):
|
||
self.assertIsNotNone(tab.original_list.item(row).data(ORIGINAL_CHECK_STATE_ROLE))
|
||
for row in range(2, 6):
|
||
self.assertIsNone(tab.original_list.item(row).data(ORIGINAL_CHECK_STATE_ROLE))
|
||
|
||
tab.select_all_originals_button.click()
|
||
self.assertEqual([asset.id for asset in assets], tab.original_list.checked_asset_ids())
|
||
self.assertEqual("已选 2 张", tab.original_selected_label.text())
|
||
tab.invert_originals_button.click()
|
||
self.assertEqual([], tab.original_list.checked_asset_ids())
|
||
|
||
tab.original_list.set_checked_asset_ids([assets[1].id])
|
||
tab._refresh_originals(state)
|
||
self.assertEqual([assets[1].id], tab.original_list.checked_asset_ids())
|
||
tab.reorder_originals([assets[1].id, assets[0].id])
|
||
self.assertEqual([assets[1].id], tab.original_list.checked_asset_ids())
|
||
|
||
tab.add_task(inherit=True)
|
||
self.assertEqual([], tab.original_list.checked_asset_ids())
|
||
self.assertFalse(tab.select_all_originals_button.isEnabled())
|
||
tab.task_tabs.setCurrentIndex(0)
|
||
self.assertEqual([], tab.original_list.checked_asset_ids())
|
||
self.assertEqual("已选 0 张", tab.original_selected_label.text())
|
||
|
||
self.assert_removed(temp_dir)
|
||
|
||
def test_original_checkbox_click_and_keyboard_delete_keep_actions_separate(self):
|
||
original_list = ProductOriginalList()
|
||
self.addCleanup(original_list.close)
|
||
original_list.setFixedSize(240, 210)
|
||
for asset_id in (11, 12):
|
||
item = QListWidgetItem("原图%d" % asset_id)
|
||
item.setData(Qt.UserRole, asset_id)
|
||
item.setData(ORIGINAL_CHECK_STATE_ROLE, Qt.Unchecked)
|
||
pixmap = QPixmap(82, 64)
|
||
pixmap.fill(0xFF336699)
|
||
item.setIcon(QIcon(pixmap))
|
||
original_list.addItem(item)
|
||
original_list.show()
|
||
self.app.processEvents()
|
||
|
||
clicked = []
|
||
double_clicked = []
|
||
batch_deleted = []
|
||
single_deleted = []
|
||
original_list.itemClicked.connect(lambda item: clicked.append(item.data(Qt.UserRole)))
|
||
original_list.itemDoubleClicked.connect(
|
||
lambda item: double_clicked.append(item.data(Qt.UserRole))
|
||
)
|
||
original_list.deleteAssetsRequested.connect(lambda ids: batch_deleted.append(ids))
|
||
original_list.deleteRequested.connect(lambda asset_id: single_deleted.append(asset_id))
|
||
|
||
first_rect = original_list.visualItemRect(original_list.item(0))
|
||
check_point = ProductOriginalDelegate.checkbox_hit_rect(first_rect).center()
|
||
QTest.mouseClick(original_list.viewport(), Qt.LeftButton, pos=check_point)
|
||
self.assertEqual([11], original_list.checked_asset_ids())
|
||
self.assertEqual([], clicked)
|
||
QTest.mouseDClick(original_list.viewport(), Qt.LeftButton, pos=check_point)
|
||
self.assertEqual([], double_clicked)
|
||
|
||
QTest.mouseClick(original_list.viewport(), Qt.LeftButton, pos=first_rect.center())
|
||
self.assertEqual([11], clicked)
|
||
original_list.set_checked_asset_ids([11, 12])
|
||
self.assertEqual(
|
||
[("删除选中的2张图片…", [11, 12])],
|
||
original_list.context_delete_options(11),
|
||
)
|
||
original_list.set_checked_asset_ids([12])
|
||
self.assertEqual(
|
||
[("删除这张图片…", [11]), ("删除选中图片…", [12])],
|
||
original_list.context_delete_options(11),
|
||
)
|
||
|
||
original_list.setFocus()
|
||
original_list.set_checked_asset_ids([11, 12])
|
||
QTest.keyClick(original_list, Qt.Key_Delete)
|
||
self.assertEqual([[11, 12]], batch_deleted)
|
||
original_list.clear_checks()
|
||
original_list.setCurrentRow(1)
|
||
QTest.keyClick(original_list, Qt.Key_Backspace)
|
||
self.assertEqual([12], single_deleted)
|
||
|
||
def test_batch_delete_confirms_main_image_and_blocks_running_or_downloading(self):
|
||
with self.make_temp_dir() as temp_dir:
|
||
config = self._config(temp_dir)
|
||
project, assets = self._create_project_with_assets(temp_dir, config, 3)
|
||
statuses = []
|
||
tab = ProductSuiteTab(
|
||
config=config,
|
||
db_path=config["db_path"],
|
||
status_callback=lambda message, level=None: statuses.append((message, level)),
|
||
)
|
||
self.addCleanup(tab.close)
|
||
state = tab._displayed_state
|
||
state.account_alias = "alias-a"
|
||
state.item_id = "51100639510"
|
||
state.project_id = project.id
|
||
tab._load_state(state)
|
||
|
||
confirmations = []
|
||
tab._confirm = lambda title, message, **kwargs: confirmations.append(
|
||
(title, message, kwargs)
|
||
) or True
|
||
tab.original_list.set_checked_asset_ids([assets[0].id, assets[1].id])
|
||
tab.delete_originals(tab.original_list.checked_asset_ids())
|
||
|
||
self.assertEqual([assets[2].id], tab.original_list.asset_ids())
|
||
self.assertEqual([], tab.original_list.checked_asset_ids())
|
||
self.assertIn("选中的2张", confirmations[0][1])
|
||
self.assertIn("下一张图片将成为主图", confirmations[0][1])
|
||
self.assertIn("不会删除蝦皮线上图片", confirmations[0][1])
|
||
self.assertEqual(("已移除2张商品原图", "success"), statuses[-1])
|
||
|
||
messages = []
|
||
tab._message = lambda title, message, **kwargs: messages.append((title, message))
|
||
state.download_queue = [assets[2].id]
|
||
tab.delete_originals([assets[2].id])
|
||
self.assertIn("仍在后台下载", messages[-1][1])
|
||
self.assertIsNotNone(image_studio.get_asset(assets[2].id, path=config["db_path"]))
|
||
|
||
class RunningWorker:
|
||
def cancel(self):
|
||
return None
|
||
|
||
state.download_queue = []
|
||
state.worker = RunningWorker()
|
||
tab.delete_originals([assets[2].id])
|
||
self.assertEqual(("生成中不能删除当前任务的商品原图", "warning"), statuses[-1])
|
||
self.assertIsNotNone(image_studio.get_asset(assets[2].id, path=config["db_path"]))
|
||
state.worker = None
|
||
|
||
self.assert_removed(temp_dir)
|
||
|
||
|
||
if __name__ == "__main__":
|
||
unittest.main()
|