mirror of
https://github.com/usestrix/strix.git
synced 2026-08-16 01:16:40 +02:00
79 lines
2.5 KiB
Python
79 lines
2.5 KiB
Python
"""Tests for stripping the markdown code fence off stored code fields."""
|
|
|
|
from __future__ import annotations
|
|
|
|
from pygments.lexers import BashLexer, PythonLexer
|
|
|
|
from strix.report.writer import (
|
|
guess_language_name,
|
|
parse_fenced_code,
|
|
resolve_lexer,
|
|
safe_fence,
|
|
)
|
|
from strix.viewer.report_pdf import _strip_code_fence
|
|
|
|
|
|
def test_parse_fenced_code_extracts_language_and_body() -> None:
|
|
language, code = parse_fenced_code("```python\nimport requests\nprint(1)\n```")
|
|
assert language == "python"
|
|
assert code == "import requests\nprint(1)"
|
|
|
|
|
|
def test_parse_fenced_code_uses_first_token_of_info_string() -> None:
|
|
language, code = parse_fenced_code("```python title=app.py\nx = 1\n```")
|
|
assert language == "python"
|
|
assert code == "x = 1"
|
|
|
|
|
|
def test_parse_fenced_code_handles_non_python_language() -> None:
|
|
language, code = parse_fenced_code("```http\nGET / HTTP/1.1\n```")
|
|
assert language == "http"
|
|
assert code == "GET / HTTP/1.1"
|
|
|
|
|
|
def test_parse_fenced_code_passes_through_unfenced() -> None:
|
|
language, code = parse_fenced_code("import requests\nprint(1)")
|
|
assert language is None
|
|
assert code == "import requests\nprint(1)"
|
|
|
|
|
|
def test_parse_fenced_code_fence_without_language() -> None:
|
|
language, code = parse_fenced_code("```\nplain\n```")
|
|
assert language is None
|
|
assert code == "plain"
|
|
|
|
|
|
def test_strip_code_fence_removes_fence() -> None:
|
|
assert _strip_code_fence("```python\nx = 1\n```") == "x = 1"
|
|
|
|
|
|
def test_strip_code_fence_passes_through_non_string_and_unfenced() -> None:
|
|
assert _strip_code_fence(None) is None
|
|
assert _strip_code_fence("x = 1") == "x = 1"
|
|
|
|
|
|
def test_resolve_lexer_honors_explicit_language() -> None:
|
|
assert isinstance(resolve_lexer("bash", "echo hi"), BashLexer)
|
|
|
|
|
|
def test_resolve_lexer_falls_back_to_python_when_unresolvable() -> None:
|
|
# Unknown language name and empty body -> nothing to auto-detect -> Python.
|
|
assert isinstance(resolve_lexer("not-a-language", ""), PythonLexer)
|
|
|
|
|
|
def test_guess_language_name_defaults_to_python_when_inconclusive() -> None:
|
|
assert guess_language_name("") == "python"
|
|
|
|
|
|
def test_parse_fenced_code_handles_crlf() -> None:
|
|
language, code = parse_fenced_code("```python\r\nx = 1\r\n```")
|
|
assert language == "python"
|
|
assert code == "x = 1"
|
|
|
|
|
|
def test_safe_fence_widens_past_embedded_backticks() -> None:
|
|
# A PoC body containing a ``` run must be wrapped in a longer fence so it
|
|
# can't terminate the block early.
|
|
assert safe_fence("plain code") == "```"
|
|
assert safe_fence("has ```\nfence inside") == "````"
|