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,63 @@
+import tempfile
+import unittest
+from pathlib import Path
+from unittest.mock import MagicMock, patch
+
+from music_server.services.cache_uploaders import S3CacheTargetUploader, SFTPCacheTargetUploader
+
+
+class CacheUploaderTests(unittest.TestCase):
+    @patch("music_server.services.cache_uploaders.paramiko.SSHClient")
+    def test_sftp_uploader_upload_and_delete(self, ssh_client_class):
+        ssh_client = MagicMock()
+        sftp_client = MagicMock()
+        ssh_client.open_sftp.return_value = sftp_client
+        ssh_client_class.return_value = ssh_client
+
+        with tempfile.TemporaryDirectory() as tmpdir:
+            file_path = Path(tmpdir) / "song.flac"
+            file_path.write_bytes(b"flac-data")
+            uploader = SFTPCacheTargetUploader(
+                host="127.0.0.1",
+                port=22,
+                username="tester",
+                password="secret",
+                remote_root="/srv/music_server_cache",
+            )
+
+            uploader.upload_file(local_path=file_path, remote_key="music/song.flac")
+            uploader.delete_file(remote_key="music/song.flac")
+            uploader.test_connection()
+
+        self.assertEqual(3, ssh_client.connect.call_count)
+        sftp_client.put.assert_called_once_with(str(file_path), "/srv/music_server_cache/music/song.flac")
+        sftp_client.remove.assert_called_once_with("/srv/music_server_cache/music/song.flac")
+        sftp_client.listdir.assert_called_once_with("/srv/music_server_cache")
+
+    @patch("music_server.services.cache_uploaders._build_boto3_client")
+    def test_s3_uploader_upload_and_delete(self, build_client):
+        s3_client = MagicMock()
+        build_client.return_value = s3_client
+
+        with tempfile.TemporaryDirectory() as tmpdir:
+            file_path = Path(tmpdir) / "song.flac"
+            file_path.write_bytes(b"flac-data")
+            uploader = S3CacheTargetUploader(
+                bucket="music-cache",
+                region="ap-shanghai",
+                endpoint_url="https://s3.example",
+                access_key_id="AKIA",
+                secret_access_key="SECRET",
+            )
+
+            uploader.upload_file(local_path=file_path, remote_key="music/song.flac")
+            uploader.delete_file(remote_key="music/song.flac")
+            uploader.test_connection()
+
+        s3_client.upload_file.assert_called_once()
+        s3_client.delete_object.assert_called_once_with(Bucket="music-cache", Key="music/song.flac")
+        s3_client.head_bucket.assert_called_once_with(Bucket="music-cache")
+
+
+if __name__ == "__main__":
+    unittest.main()