添加技能包测试

tests/test_generate_image.py

@@ -0,0 +1,72 @@
+import pathlib
+import sys
+import unittest
+
+SCRIPT_DIR = pathlib.Path(__file__).resolve().parents[1] / "scripts"
+SKILL_DIR = SCRIPT_DIR.parent
+sys.path.insert(0, str(SCRIPT_DIR))
+
+try:
+    from generate_image import DEFAULT_API_URL, extract_image_bytes, images_endpoint, png_metadata
+except ModuleNotFoundError:
+    DEFAULT_API_URL = None
+    extract_image_bytes = None
+    images_endpoint = None
+    png_metadata = None
+
+
+class ExtractImageBytesTests(unittest.TestCase):
+    def test_decodes_inline_base64_image(self):
+        self.assertIsNotNone(extract_image_bytes, "generate_image.extract_image_bytes is required")
+        payload = {"data": [{"b64_json": "aGVsbG8="}]}
+        self.assertEqual(extract_image_bytes(payload), b"hello")
+
+    def test_downloads_url_response_through_injected_reader(self):
+        self.assertIsNotNone(extract_image_bytes, "generate_image.extract_image_bytes is required")
+        payload = {"data": [{"url": "https://images.example.test/output.png"}]}
+        self.assertEqual(extract_image_bytes(payload, download_url=lambda url: b"png-bytes"), b"png-bytes")
+
+    def test_rejects_unknown_image_payload(self):
+        self.assertIsNotNone(extract_image_bytes, "generate_image.extract_image_bytes is required")
+        with self.assertRaisesRegex(ValueError, "b64_json or url"):
+            extract_image_bytes({"data": [{}]})
+
+
+class ImageMetadataTests(unittest.TestCase):
+    def test_uses_pinai_as_the_default_api_url(self):
+        self.assertEqual(DEFAULT_API_URL, "https://api.pinaic.com/v1")
+
+    def test_uses_a_base_url_or_complete_endpoint(self):
+        self.assertIsNotNone(images_endpoint, "generate_image.images_endpoint is required")
+        self.assertEqual(
+            images_endpoint("https://provider.example/v1"),
+            "https://provider.example/v1/images/generations",
+        )
+        self.assertEqual(
+            images_endpoint("https://provider.example/v1/images/generations/"),
+            "https://provider.example/v1/images/generations",
+        )
+
+    def test_reads_png_dimensions_and_color_mode(self):
+        self.assertIsNotNone(png_metadata, "generate_image.png_metadata is required")
+        png = b"\x89PNG\r\n\x1a\n" + b"\x00\x00\x00\rIHDR" + b"\x00\x00\x04\x00\x00\x00\x02\x00\x08\x06\x00\x00\x00"
+        self.assertEqual(png_metadata(png), (1024, 512, 8, "RGBA"))
+
+
+class KeySetupPackageTests(unittest.TestCase):
+    def test_includes_a_double_click_key_setup_launcher(self):
+        launcher = SKILL_DIR / "设置 PinAI Key.cmd"
+        setup_script = SCRIPT_DIR / "setup_pinai_key.ps1"
+        self.assertTrue(launcher.is_file())
+        self.assertTrue(setup_script.is_file())
+        self.assertIn("setup_pinai_key.ps1", launcher.read_text(encoding="utf-8"))
+
+    def test_key_setup_uses_a_masked_local_input_and_user_variable(self):
+        setup_script = (SCRIPT_DIR / "setup_pinai_key.ps1").read_text(encoding="utf-8")
+        self.assertIn("UseSystemPasswordChar = $true", setup_script)
+        self.assertIn("SetEnvironmentVariable('PINAI_API_KEY', $key, 'User')", setup_script)
+        self.assertNotIn("Write-Output $key", setup_script)
+
+
+if __name__ == "__main__":
+    unittest.main()