"""Tests for AI outfit Excel service. Pure service tests: no GUI dependency. """ import shutil import sys import tempfile import unittest from pathlib import Path from unittest.mock import patch from openpyxl import Workbook, load_workbook sys.path.insert(0, str(Path(__file__).parent.parent / "src")) class TestOutfitExcelService(unittest.TestCase): def setUp(self): self.tmp = Path(tempfile.mkdtemp()) self.excel_path = self.tmp / "outfit.xlsx" self._write_workbook() def tearDown(self): shutil.rmtree(str(self.tmp), ignore_errors=True) def _write_workbook(self): wb = Workbook() ws = wb.active ws.append(["标题", "货号", "衣服图", "结果图", "状态", "原因"]) ws.append(["T恤", "TY001", r"D:\img\ty001.png", "", "", ""]) ws.append(["已完成", "TY002", r"D:\img\ty002.png", r"D:\out\ty002.jpg", "完成", ""]) ws.append(["失败行", "TY003", r"D:\img\ty003.png", "", "失败", "旧错误"]) ws.append(["缺路径", "TY004", "", "", "", ""]) ws.append([" 保留空格 ", 1005, r"D:\img\ty005.png", "", "", ""]) wb.save(str(self.excel_path)) wb.close() def _read_row(self, row): wb = load_workbook(str(self.excel_path)) try: ws = wb.active return ( ws.cell(row, 4).value, ws.cell(row, 5).value, ws.cell(row, 6).value, ) finally: wb.close() def test_load_tasks_skips_done_failed_and_incomplete_by_default(self): from services.excel_service import load_outfit_tasks tasks = load_outfit_tasks(self.excel_path) self.assertEqual([task.row_index for task in tasks], [2, 6]) self.assertEqual(tasks[0].title, "T恤") self.assertEqual(tasks[0].product_id, "TY001") self.assertEqual(tasks[0].garment_path, r"D:\img\ty001.png") def test_load_tasks_includes_failed_when_retry_enabled(self): from services.excel_service import load_outfit_tasks tasks = load_outfit_tasks(self.excel_path, retry_failed=True) self.assertEqual([task.row_index for task in tasks], [2, 4, 6]) self.assertEqual(tasks[1].status, "失败") def test_load_tasks_preserves_user_cell_text(self): from services.excel_service import load_outfit_tasks tasks = load_outfit_tasks(self.excel_path) task = tasks[-1] self.assertEqual(task.title, " 保留空格 ") self.assertEqual(task.product_id, "1005") def test_write_success_result_updates_d_e_f_and_saves(self): from core.models import OutfitResult, OutfitTask from services.excel_service import write_outfit_result task = OutfitTask(row_index=2, title="T恤", product_id="TY001", garment_path="g.png") result = OutfitResult(task=task, success=True, output_path=r"D:\out\TY001.jpg") write_outfit_result(self.excel_path, result) self.assertEqual(self._read_row(2), (r"D:\out\TY001.jpg", "完成", None)) def test_write_failure_result_updates_e_f_and_saves(self): from core.models import OutfitResult, OutfitTask from services.excel_service import write_outfit_result task = OutfitTask(row_index=2, title="T恤", product_id="TY001", garment_path="g.png") result = OutfitResult(task=task, success=False, error="衣服图打不开") write_outfit_result(self.excel_path, result) self.assertEqual(self._read_row(2), (None, "失败", "衣服图打不开")) def test_read_all_rows_includes_done_failed_skips_incomplete(self): from services.excel_service import read_all_rows rows = read_all_rows(self.excel_path) # rows 2(待处理) 3(完成) 4(失败) 6(待处理) — row 5 (空衣服图) skipped self.assertEqual([r.row_index for r in rows], [2, 3, 4, 6]) by_idx = {r.row_index: r for r in rows} self.assertEqual(by_idx[3].status, "完成") self.assertEqual(by_idx[4].status, "失败") self.assertEqual(by_idx[6].title, " 保留空格 ") # raw text preserved def test_empty_product_id_rows_are_included(self): """货号(B)可空：只要求 标题 + 衣服图路径（docs/11 §4）。""" from services.excel_service import load_outfit_tasks, read_all_rows p = self.tmp / "nopid.xlsx" wb = Workbook() ws = wb.active ws.append(["标题", "货号", "衣服图", "结果", "状态", "原因"]) ws.append(["有标题无货号", "", r"D:\img\a.png", "", "", ""]) # 收录 ws.append(["缺图也缺货号", "", "", "", "", ""]) # 缺图 → 仍跳过 wb.save(str(p)) wb.close() tasks = load_outfit_tasks(p) self.assertEqual([t.row_index for t in tasks], [2]) self.assertEqual(tasks[0].product_id, "") self.assertEqual([r.row_index for r in read_all_rows(p)], [2]) def test_read_all_rows_require_title_false_keeps_empty_title_rows(self): """§19.27: 标题生成的行来源——A 空、C 非空的行 require_title=False 收录， 默认(True)跳过；C 空的行两种都跳过。""" from services.excel_service import read_all_rows p = self.tmp / "notitle.xlsx" wb = Workbook() ws = wb.active ws.append(["标题", "商品id", "原始图片路径", "结果", "状态", "原因"]) ws.append([None, None, "D:\\out\\TY030\\", "", "", ""]) # A空、C有（印花表） ws.append([None, None, "D:\\out\\TY037\\", "", "", ""]) # A空、C有 ws.append([None, None, "", "", "", ""]) # C空 → 两种都跳过 wb.save(str(p)) wb.close() # 默认要求标题 → A 全空 → 跳过印花行（旧 catch-22 现象） self.assertEqual([r.row_index for r in read_all_rows(p)], []) # require_title=False → A 空也收录，只按 C 过滤 rows = read_all_rows(p, require_title=False) self.assertEqual([r.row_index for r in rows], [2, 3]) self.assertEqual(rows[0].title, "") # A 留空（待生成） self.assertTrue(rows[0].garment_path.endswith("TY030" + "\\")) def test_write_title_result_updates_a_only_and_saves(self): """标题生成回填只改 A 列，不动 D/E/F（docs/11 §17）。""" from services.excel_service import write_title_result # Row 3 starts as 完成 with D/E filled; writing a title must not touch them. write_title_result(self.excel_path, 3, "AI 生成的电商标题") wb = load_workbook(str(self.excel_path)) try: ws = wb.active self.assertEqual(ws.cell(3, 1).value, "AI 生成的电商标题") # A 改写 self.assertEqual(ws.cell(3, 4).value, r"D:\out\ty002.jpg") # D 不动 self.assertEqual(ws.cell(3, 5).value, "完成") # E 不动 finally: wb.close() def test_check_excel_writable_true_for_existing_workbook(self): from services.excel_service import check_excel_writable self.assertTrue(check_excel_writable(self.excel_path)) def test_ensure_excel_writable_raises_when_probe_fails(self): from services.excel_service import ExcelInUseError, ensure_excel_writable with patch("builtins.open", side_effect=OSError("locked")): with self.assertRaises(ExcelInUseError): ensure_excel_writable(self.excel_path) # -- write_outfit_source_excel (docs/02 §6.12) ---------------------------- def test_write_outfit_source_excel_header_and_rows(self): """One row per (print_name, subdir_path); header + A/B/C values correct.""" import os from services.excel_service import write_outfit_source_excel p = self.tmp / "batch.xlsx" rows = [ ("floral", r"D:\out\20260623\floral" + os.sep), ("stripe", r"D:\out\20260623\stripe" + os.sep), ] write_outfit_source_excel(p, rows) wb = load_workbook(str(p)) ws = wb.active try: self.assertEqual( [ws.cell(1, c).value for c in range(1, 8)], ["标题", "商品id", "原始图片路径", "生成结果图片路径", "完成状态", "失败原因", "货憨憨上传状态"], ) self.assertEqual(ws.max_row, 3) # 1 header + 2 data self.assertEqual(ws.cell(2, 1).value, "floral") self.assertIsNone(ws.cell(2, 2).value) # 货号留空 self.assertIn("floral", ws.cell(2, 3).value) self.assertTrue(ws.cell(2, 3).value.endswith(os.sep)) self.assertEqual(ws.cell(3, 1).value, "stripe") for col in (4, 5, 6): # D/E/F 留空 self.assertIsNone(ws.cell(2, col).value) finally: wb.close() def test_write_outfit_source_excel_ai_outfit_can_read_it(self): """AI穿搭 read_all_rows picks up the generated Excel as a directory row.""" import os from services.excel_service import read_all_rows, write_outfit_source_excel subdir = self.tmp / "floral" subdir.mkdir() p = self.tmp / "batch.xlsx" write_outfit_source_excel(p, [("floral", str(subdir) + os.sep)]) rows = read_all_rows(p) self.assertEqual(len(rows), 1) self.assertEqual(rows[0].title, "floral") self.assertEqual(rows[0].product_id, "") self.assertTrue(rows[0].garment_path.endswith(os.sep)) if __name__ == "__main__": unittest.main()