import unittest from pathlib import Path from tempfile import TemporaryDirectory from unittest.mock import Mock, patch from upload_video_to_guanghe import ( GuangHeUploader, TmallTitleGenerationError, _match_tm_item_ids_from_product_titles, _select_video_product_names, _tm_product_title_search_pattern, download_videos, ) def test_tm_product_title_search_pattern_uses_model_token() -> None: assert _tm_product_title_search_pattern("拾影单肩包") == "%拾影%" assert _tm_product_title_search_pattern("星云mini") == "%星云mini%" def test_multi_style_record_prefers_style_named_by_attachment() -> None: assert _select_video_product_names( ["阿波罗2", "宙斯3"], "宙斯3 7.30.mp4" ) == ["宙斯3"] assert _select_video_product_names( ["阿波罗2", "宙斯3"], "7月30日.mp4" ) == ["阿波罗2", "宙斯3"] def test_attachment_style_matching_handles_combos_and_overlapping_aliases() -> None: assert _select_video_product_names( ["阿波罗2", "宙斯3", "星云2", "逐星"], "宙斯3+阿波罗2+星云2+逐星.m4v", ) == ["宙斯3", "阿波罗2", "星云2", "逐星"] assert _select_video_product_names( ["极星双肩", "极星双肩2"], "极星双肩2 新新新.mp4" ) == ["极星双肩2"] class GuangHeMetadataTests(unittest.TestCase): def test_formal_title_rejection_uses_non_browser_failure_type(self): GuangHeUploader.configure_catalog_product_titles( {"极星双肩2": ["极星双肩2双肩包轻便电脑包"]} ) uploader = GuangHeUploader("", "", require_llm_titles=True) try: with patch( "upload_video_to_guanghe.JdVideoTitleGenerator.generate", side_effect=RuntimeError("候选不合格"), ): with self.assertRaisesRegex( TmallTitleGenerationError, "候选不合格" ): uploader._generate_title(["极星双肩2"], video_key="token") finally: GuangHeUploader.configure_catalog_product_titles({}) def test_formal_collection_title_forbids_specific_style_names(self): GuangHeUploader.configure_catalog_product_titles( { "宙斯3": ["宙斯3双肩包男士大容量"], "阿波罗2": ["阿波罗2电脑包手提轻便"], } ) uploader = GuangHeUploader("", "", require_llm_titles=True) try: with patch( "upload_video_to_guanghe.JdVideoTitleGenerator" ) as generator_class: generator = generator_class.return_value generator.generate.return_value = ( "多款通勤包袋,大容量轻便按场景切换" ) title = uploader._generate_title( ["宙斯3", "阿波罗2"], video_key="combo-token" ) self.assertEqual(title, "多款通勤包袋,大容量轻便按场景切换") self.assertTrue(generator_class.call_args.kwargs["generic_collection"]) self.assertFalse(generator_class.call_args.kwargs["require_product_name"]) self.assertEqual(generator_class.call_args.kwargs["llm_timeout"], 300) generator.generate.assert_called_once() self.assertEqual( generator.generate.call_args.args[0], ["宙斯3", "阿波罗2"] ) finally: GuangHeUploader.configure_catalog_product_titles({}) def test_downloads_with_same_attachment_name_are_isolated_by_record(self): attachment = {"file_token": "file-token", "name": "same-name.mp4"} with TemporaryDirectory() as tmp_dir, patch( "upload_video_to_guanghe.DOWNLOAD_DIR", Path(tmp_dir) ), patch("upload_video_to_guanghe.run_lark") as run_lark_mock: def fake_download(args, *, cwd=None, timeout=None): self.assertEqual(timeout, 600) output_dir = Path(cwd) output_dir.mkdir(parents=True, exist_ok=True) (output_dir / args[args.index("--output") + 1]).write_bytes(b"video") return 0, "", "" run_lark_mock.side_effect = fake_download first = download_videos("record-one", [attachment]) second = download_videos("record-two", [attachment]) self.assertEqual(first[0].parent.parent.name, "record-one") self.assertEqual(second[0].parent.parent.name, "record-two") self.assertEqual(first[0].parent.name, "file-token") self.assertNotEqual(first[0], second[0]) def test_unresolved_style_uses_exact_database_product_title_match(self): result = _match_tm_item_ids_from_product_titles( ["宙斯3", "未建档款"], [ ( "986141740958", "GYXX/光影行星宙斯3双肩包男士背包机能大容量", ), ("655191144133", "GYXX光影行星宙斯双肩包男士电脑包"), ], ) self.assertEqual(result, {"宙斯3": ["986141740958"]}) def test_product_search_term_removes_series_suffix(self): self.assertEqual( GuangHeUploader._normalize_product_search_term("宙斯系列"), "宙斯", ) self.assertEqual( GuangHeUploader._normalize_product_search_term(" 宙斯 系列款 "), "宙斯", ) self.assertEqual( GuangHeUploader._normalize_product_search_term("星迹2"), "星迹2", ) def test_cover_generation_pending_marker_is_not_upload_done(self): self.assertTrue(GuangHeUploader._cover_generation_is_pending("封面生成中")) self.assertFalse(GuangHeUploader._cover_generation_is_pending("智能封面图")) def test_content_label_option_allows_browse_count_but_rejects_longer_label(self): self.assertTrue( GuangHeUploader._content_label_text_matches("# 斜挎包 41673031次浏览", "斜挎包") ) self.assertTrue( GuangHeUploader._content_label_text_matches("商品展示", "商品展示") ) self.assertTrue( GuangHeUploader._content_label_text_matches("#斜挎包", "斜挎包") ) self.assertFalse( GuangHeUploader._content_label_text_matches("# 单肩斜挎包 32871826次浏览", "斜挎包") ) def test_topic_result_card_requires_topic_and_activity_statistics(self): self.assertTrue( GuangHeUploader._topic_result_card_text_matches( "# 我的夏日焕新清单 小二推荐 作品数469.5万 参与数16.2万", "我的夏日焕新清单", ) ) self.assertTrue( GuangHeUploader._topic_result_card_text_matches( "# 秋上新开拍了 时尚 作品数198.3万 参与数11.3万", "秋上新", ) ) self.assertFalse( GuangHeUploader._topic_result_card_text_matches( "我的夏日焕新清单", "我的夏日焕新清单", ) ) self.assertTrue(GuangHeUploader._topic_card_box_is_clickable(620, 120)) self.assertFalse(GuangHeUploader._topic_card_box_is_clickable(620, 45)) self.assertFalse(GuangHeUploader._topic_card_box_is_clickable(720, 600)) def test_poseidon_edge_uses_authoritative_tote_category(self): self.assertEqual( GuangHeUploader._resolve_title_category( "波塞冬edge", "双肩包、电脑包、托特包、大容量", ), "托特包", ) self.assertEqual( GuangHeUploader._resolve_title_category("星迹2", ""), "斜挎包", ) def test_invalid_online_info_is_rejected_when_main_category_conflicts(self): result = GuangHeUploader._validate_online_product_info( "波塞冬edge", "主包型：双肩包；尺寸：39cm；容量/适配：16寸；材质：尼龙；卖点：分区", ) self.assertEqual(result, "") def test_verified_fallback_info_contains_all_product_fields(self): info = GuangHeUploader._verified_product_info("星迹2") for field in ["主包型：", "尺寸：", "容量/适配：", "材质：", "卖点："]: self.assertIn(field, info) self.assertIn("斜挎包", info) def test_title_category_uses_name_then_search_features(self): self.assertEqual( GuangHeUploader._resolve_title_category("盖亚斜挎", "轻量、大容量"), "斜挎包", ) self.assertEqual( GuangHeUploader._resolve_title_category("宙斯", "双肩包、大容量、笔电仓"), "双肩包", ) def test_title_normalization_keeps_search_structure_and_uses_computer_bag_term(self): title = GuangHeUploader._normalize_generated_title( "宙斯笔电大容量通勤真能装", ["宙斯"], "双肩包", ) self.assertTrue(title.startswith("宙斯双肩包,")) self.assertTrue(title.endswith("!")) self.assertIn("电脑包", title) self.assertNotIn("笔电", title) self.assertNotIn("笔记本电脑", title) self.assertNotIn("电脑包仓", title) self.assertGreaterEqual(len(title), 26) self.assertLessEqual(len(title), 30) def test_title_normalization_removes_compact_duplicate_model_and_broken_tail(self): title = GuangHeUploader._normalize_generated_title( "光影行星宙斯zair双肩包,宙斯zair,大容量分区收纳,从容应", ["宙斯z air"], "双肩包", ) self.assertTrue(title.startswith("宙斯z air双肩包,")) self.assertEqual(title.replace(" ", "").count("宙斯zair"), 1) self.assertNotIn("从容应!", title) self.assertGreaterEqual(len(title), 26) self.assertLessEqual(len(title), 30) def test_same_style_multiple_videos_get_distinct_titles_when_model_repeats(self): uploader = object.__new__(GuangHeUploader) repeated = "光影行星宙斯双肩包科学分区收纳轻量通勤出行" with ( patch.object(uploader, "_search_product_features", return_value=""), patch("upload_video_to_guanghe.call_direct_llm", return_value=(repeated, {})), ): first = uploader._generate_title(["宙斯"], video_index=1, video_total=2) second = uploader._generate_title(["宙斯"], video_index=2, video_total=2) self.assertNotEqual(first, second) self.assertTrue(first.startswith("宙斯双肩包,")) self.assertTrue(second.startswith("宙斯双肩包,")) def test_same_style_repeated_model_copy_gets_low_similarity_bodies(self): uploader = object.__new__(GuangHeUploader) repeated = "光影行星宙斯双肩包,科学分区收纳,轻量通勤出行!" with ( patch.object(uploader, "_search_product_features", return_value=""), patch("upload_video_to_guanghe.call_direct_llm", return_value=(repeated, {})), ): titles = [ uploader._generate_title(["宙斯"], video_index=index, video_total=4) for index in range(1, 5) ] bodies = [GuangHeUploader._title_body(title) for title in titles] self.assertEqual(len(set(bodies)), 4) for index, body in enumerate(bodies): for other in bodies[index + 1:]: self.assertLess( GuangHeUploader._title_similarity(body, other), GuangHeUploader.TITLE_SIMILARITY_LIMIT, ) def test_different_styles_cannot_reuse_same_title_skeleton(self): uploader = object.__new__(GuangHeUploader) repeated = "科学分区收纳,轻装通勤出行" with ( patch.object(uploader, "_search_product_features", return_value=""), patch("upload_video_to_guanghe.call_direct_llm", return_value=(repeated, {})), ): first = uploader._generate_title(["宙斯"], video_index=1) second = uploader._generate_title(["星云2"], video_index=1) first_body = GuangHeUploader._title_body(first) second_body = GuangHeUploader._title_body(second) self.assertLess( GuangHeUploader._title_similarity(first_body, second_body), GuangHeUploader.TITLE_SIMILARITY_LIMIT, ) def test_title_similarity_ignores_product_prefix(self): first = "宙斯双肩包,科学分区收纳,轻装通勤出行!" second = "星云2斜挎包,科学分区收纳,轻装通勤出行!" self.assertEqual(GuangHeUploader._title_similarity(first, second), 1.0) def test_full_batch_can_claim_48_globally_distinct_title_bodies(self): uploader = object.__new__(GuangHeUploader) titles = [] for index in range(48): product_names = [f"款式{index % 12 + 1}"] video_index = index // 12 + 1 candidate = uploader._fallback_title( product_names, "双肩包", variant_index=video_index, ) titles.append( uploader._claim_unique_title( candidate, product_names, "双肩包", variant_index=video_index, ) ) self.assertEqual(len(set(titles)), 48) self.assertTrue(all(26 <= len(title) <= 30 for title in titles)) for index, title in enumerate(titles): for other in titles[index + 1:]: self.assertLess( GuangHeUploader._title_similarity(title, other), GuangHeUploader.TITLE_SIMILARITY_LIMIT, ) def test_two_store_uploaders_can_share_one_global_title_registry(self): shared_global = [] shared_styles = {} shared_usage = {} first_uploader = object.__new__(GuangHeUploader) second_uploader = object.__new__(GuangHeUploader) for uploader in (first_uploader, second_uploader): uploader._used_titles_global = shared_global uploader._used_titles_by_products = shared_styles uploader._title_variant_usage = shared_usage first = first_uploader._claim_unique_title( first_uploader._fallback_title(["宙斯"], "双肩包", 1), ["宙斯"], "双肩包", 1, ) second = second_uploader._claim_unique_title( second_uploader._fallback_title(["星云2"], "斜挎包", 1), ["星云2"], "斜挎包", 1, ) self.assertLess( GuangHeUploader._title_similarity(first, second), GuangHeUploader.TITLE_SIMILARITY_LIMIT, ) def test_product_search_specs_prefer_tm_item_ids_and_fallback_to_keyword(self): specs = GuangHeUploader._build_product_search_specs( ["星云2", "未建档系列款"], {"星云2": ["900176117674", "865283623976"]}, ) self.assertEqual( specs, [ { "product_name": "星云2", "search_term": "900176117674", "exact_id": True, }, { "product_name": "未建档系列款", "search_term": "未建档", "exact_id": False, }, { "product_name": "星云2", "search_term": "865283623976", "exact_id": True, }, ], ) def test_exact_id_searches_get_one_store_keyword_fallback(self): specs = GuangHeUploader._build_product_search_specs( ["极星双肩", "未建档款"], {"极星双肩": ["972285625931", "966753757238"]}, ) self.assertEqual( GuangHeUploader._build_exact_search_fallback_specs(specs), [ { "product_name": "极星双肩", "search_term": "极星双肩", "exact_id": False, } ], ) def test_title_fallback_contains_product_category_solution_and_result(self): title = GuangHeUploader._fallback_title(["盖亚斜挎"], "斜挎包") self.assertTrue(title.startswith("盖亚斜挎包,")) self.assertIn(",", title) self.assertTrue(title.endswith("!")) self.assertIn("钥匙卡包各归其位", title) self.assertIn("早高峰取物", title) self.assertGreaterEqual(title.count(","), 2) self.assertGreaterEqual(len(title), 26) self.assertLessEqual(len(title), 30) def test_generated_title_replaces_generic_category_with_model_category(self): uploader = object.__new__(GuangHeUploader) with ( patch.object(uploader, "_search_product_features", return_value=""), patch( "upload_video_to_guanghe.call_direct_llm", return_value=("光影行星宙斯双肩包大容量笔记本电脑轻量通勤", {}), ), ): title = uploader._generate_title(["宙斯"]) self.assertTrue(title.startswith("宙斯双肩包,")) self.assertNotIn("包袋双肩包", title) def test_generated_poseidon_title_cannot_be_changed_to_backpack_by_model(self): uploader = object.__new__(GuangHeUploader) with ( patch.object( uploader, "_search_product_features", return_value="双肩包、电脑包、托特包、大容量", ), patch( "upload_video_to_guanghe.call_direct_llm", return_value=( "光影行星波塞冬edge双肩包暴雨通勤防泼水电脑仓从容出行", {}, ), ), ): title = uploader._generate_title(["波塞冬edge"]) self.assertTrue(title.startswith("波塞冬edge托特包,")) self.assertNotIn("双肩包", title) self.assertTrue(title.endswith("!")) self.assertGreaterEqual(len(title), 26) self.assertLessEqual(len(title), 30) def test_primary_content_label_uses_first_product_bag_type(self): self.assertEqual( GuangHeUploader._content_label_for_products(["盖亚斜挎", "极星双肩"]), "斜挎包", ) self.assertEqual( GuangHeUploader._content_label_for_products(["盖亚微单"]), "相机包", ) self.assertEqual( GuangHeUploader._content_label_for_products(["波塞冬edge"]), "托特包", ) self.assertEqual( GuangHeUploader._content_label_for_products(["波塞冬手提电脑包"]), "电脑包", ) def test_catalog_product_titles_supply_style_category_and_verified_info(self): try: GuangHeUploader.configure_catalog_product_titles( { "星云2": [ "GYXX光影行星星云2斜挎包男士单肩包平板胸包", "光影行星星云斜挎包男款邮差单肩包", ], "宙斯": ["GYXX光影行星宙斯双肩包男大容量通勤背包"], "宙斯z air": ["光影行星宙斯zair双肩包电脑包通勤背包"], "黑曜石托特": ["光影行星黑曜石双肩托特多用包"], } ) self.assertEqual( GuangHeUploader._content_label_for_products(["星云2"]), "斜挎包", ) self.assertEqual( GuangHeUploader._resolve_title_category("宙斯"), "双肩包", ) self.assertIn( "主包型：斜挎包", GuangHeUploader._verified_product_info("星云2"), ) self.assertEqual( GuangHeUploader._content_label_for_products(["宙斯z air"]), "双肩包", ) self.assertEqual( GuangHeUploader._content_label_for_products(["黑曜石托特"]), "托特包", ) finally: GuangHeUploader.configure_catalog_product_titles({}) def test_metadata_selects_style_category_and_product_showcase_before_topic(self): cases = [ (["盖亚斜挎"], ["斜挎包", "商品展示"]), (["极星双肩"], ["双肩包", "商品展示"]), ] for product_names, expected_labels in cases: with self.subTest(product_names=product_names): uploader = object.__new__(GuangHeUploader) calls = [] uploader._pick_content_label = Mock( side_effect=lambda label: calls.append(("label", label)) ) uploader._pick_topic = Mock( side_effect=lambda topic: calls.append(("topic", topic)) ) uploader._snapshot = Mock() uploader._prepare_publish_metadata(product_names) self.assertEqual( calls, [ *[("label", label) for label in expected_labels], ("topic", "我的夏日焕新清单"), ], ) def test_fixed_content_labels_only_keep_product_showcase(self): self.assertEqual(GuangHeUploader.FIXED_CONTENT_LABELS, ("商品展示",)) self.assertNotIn("户外", GuangHeUploader.FIXED_CONTENT_LABELS) def test_missing_user_selected_topic_stops_metadata_preparation(self): uploader = object.__new__(GuangHeUploader) uploader._pick_content_label = Mock() uploader._pick_topic = Mock( side_effect=RuntimeError("没有找到固定话题: #我的夏日焕新清单") ) uploader._snapshot = Mock() with self.assertRaisesRegex(RuntimeError, "没有找到固定话题"): uploader._prepare_publish_metadata(["极星双肩"]) self.assertEqual( uploader._pick_content_label.call_args_list, [unittest.mock.call("双肩包"), unittest.mock.call("商品展示")], ) def test_optional_content_label_failures_do_not_skip_required_topic(self): uploader = object.__new__(GuangHeUploader) uploader.page = Mock() uploader._pick_content_label = Mock( side_effect=[ RuntimeError("内容标签没有精确匹配项: #双肩包"), RuntimeError("内容标签没有精确匹配项: #商品展示"), ] ) uploader._pick_topic = Mock() uploader._snapshot = Mock() uploader._prepare_publish_metadata(["极星双肩"]) self.assertEqual( uploader._pick_content_label.call_args_list, [unittest.mock.call("双肩包"), unittest.mock.call("商品展示")], ) uploader._pick_topic.assert_called_once_with("我的夏日焕新清单") def test_metadata_uses_operator_topic_keyword(self): uploader = object.__new__(GuangHeUploader) calls = [] uploader._pick_content_label = Mock() uploader._pick_topic = Mock(side_effect=lambda topic: calls.append(topic)) uploader._snapshot = Mock() uploader._prepare_publish_metadata(["极星双肩"], topic_keyword="秋上新") self.assertEqual(calls, ["秋上新"]) def test_fixed_topic_is_not_a_recommended_first_result(self): self.assertEqual(GuangHeUploader.FIXED_TOPIC, "我的夏日焕新清单") if __name__ == "__main__": unittest.main()