mirror of
https://github.com/jxxghp/MoviePilot.git
synced 2026-07-06 07:11:26 +08:00
feat: add skills marketplace management
This commit is contained in:
jxxghp
183
tests/test_skills_command.py
Normal file
183
tests/test_skills_command.py
Normal file
@@ -0,0 +1,183 @@
|
||||
import io
|
||||
import sys
|
||||
import tempfile
|
||||
import unittest
|
||||
import zipfile
|
||||
from pathlib import Path
|
||||
from types import ModuleType
|
||||
from unittest.mock import patch
|
||||
|
||||
sys.modules.setdefault("qbittorrentapi", ModuleType("qbittorrentapi"))
|
||||
setattr(sys.modules["qbittorrentapi"], "TorrentFilesList", list)
|
||||
sys.modules.setdefault("transmission_rpc", ModuleType("transmission_rpc"))
|
||||
setattr(sys.modules["transmission_rpc"], "File", object)
|
||||
sys.modules.setdefault("psutil", ModuleType("psutil"))
|
||||
|
||||
from app.chain.message import MessageChain
|
||||
from app.chain.skills import SkillsChain, skills_interaction_manager
|
||||
from app.helper.skill import SkillHelper, SkillInfo
|
||||
from app.schemas.types import MessageChannel
|
||||
|
||||
|
||||
def _build_skill_zip(skill_dir: str, skill_name: str) -> bytes:
|
||||
buf = io.BytesIO()
|
||||
with zipfile.ZipFile(buf, "w") as zf:
|
||||
zf.writestr(
|
||||
f"demo-main/{skill_dir}/SKILL.md",
|
||||
(
|
||||
f"---\n"
|
||||
f"name: {skill_name}\n"
|
||||
f"version: 1\n"
|
||||
f"description: demo skill\n"
|
||||
f"---\n\n"
|
||||
f"# {skill_name}\n"
|
||||
),
|
||||
)
|
||||
zf.writestr(f"demo-main/{skill_dir}/scripts/example.py", "print('ok')\n")
|
||||
return buf.getvalue()
|
||||
|
||||
|
||||
class TestSkillsCommand(unittest.TestCase):
|
||||
def tearDown(self):
|
||||
skills_interaction_manager.clear()
|
||||
|
||||
def test_message_routes_text_reply_to_skills_interaction_before_ai(self):
|
||||
chain = MessageChain()
|
||||
skills_interaction_manager.create_or_replace(
|
||||
user_id="10001",
|
||||
channel=MessageChannel.Wechat,
|
||||
source="wechat-test",
|
||||
username="tester",
|
||||
)
|
||||
|
||||
with patch.object(chain, "_record_user_message"), patch(
|
||||
"app.chain.message.SkillsChain.handle_text_interaction",
|
||||
return_value=True,
|
||||
) as handle_text, patch.object(chain, "_handle_ai_message") as handle_ai:
|
||||
chain.handle_message(
|
||||
channel=MessageChannel.Wechat,
|
||||
source="wechat-test",
|
||||
userid="10001",
|
||||
username="tester",
|
||||
text="2",
|
||||
)
|
||||
|
||||
handle_text.assert_called_once()
|
||||
handle_ai.assert_not_called()
|
||||
|
||||
def test_callback_routes_to_skills_chain(self):
|
||||
chain = MessageChain()
|
||||
request = skills_interaction_manager.create_or_replace(
|
||||
user_id="10001",
|
||||
channel=MessageChannel.Telegram,
|
||||
source="telegram-test",
|
||||
username="tester",
|
||||
)
|
||||
|
||||
with patch(
|
||||
"app.chain.message.SkillsChain.handle_callback_interaction",
|
||||
return_value=True,
|
||||
) as handle_callback:
|
||||
chain._handle_callback(
|
||||
text=f"CALLBACK:skills:{request.request_id}:market",
|
||||
channel=MessageChannel.Telegram,
|
||||
source="telegram-test",
|
||||
userid="10001",
|
||||
username="tester",
|
||||
)
|
||||
|
||||
handle_callback.assert_called_once()
|
||||
|
||||
def test_skillhelper_install_and_remove_market_skill(self):
|
||||
helper = SkillHelper()
|
||||
skill = SkillInfo(
|
||||
id="demo-skill",
|
||||
name="demo-skill",
|
||||
description="demo",
|
||||
source_type="market",
|
||||
source_label="市场 · acme/demo",
|
||||
repo_url="https://github.com/acme/demo",
|
||||
repo_name="acme/demo",
|
||||
skill_path="skills/demo-skill",
|
||||
)
|
||||
zip_bytes = _build_skill_zip("skills/demo-skill", "demo-skill")
|
||||
|
||||
with tempfile.TemporaryDirectory() as tempdir:
|
||||
user_root = Path(tempdir) / "user-skills"
|
||||
bundled_root = Path(tempdir) / "bundled-skills"
|
||||
user_root.mkdir(parents=True, exist_ok=True)
|
||||
bundled_root.mkdir(parents=True, exist_ok=True)
|
||||
|
||||
with patch.object(
|
||||
SkillHelper, "get_user_skills_dir", return_value=user_root
|
||||
), patch.object(
|
||||
SkillHelper, "get_bundled_skills_dir", return_value=bundled_root
|
||||
), patch.object(
|
||||
helper, "_download_repo_archive", return_value=zip_bytes
|
||||
):
|
||||
success, message = helper.install_market_skill(skill)
|
||||
self.assertTrue(success, message)
|
||||
self.assertTrue((user_root / "demo-skill" / "SKILL.md").exists())
|
||||
self.assertTrue(
|
||||
(user_root / "demo-skill" / ".moviepilot-skill-source.json").exists()
|
||||
)
|
||||
|
||||
local_skills = helper.list_local_skills()
|
||||
self.assertEqual(len(local_skills), 1)
|
||||
self.assertEqual(local_skills[0].source_type, "market")
|
||||
self.assertTrue(local_skills[0].removable)
|
||||
|
||||
removed, remove_message = helper.remove_local_skill("demo-skill")
|
||||
self.assertTrue(removed, remove_message)
|
||||
self.assertFalse((user_root / "demo-skill").exists())
|
||||
|
||||
bundled_skill_dir = bundled_root / "builtin-skill"
|
||||
bundled_skill_dir.mkdir(parents=True, exist_ok=True)
|
||||
(bundled_skill_dir / "SKILL.md").write_text(
|
||||
"---\nname: builtin-skill\ndescription: builtin\n---\n",
|
||||
encoding="utf-8",
|
||||
)
|
||||
installed_builtin = user_root / "builtin-skill"
|
||||
installed_builtin.mkdir(parents=True, exist_ok=True)
|
||||
(installed_builtin / "SKILL.md").write_text(
|
||||
"---\nname: builtin-skill\ndescription: builtin\n---\n",
|
||||
encoding="utf-8",
|
||||
)
|
||||
|
||||
removed, remove_message = helper.remove_local_skill("builtin-skill")
|
||||
self.assertFalse(removed)
|
||||
self.assertIn("内置技能", remove_message)
|
||||
|
||||
def test_skills_chain_updates_buttons_via_edit_message(self):
|
||||
chain = SkillsChain()
|
||||
buttons = [[{"text": "安装 1", "callback_data": "skills:req:install:1"}]]
|
||||
|
||||
with patch.object(chain, "edit_message", return_value=True) as edit_message, patch.object(
|
||||
chain, "post_message"
|
||||
) as post_message:
|
||||
chain._update_or_post_message(
|
||||
channel=MessageChannel.Telegram,
|
||||
source="telegram-test",
|
||||
userid="10001",
|
||||
username="tester",
|
||||
title="技能市场",
|
||||
text="请选择技能",
|
||||
buttons=buttons,
|
||||
original_message_id=123,
|
||||
original_chat_id="456",
|
||||
)
|
||||
|
||||
edit_message.assert_called_once_with(
|
||||
channel=MessageChannel.Telegram,
|
||||
source="telegram-test",
|
||||
message_id=123,
|
||||
chat_id="456",
|
||||
title="技能市场",
|
||||
text="请选择技能",
|
||||
buttons=buttons,
|
||||
)
|
||||
post_message.assert_not_called()
|
||||
|
||||
|
||||
if __name__ == "__main__":
|
||||
unittest.main()
|
||||
Reference in New Issue
Block a user