mirror of
https://github.com/mistralai/mistral-vibe
synced 2026-04-25 17:14:55 +02:00
04305bd77c
Co-authored-by: Kim-Adeline Miguel <51720070+kimadeline@users.noreply.github.com> Co-authored-by: Mathias Gesbert <mathias.gesbert@mistral.ai> Co-authored-by: Michel Thomazo <51709227+michelTho@users.noreply.github.com> Co-authored-by: Pierre Rossinès <pierre.rossines@mistral.ai> Co-authored-by: Mistral Vibe <vibe@mistral.ai>
116 lines
3.0 KiB
Python
116 lines
3.0 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_file
|
|
---
|
|
|
|
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_file"
|
|
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() == ""
|