Initial import: Music_Server, MusicFree, catalog-sync

2026-05-23 16:51:14 +08:00
commit 069af30dba
847 changed files with 179878 additions and 0 deletions
@@ -0,0 +1,121 @@
+import json
+import unittest
+from pathlib import Path
+from unittest.mock import patch
+
+
+FIXTURES_DIR = Path(__file__).resolve().parent / "fixtures"
+
+
+def load_fixture(name: str) -> str:
+    return (FIXTURES_DIR / name).read_text(encoding="utf-8")
+
+
+class CollectorParsingTests(unittest.TestCase):
+    def test_netease_playlist_square_parser_extracts_playlist_candidates(self):
+        from musicdl.catalogsync.collectors.netease import parse_playlist_square_html
+
+        items = parse_playlist_square_html(load_fixture("netease_playlist_square.html"))
+
+        self.assertEqual(2, len(items))
+        self.assertEqual("netease", items[0].platform)
+        self.assertEqual("7583298906", items[0].remote_id)
+        self.assertEqual("playlist_url", items[0].parse_strategy)
+        self.assertEqual("https://music.163.com/#/playlist?id=7583298906", items[0].url)
+        self.assertEqual(1230000, items[0].play_count)
+        self.assertEqual(4567, items[1].play_count)
+
+    def test_netease_toplist_parser_extracts_playlist_style_rankings(self):
+        from musicdl.catalogsync.collectors.netease import parse_toplist_payload
+
+        items = parse_toplist_payload(json.loads(load_fixture("netease_toplist.json")))
+
+        self.assertEqual(2, len(items))
+        self.assertEqual("19723756", items[0].remote_id)
+        self.assertEqual("netease_toplist", items[0].parse_strategy)
+        self.assertEqual("https://music.163.com/#/playlist?id=19723756", items[0].url)
+        self.assertEqual(7654321, items[0].play_count)
+        self.assertEqual(345678, items[1].play_count)
+
+    def test_qq_playlist_square_parser_extracts_playlist_candidates(self):
+        from musicdl.catalogsync.collectors.qq import parse_playlist_square_payload
+
+        items = parse_playlist_square_payload(json.loads(load_fixture("qq_playlist_square.json")))
+
+        self.assertEqual(2, len(items))
+        self.assertEqual("qq", items[0].platform)
+        self.assertEqual("7707261125", items[0].remote_id)
+        self.assertEqual("playlist_url", items[0].parse_strategy)
+        self.assertEqual("https://y.qq.com/n/ryqq/playlist/7707261125", items[0].url)
+
+    def test_qq_playlist_square_parser_extracts_collected_song_count_when_present(self):
+        from musicdl.catalogsync.collectors.qq import parse_playlist_square_payload
+
+        payload = {
+            "data": {
+                "list": [
+                    {
+                        "dissid": "7707261125",
+                        "dissname": "QQ Count Playlist",
+                        "songnum": 42,
+                        "creator": {"name": "Collector"},
+                    }
+                ]
+            }
+        }
+
+        items = parse_playlist_square_payload(payload)
+
+        self.assertEqual(1, len(items))
+        self.assertEqual(42, items[0].collected_song_count)
+
+    def test_qq_toplist_parser_marks_entries_for_special_detail_resolution(self):
+        from musicdl.catalogsync.collectors.qq import parse_toplist_payload
+
+        items = parse_toplist_payload(json.loads(load_fixture("qq_toplist.json")))
+
+        self.assertEqual(2, len(items))
+        self.assertEqual("4", items[0].remote_id)
+        self.assertEqual("qq_toplist", items[0].parse_strategy)
+        self.assertEqual("https://y.qq.com/n/ryqq/toplist/4", items[0].url)
+
+    def test_kuwo_playlist_square_parser_extracts_ssr_playlist_links(self):
+        from musicdl.catalogsync.collectors.kuwo import parse_playlist_square_html
+
+        items = parse_playlist_square_html(load_fixture("kuwo_playlist_square.html"))
+
+        self.assertEqual(2, len(items))
+        self.assertEqual("kuwo", items[0].platform)
+        self.assertEqual("3694434192", items[0].remote_id)
+        self.assertEqual("playlist_url", items[0].parse_strategy)
+        self.assertEqual("https://www.kuwo.cn/playlist_detail/3694434192", items[0].url)
+        self.assertEqual(3215000, items[0].play_count)
+        self.assertEqual(6789, items[1].play_count)
+
+    def test_kuwo_toplist_parser_extracts_rank_entries_from_ssr_state(self):
+        from musicdl.catalogsync.collectors.kuwo import parse_toplist_html
+
+        items = parse_toplist_html(load_fixture("kuwo_toplist.html"))
+
+        self.assertEqual(2, len(items))
+        self.assertEqual("489927", items[0].remote_id)
+        self.assertEqual("kuwo_toplist", items[0].parse_strategy)
+        self.assertEqual("16", items[0].metadata["sourceid"])
+        self.assertEqual("https://www.kuwo.cn/rankList?bangId=489927", items[0].url)
+        self.assertEqual(1234567, items[0].play_count)
+        self.assertEqual(765432, items[1].play_count)
+
+    def test_kuwo_toplist_parser_can_fallback_without_node(self):
+        from musicdl.catalogsync.collectors.kuwo import parse_toplist_html
+
+        with patch("musicdl.catalogsync.collectors.kuwo.subprocess.run", side_effect=RuntimeError("node missing")):
+            items = parse_toplist_html(load_fixture("kuwo_toplist.html"))
+
+        self.assertEqual(2, len(items))
+        self.assertEqual("489927", items[0].remote_id)
+        self.assertEqual("16", items[0].metadata["sourceid"])
+        self.assertEqual(1234567, items[0].play_count)
+
+
+if __name__ == "__main__":
+    unittest.main()