wehub-resource-sync
70 lines
1.9 KiB
Python
70 lines
1.9 KiB
Python
from unittest import mock
|
|
|
|
from mlflow.assistant.skill_installer import (
|
|
install_skills,
|
|
list_bundled_skills,
|
|
list_installed_skills,
|
|
)
|
|
|
|
|
|
def test_install_skills_copies_to_destination(tmp_path):
|
|
destination = tmp_path / "skills"
|
|
installed = install_skills(destination)
|
|
|
|
assert destination.exists()
|
|
assert "agent-evaluation" in installed
|
|
assert (destination / "agent-evaluation" / "SKILL.md").exists()
|
|
|
|
|
|
def test_install_skills_overwrites_existing(tmp_path):
|
|
destination = tmp_path / "skills"
|
|
destination.mkdir(parents=True)
|
|
|
|
install_skills(destination)
|
|
assert (destination / "agent-evaluation" / "SKILL.md").exists()
|
|
|
|
|
|
def test_list_installed_skills(tmp_path):
|
|
# Create mock installed skills
|
|
skill1 = tmp_path / "alpha-skill"
|
|
skill1.mkdir()
|
|
(skill1 / "SKILL.md").touch()
|
|
|
|
skill2 = tmp_path / "beta-skill"
|
|
skill2.mkdir()
|
|
(skill2 / "SKILL.md").touch()
|
|
|
|
skills = list_installed_skills(tmp_path)
|
|
|
|
assert skills == ["alpha-skill", "beta-skill"] # Sorted
|
|
|
|
|
|
def test_list_installed_skills_empty(tmp_path):
|
|
skills = list_installed_skills(tmp_path)
|
|
assert skills == []
|
|
|
|
|
|
def test_list_installed_skills_nonexistent_path(tmp_path):
|
|
nonexistent = tmp_path / "does-not-exist"
|
|
skills = list_installed_skills(nonexistent)
|
|
assert skills == []
|
|
|
|
|
|
def test_list_bundled_skills_returns_empty_when_package_missing():
|
|
with mock.patch(
|
|
"mlflow.assistant.skill_installer.resources.files",
|
|
side_effect=ModuleNotFoundError,
|
|
) as mock_files:
|
|
assert list_bundled_skills() == []
|
|
mock_files.assert_called_once()
|
|
|
|
|
|
def test_list_bundled_skills():
|
|
skills = list_bundled_skills()
|
|
|
|
by_name = {skill.name: skill for skill in skills}
|
|
assert "agent-evaluation" in by_name
|
|
assert by_name["agent-evaluation"].description
|
|
assert (by_name["agent-evaluation"].path / "SKILL.md").is_file()
|
|
assert [skill.name for skill in skills] == sorted(skill.name for skill in skills)
|