wehub-resource-sync
93 lines
3.1 KiB
Python
93 lines
3.1 KiB
Python
import importlib.util
|
||
import os
|
||
import tempfile
|
||
import unittest
|
||
from pathlib import Path
|
||
from unittest import mock
|
||
|
||
|
||
def _load_skill_manager(openclaw_home, hub_base=None):
|
||
root = Path(__file__).resolve().parents[1]
|
||
script_path = root / "scripts" / "skill_manager.py"
|
||
|
||
env = {"OPENCLAW_HOME": str(openclaw_home)}
|
||
if hub_base is not None:
|
||
env["OPENCLAW_SKILLS_HUB_BASE"] = hub_base
|
||
|
||
spec = importlib.util.spec_from_file_location("skill_manager_under_test", script_path)
|
||
module = importlib.util.module_from_spec(spec)
|
||
with mock.patch.dict(os.environ, env, clear=False):
|
||
if hub_base is None:
|
||
os.environ.pop("OPENCLAW_SKILLS_HUB_BASE", None)
|
||
spec.loader.exec_module(module)
|
||
return module
|
||
|
||
|
||
class SkillManagerTests(unittest.TestCase):
|
||
def test_default_skills_do_not_use_removed_openclaw_hub(self):
|
||
with tempfile.TemporaryDirectory() as tmp:
|
||
skill_manager = _load_skill_manager(Path(tmp) / ".openclaw")
|
||
|
||
self.assertIn("mmx_cli", skill_manager.OFFICIAL_SKILLS_HUB)
|
||
self.assertTrue(
|
||
all(
|
||
"openclaw-ai/skills-hub" not in url
|
||
for url in skill_manager.OFFICIAL_SKILLS_HUB.values()
|
||
)
|
||
)
|
||
|
||
def test_custom_hub_base_restores_hub_skill_urls(self):
|
||
with tempfile.TemporaryDirectory() as tmp:
|
||
skill_manager = _load_skill_manager(
|
||
Path(tmp) / ".openclaw",
|
||
hub_base="https://example.com/openclaw-skills",
|
||
)
|
||
|
||
self.assertEqual(
|
||
skill_manager.OFFICIAL_SKILLS_HUB["code_review"],
|
||
"https://example.com/openclaw-skills/code_review/SKILL.md",
|
||
)
|
||
self.assertEqual(
|
||
skill_manager.OFFICIAL_SKILLS_HUB["test_framework"],
|
||
"https://example.com/openclaw-skills/test_framework/SKILL.md",
|
||
)
|
||
self.assertEqual(
|
||
skill_manager.OFFICIAL_SKILLS_HUB["mmx_cli"],
|
||
"https://raw.githubusercontent.com/MiniMax-AI/cli/main/skill/SKILL.md",
|
||
)
|
||
|
||
def test_import_official_hub_uses_per_skill_recommended_agents(self):
|
||
with tempfile.TemporaryDirectory() as tmp:
|
||
skill_manager = _load_skill_manager(Path(tmp) / ".openclaw")
|
||
|
||
calls = []
|
||
|
||
def fake_add_remote(agent_id, skill_name, source_url, description=""):
|
||
calls.append((agent_id, skill_name, source_url, description))
|
||
return True
|
||
|
||
with mock.patch.object(skill_manager, "add_remote", fake_add_remote):
|
||
self.assertTrue(skill_manager.import_official_hub([]))
|
||
|
||
self.assertEqual(
|
||
calls,
|
||
[
|
||
(
|
||
"menxia",
|
||
"mmx_cli",
|
||
"https://raw.githubusercontent.com/MiniMax-AI/cli/main/skill/SKILL.md",
|
||
"默认 skill：mmx_cli",
|
||
),
|
||
(
|
||
"shangshu",
|
||
"mmx_cli",
|
||
"https://raw.githubusercontent.com/MiniMax-AI/cli/main/skill/SKILL.md",
|
||
"默认 skill：mmx_cli",
|
||
),
|
||
],
|
||
)
|
||
|
||
|
||
if __name__ == "__main__":
|
||
unittest.main()
|