3f8487f761
Co-authored-by: Alexis Tacnet <alexis@mistral.ai> Co-authored-by: Clément Drouin <clement.drouin@mistral.ai> Co-authored-by: Guillaume LE GOFF <guillaume.lgf@gmail.com> Co-authored-by: Lucas Marandat <31749711+lucasmrdt@users.noreply.github.com> Co-authored-by: Maxime Dolores <maxime.dolores@ext.mistral.ai> Co-authored-by: Pierre Rossinès <pierre.rossines@mistral.ai> Co-authored-by: Quentin <quentin.torroba@mistral.ai> Co-authored-by: Val <102326092+vdeva@users.noreply.github.com> Co-authored-by: Vincent G <10739306+VinceOPS@users.noreply.github.com> Co-authored-by: p.vezia <166131032+le-codeur-rapide@users.noreply.github.com> Co-authored-by: Hiba Chaabnia <Hiba-Chaabnia@users.noreply.github.com> Co-authored-by: Nikhil Bhima <nikhilbhima@users.noreply.github.com> Co-authored-by: Nkipohcs <Nkipohcs@users.noreply.github.com> Co-authored-by: Mistral Vibe <vibe@mistral.ai>
115 lines
3 KiB
Python
115 lines
3 KiB
Python
from __future__ import annotations
|
|
|
|
import pytest
|
|
|
|
from vibe.core.skills.parser import SkillParseError, parse_skill_markdown
|
|
|
|
|
|
class TestParseSkillMarkdown:
|
|
def test_parses_valid_frontmatter(self) -> None:
|
|
content = """---
|
|
name: test-skill
|
|
description: A test skill
|
|
---
|
|
|
|
## Body content here
|
|
"""
|
|
frontmatter, body = parse_skill_markdown(content)
|
|
|
|
assert frontmatter["name"] == "test-skill"
|
|
assert frontmatter["description"] == "A test skill"
|
|
assert "## Body content here" in body
|
|
|
|
def test_parses_frontmatter_with_all_fields(self) -> None:
|
|
content = """---
|
|
name: full-skill
|
|
description: A skill with all fields
|
|
license: MIT
|
|
compatibility: Requires git
|
|
metadata:
|
|
author: Test Author
|
|
version: "1.0"
|
|
allowed-tools: bash read
|
|
---
|
|
|
|
Instructions here.
|
|
"""
|
|
frontmatter, body = parse_skill_markdown(content)
|
|
|
|
assert frontmatter["name"] == "full-skill"
|
|
assert frontmatter["description"] == "A skill with all fields"
|
|
assert frontmatter["license"] == "MIT"
|
|
assert frontmatter["compatibility"] == "Requires git"
|
|
assert frontmatter["metadata"]["author"] == "Test Author"
|
|
assert frontmatter["metadata"]["version"] == "1.0"
|
|
assert frontmatter["allowed-tools"] == "bash read"
|
|
assert "Instructions here." in body
|
|
|
|
def test_raises_error_for_missing_frontmatter(self) -> None:
|
|
content = "Just markdown content without frontmatter"
|
|
|
|
with pytest.raises(SkillParseError) as exc_info:
|
|
parse_skill_markdown(content)
|
|
|
|
assert "Missing or invalid YAML frontmatter" in str(exc_info.value)
|
|
|
|
def test_raises_error_for_unclosed_frontmatter(self) -> None:
|
|
content = """---
|
|
name: incomplete
|
|
description: Missing closing delimiter
|
|
"""
|
|
|
|
with pytest.raises(SkillParseError) as exc_info:
|
|
parse_skill_markdown(content)
|
|
|
|
assert "Missing or invalid YAML frontmatter" in str(exc_info.value)
|
|
|
|
def test_raises_error_for_invalid_yaml(self) -> None:
|
|
content = """---
|
|
name: [invalid yaml
|
|
description: broken
|
|
---
|
|
|
|
Body here.
|
|
"""
|
|
|
|
with pytest.raises(SkillParseError) as exc_info:
|
|
parse_skill_markdown(content)
|
|
|
|
assert "Invalid YAML frontmatter" in str(exc_info.value)
|
|
|
|
def test_raises_error_for_non_dict_frontmatter(self) -> None:
|
|
content = """---
|
|
- item1
|
|
- item2
|
|
---
|
|
|
|
Body here.
|
|
"""
|
|
|
|
with pytest.raises(SkillParseError) as exc_info:
|
|
parse_skill_markdown(content)
|
|
|
|
assert "must be a mapping" in str(exc_info.value)
|
|
|
|
def test_handles_empty_frontmatter(self) -> None:
|
|
content = """---
|
|
---
|
|
|
|
Body content.
|
|
"""
|
|
frontmatter, body = parse_skill_markdown(content)
|
|
|
|
assert frontmatter == {}
|
|
assert "Body content." in body
|
|
|
|
def test_handles_frontmatter_with_no_body(self) -> None:
|
|
content = """---
|
|
name: minimal
|
|
description: No body
|
|
---
|
|
"""
|
|
frontmatter, body = parse_skill_markdown(content)
|
|
|
|
assert frontmatter["name"] == "minimal"
|
|
assert body.strip() == ""
|