fixed # 修复工作流

2026-10-29 21:50:01 +08:00 · 2026-10-23 19:19:01 +08:00
parent 685cff9293
commit dc17f7f65c
10 changed files with 135 additions and 2 deletions
@@ -0,0 +1,92 @@
+from concurrent.futures import ThreadPoolExecutor
+from io import BytesIO
+from pathlib import Path
+import tempfile
+import unittest
+from unittest.mock import patch
+
+from farm_render import bundle
+
+ROOT = Path(__file__).resolve().parents[1]
+ZIP = ROOT / 'src/farm_render/data/assets.zip'
+
+
+class BundleTests(unittest.TestCase):
+    def setUp(self):
+        self.temp = tempfile.TemporaryDirectory()
+        self.addCleanup(self.temp.cleanup)
+        self.root = Path(self.temp.name)
+        self.expected = bundle.expected_manifest()
+
+    def download(self, url, destination, limit):
+        content = ZIP.read_bytes()
+        self.assertEqual(limit, len(content))
+        destination.write(content)
+
+    def test_download_deduplicates_and_cache_hash_checked_each_start(self):
+        with patch.object(bundle, '_copy_remote',
+                          side_effect=self.download) as fetch:
+            with ThreadPoolExecutor(4) as pool:
+                results = list(pool.map(
+                    lambda _: bundle.ensure_assets(self.root), range(4),
+                ))
+            self.assertEqual(len(set(results)), 1)
+            self.assertEqual(fetch.call_count, 1)
+            path = results[0]
+            content = bytearray(path.read_bytes())
+            content[-1] ^= 1
+            path.write_bytes(content)
+            bundle.ensure_assets(self.root)
+            self.assertEqual(fetch.call_count, 2)
+            self.assertEqual(path.read_bytes(), ZIP.read_bytes())
+
+    def test_gitea_and_service_use_service_repository_and_configured_base(self):
+        with patch.object(bundle, '_copy_remote',
+                          side_effect=self.download) as fetch:
+            bundle.ensure_assets(self.root / 'gitea', 'Gitea')
+            self.assertEqual(fetch.call_args.args[0],
+                'https://gitea.diuse.work/Diuse/farm_service/'
+                'raw/branch/main/assets/assets.zip')
+            bundle.ensure_assets(self.root / 'service', 'farm_service',
+                                 'https://farm.example/prefix/')
+            self.assertEqual(fetch.call_args.args[0],
+                'https://farm.example/prefix/v1/assets/'
+                + self.expected['sha256'] + '/assets.zip')
+
+    def test_failed_download_leaves_no_partial_and_wrong_hash_is_rejected(self):
+        def corrupt(url, destination, limit):
+            data = bytearray(ZIP.read_bytes())
+            data[-1] ^= 1
+            destination.write(data)
+        with patch.object(bundle, '_copy_remote', side_effect=corrupt):
+            with self.assertRaisesRegex(ValueError, 'SHA256'):
+                bundle.ensure_assets(self.root)
+        self.assertFalse(list(self.root.rglob('assets.zip')))
+        self.assertFalse(list(self.root.rglob('*.download')))
+        with patch.object(bundle, '_copy_remote', side_effect=OSError()):
+            with self.assertRaises(OSError):
+                bundle.ensure_assets(self.root)
+        self.assertFalse(list(self.root.rglob('*.download')))
+
+    def test_invalid_source_and_service_configuration(self):
+        for source, url in [('unknown', ''), ('farm_service', ''),
+                            ('farm_service', 'http://farm.example'),
+                            ('Github', 'https://user:pass@farm.example'),
+                            ('Gitea', 'https://farm.example/?token=secret')]:
+            with self.subTest(source=source, url=url):
+                with self.assertRaises(ValueError):
+                    bundle.ensure_assets(self.root, source, url)
+
+    def test_download_enforces_response_limit_and_https_redirect(self):
+        class Response(BytesIO):
+            headers = {'Content-Length': '20'}
+            def geturl(self):
+                return 'https://farm.example/assets.zip'
+        with patch.object(bundle, 'urlopen', return_value=Response(b'x'*20)):
+            with self.assertRaisesRegex(ValueError, 'limit'):
+                bundle._copy_remote('https://farm.example/a', BytesIO(), 10)
+        response = Response(b'')
+        response.geturl = lambda: 'http://farm.example/a'
+        with patch.object(bundle, 'urlopen', return_value=response):
+            with self.assertRaises(ValueError):
+                bundle._copy_remote('https://farm.example/a', BytesIO(), 10)