5d2e01a6d7
Co-authored-by: Carlo <carloantonio.patti@mistral.ai> Co-authored-by: Mathias Gesbert <mathias.gesbert@mistral.ai> Co-authored-by: Clement Sirieix <clem.sirieix@gmail.com> Co-authored-by: Michel Thomazo <michel.thomazo@mistral.ai> Co-authored-by: Clément Drouin <clement.drouin@mistral.ai> Co-authored-by: Vincent Guilloux <vincent.guilloux@mistral.ai> Co-authored-by: Thomas Kenbeek <thomas.kenbeek@mistral.ai> Co-authored-by: Mistral Vibe <vibe@mistral.ai>
184 lines
8.7 KiB
Python
184 lines
8.7 KiB
Python
from __future__ import annotations
|
|
|
|
from pathlib import Path
|
|
from unittest.mock import patch
|
|
|
|
from vibe.core.paths.config_paths import (
|
|
discover_local_agents_dirs,
|
|
discover_local_skills_dirs,
|
|
discover_local_tools_dirs,
|
|
)
|
|
|
|
|
|
class TestDiscoverLocalSkillsDirs:
|
|
def test_returns_empty_list_when_dir_not_trusted(self, tmp_path: Path) -> None:
|
|
(tmp_path / ".vibe" / "skills").mkdir(parents=True)
|
|
with patch("vibe.core.paths.config_paths.trusted_folders_manager") as mock_tfm:
|
|
mock_tfm.is_trusted.return_value = False
|
|
assert discover_local_skills_dirs(tmp_path) == []
|
|
|
|
def test_returns_empty_list_when_trusted_but_no_skills_dirs(
|
|
self, tmp_path: Path
|
|
) -> None:
|
|
with patch("vibe.core.paths.config_paths.trusted_folders_manager") as mock_tfm:
|
|
mock_tfm.is_trusted.return_value = True
|
|
assert discover_local_skills_dirs(tmp_path) == []
|
|
|
|
def test_returns_vibe_skills_only_when_only_it_exists(self, tmp_path: Path) -> None:
|
|
vibe_skills = tmp_path / ".vibe" / "skills"
|
|
vibe_skills.mkdir(parents=True)
|
|
with patch("vibe.core.paths.config_paths.trusted_folders_manager") as mock_tfm:
|
|
mock_tfm.is_trusted.return_value = True
|
|
result = discover_local_skills_dirs(tmp_path)
|
|
assert result == [vibe_skills]
|
|
|
|
def test_returns_agents_skills_only_when_only_it_exists(
|
|
self, tmp_path: Path
|
|
) -> None:
|
|
agents_skills = tmp_path / ".agents" / "skills"
|
|
agents_skills.mkdir(parents=True)
|
|
with patch("vibe.core.paths.config_paths.trusted_folders_manager") as mock_tfm:
|
|
mock_tfm.is_trusted.return_value = True
|
|
result = discover_local_skills_dirs(tmp_path)
|
|
assert result == [agents_skills]
|
|
|
|
def test_returns_both_in_order_when_both_exist(self, tmp_path: Path) -> None:
|
|
vibe_skills = tmp_path / ".vibe" / "skills"
|
|
agents_skills = tmp_path / ".agents" / "skills"
|
|
vibe_skills.mkdir(parents=True)
|
|
agents_skills.mkdir(parents=True)
|
|
with patch("vibe.core.paths.config_paths.trusted_folders_manager") as mock_tfm:
|
|
mock_tfm.is_trusted.return_value = True
|
|
result = discover_local_skills_dirs(tmp_path)
|
|
assert result == [vibe_skills, agents_skills]
|
|
|
|
def test_ignores_vibe_skills_when_file_not_dir(self, tmp_path: Path) -> None:
|
|
(tmp_path / ".vibe").mkdir()
|
|
(tmp_path / ".vibe" / "skills").write_text("", encoding="utf-8")
|
|
with patch("vibe.core.paths.config_paths.trusted_folders_manager") as mock_tfm:
|
|
mock_tfm.is_trusted.return_value = True
|
|
result = discover_local_skills_dirs(tmp_path)
|
|
assert result == []
|
|
|
|
def test_finds_skills_dirs_recursively_in_trusted_folder(
|
|
self, tmp_path: Path
|
|
) -> None:
|
|
(tmp_path / ".vibe" / "skills").mkdir(parents=True)
|
|
(tmp_path / "sub" / ".agents" / "skills").mkdir(parents=True)
|
|
(tmp_path / "sub" / "deep" / ".vibe" / "skills").mkdir(parents=True)
|
|
with patch("vibe.core.paths.config_paths.trusted_folders_manager") as mock_tfm:
|
|
mock_tfm.is_trusted.return_value = True
|
|
result = discover_local_skills_dirs(tmp_path)
|
|
assert result == [
|
|
tmp_path / ".vibe" / "skills",
|
|
tmp_path / "sub" / ".agents" / "skills",
|
|
tmp_path / "sub" / "deep" / ".vibe" / "skills",
|
|
]
|
|
|
|
def test_does_not_descend_into_ignored_dirs(self, tmp_path: Path) -> None:
|
|
(tmp_path / ".vibe" / "skills").mkdir(parents=True)
|
|
(tmp_path / "node_modules" / ".vibe" / "skills").mkdir(parents=True)
|
|
with patch("vibe.core.paths.config_paths.trusted_folders_manager") as mock_tfm:
|
|
mock_tfm.is_trusted.return_value = True
|
|
result = discover_local_skills_dirs(tmp_path)
|
|
assert result == [tmp_path / ".vibe" / "skills"]
|
|
|
|
|
|
class TestDiscoverLocalToolsDirs:
|
|
def test_returns_empty_list_when_dir_not_trusted(self, tmp_path: Path) -> None:
|
|
(tmp_path / ".vibe" / "tools").mkdir(parents=True)
|
|
with patch("vibe.core.paths.config_paths.trusted_folders_manager") as mock_tfm:
|
|
mock_tfm.is_trusted.return_value = False
|
|
assert discover_local_tools_dirs(tmp_path) == []
|
|
|
|
def test_returns_empty_list_when_trusted_but_no_tools_dir(
|
|
self, tmp_path: Path
|
|
) -> None:
|
|
with patch("vibe.core.paths.config_paths.trusted_folders_manager") as mock_tfm:
|
|
mock_tfm.is_trusted.return_value = True
|
|
assert discover_local_tools_dirs(tmp_path) == []
|
|
|
|
def test_returns_tools_dir_when_exists(self, tmp_path: Path) -> None:
|
|
vibe_tools = tmp_path / ".vibe" / "tools"
|
|
vibe_tools.mkdir(parents=True)
|
|
with patch("vibe.core.paths.config_paths.trusted_folders_manager") as mock_tfm:
|
|
mock_tfm.is_trusted.return_value = True
|
|
result = discover_local_tools_dirs(tmp_path)
|
|
assert result == [vibe_tools]
|
|
|
|
def test_ignores_tools_when_file_not_dir(self, tmp_path: Path) -> None:
|
|
(tmp_path / ".vibe").mkdir()
|
|
(tmp_path / ".vibe" / "tools").write_text("", encoding="utf-8")
|
|
with patch("vibe.core.paths.config_paths.trusted_folders_manager") as mock_tfm:
|
|
mock_tfm.is_trusted.return_value = True
|
|
result = discover_local_tools_dirs(tmp_path)
|
|
assert result == []
|
|
|
|
def test_finds_tools_dirs_recursively(self, tmp_path: Path) -> None:
|
|
(tmp_path / ".vibe" / "tools").mkdir(parents=True)
|
|
(tmp_path / "sub" / ".vibe" / "tools").mkdir(parents=True)
|
|
with patch("vibe.core.paths.config_paths.trusted_folders_manager") as mock_tfm:
|
|
mock_tfm.is_trusted.return_value = True
|
|
result = discover_local_tools_dirs(tmp_path)
|
|
assert result == [
|
|
tmp_path / ".vibe" / "tools",
|
|
tmp_path / "sub" / ".vibe" / "tools",
|
|
]
|
|
|
|
def test_does_not_descend_into_ignored_dirs(self, tmp_path: Path) -> None:
|
|
(tmp_path / ".vibe" / "tools").mkdir(parents=True)
|
|
(tmp_path / ".git" / ".vibe" / "tools").mkdir(parents=True)
|
|
with patch("vibe.core.paths.config_paths.trusted_folders_manager") as mock_tfm:
|
|
mock_tfm.is_trusted.return_value = True
|
|
result = discover_local_tools_dirs(tmp_path)
|
|
assert result == [tmp_path / ".vibe" / "tools"]
|
|
|
|
|
|
class TestDiscoverLocalAgentsDirs:
|
|
def test_returns_empty_list_when_dir_not_trusted(self, tmp_path: Path) -> None:
|
|
(tmp_path / ".vibe" / "agents").mkdir(parents=True)
|
|
with patch("vibe.core.paths.config_paths.trusted_folders_manager") as mock_tfm:
|
|
mock_tfm.is_trusted.return_value = False
|
|
assert discover_local_agents_dirs(tmp_path) == []
|
|
|
|
def test_returns_empty_list_when_trusted_but_no_agents_dir(
|
|
self, tmp_path: Path
|
|
) -> None:
|
|
with patch("vibe.core.paths.config_paths.trusted_folders_manager") as mock_tfm:
|
|
mock_tfm.is_trusted.return_value = True
|
|
assert discover_local_agents_dirs(tmp_path) == []
|
|
|
|
def test_returns_agents_dir_when_exists(self, tmp_path: Path) -> None:
|
|
vibe_agents = tmp_path / ".vibe" / "agents"
|
|
vibe_agents.mkdir(parents=True)
|
|
with patch("vibe.core.paths.config_paths.trusted_folders_manager") as mock_tfm:
|
|
mock_tfm.is_trusted.return_value = True
|
|
result = discover_local_agents_dirs(tmp_path)
|
|
assert result == [vibe_agents]
|
|
|
|
def test_ignores_agents_when_file_not_dir(self, tmp_path: Path) -> None:
|
|
(tmp_path / ".vibe").mkdir()
|
|
(tmp_path / ".vibe" / "agents").write_text("", encoding="utf-8")
|
|
with patch("vibe.core.paths.config_paths.trusted_folders_manager") as mock_tfm:
|
|
mock_tfm.is_trusted.return_value = True
|
|
result = discover_local_agents_dirs(tmp_path)
|
|
assert result == []
|
|
|
|
def test_finds_agents_dirs_recursively(self, tmp_path: Path) -> None:
|
|
(tmp_path / ".vibe" / "agents").mkdir(parents=True)
|
|
(tmp_path / "sub" / "deep" / ".vibe" / "agents").mkdir(parents=True)
|
|
with patch("vibe.core.paths.config_paths.trusted_folders_manager") as mock_tfm:
|
|
mock_tfm.is_trusted.return_value = True
|
|
result = discover_local_agents_dirs(tmp_path)
|
|
assert result == [
|
|
tmp_path / ".vibe" / "agents",
|
|
tmp_path / "sub" / "deep" / ".vibe" / "agents",
|
|
]
|
|
|
|
def test_does_not_descend_into_ignored_dirs(self, tmp_path: Path) -> None:
|
|
(tmp_path / ".vibe" / "agents").mkdir(parents=True)
|
|
(tmp_path / "__pycache__" / ".vibe" / "agents").mkdir(parents=True)
|
|
with patch("vibe.core.paths.config_paths.trusted_folders_manager") as mock_tfm:
|
|
mock_tfm.is_trusted.return_value = True
|
|
result = discover_local_agents_dirs(tmp_path)
|
|
assert result == [tmp_path / ".vibe" / "agents"]
|