import unittest from pathlib import Path from tempfile import TemporaryDirectory from unittest.mock import Mock, patch from upload_video_to_guanghe import ( GuangHeUploader, _match_tm_item_ids_from_product_titles, download_videos, ) class GuangHeMetadataTests(unittest.TestCase): 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): output_dir = Path(args[args.index("--output") + 1]) output_dir.mkdir(parents=True, exist_ok=True) (output_dir / attachment["name"]).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.name, "record-one") self.assertEqual(second[0].parent.name, "record-two") 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.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_hermes_repeats(self): uploader = object.__new__(GuangHeUploader) repeated = "光影行星宙斯双肩包科学分区收纳轻量通勤出行" with ( patch.object(uploader, "_search_product_features", return_value=""), patch("upload_video_to_guanghe.HAS_HERMES", True), patch("upload_video_to_guanghe._call_hermes", 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_hermes_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.HAS_HERMES", True), patch("upload_video_to_guanghe._call_hermes", 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.HAS_HERMES", True), patch("upload_video_to_guanghe._call_hermes", 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.HAS_HERMES", True), patch( "upload_video_to_guanghe._call_hermes", return_value=("光影行星宙斯双肩包大容量笔记本电脑轻量通勤", {}), ), ): title = uploader._generate_title(["宙斯"]) self.assertTrue(title.startswith("宙斯双肩包,")) self.assertNotIn("包袋双肩包", title) def test_generated_poseidon_title_cannot_be_changed_to_backpack_by_hermes(self): uploader = object.__new__(GuangHeUploader) with ( patch.object( uploader, "_search_product_features", return_value="双肩包、电脑包、托特包、大容量", ), patch("upload_video_to_guanghe.HAS_HERMES", True), patch( "upload_video_to_guanghe._call_hermes", 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_labels = Mock( side_effect=lambda labels: calls.append(("labels", list(labels))) ) uploader._pick_topic = Mock( side_effect=lambda topic: calls.append(("topic", topic)) ) uploader._snapshot = Mock() uploader._prepare_publish_metadata(product_names) self.assertEqual( calls, [ ("labels", 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_fixed_topic_is_not_a_recommended_first_result(self): self.assertEqual(GuangHeUploader.FIXED_TOPIC, "在淘宝种草一夏") if __name__ == "__main__": unittest.main()