mirror of
https://github.com/usestrix/strix.git
synced 2026-08-20 18:38:57 +02:00
104 lines
3.0 KiB
Go
104 lines
3.0 KiB
Go
package render
|
|
|
|
import (
|
|
"strings"
|
|
"testing"
|
|
|
|
"github.com/charmbracelet/x/ansi"
|
|
)
|
|
|
|
func TestHighlightCodeColorsKnownLanguage(t *testing.T) {
|
|
out := HighlightCode("def main():\n return 1", "python")
|
|
if !strings.Contains(out, "\x1b[") {
|
|
t.Fatal("python code was not colorized")
|
|
}
|
|
if ansi.Strip(out) != "def main():\n return 1" {
|
|
t.Fatalf("highlighting changed the code text: %q", ansi.Strip(out))
|
|
}
|
|
}
|
|
|
|
func TestMarkdownCodeFenceIsHighlighted(t *testing.T) {
|
|
out := renderAssistantMarkdown("intro\n```python\nimport os\n```\ndone")
|
|
plain := ansi.Strip(out)
|
|
if !strings.Contains(plain, "import os") {
|
|
t.Fatalf("code fence content missing: %q", plain)
|
|
}
|
|
if strings.Contains(plain, "```") {
|
|
t.Fatalf("fence markers leaked into output: %q", plain)
|
|
}
|
|
}
|
|
|
|
func TestParseFencedCode(t *testing.T) {
|
|
lang, code := ParseFencedCode("```python\nprint(1)\n```")
|
|
if lang != "python" || code != "print(1)" {
|
|
t.Fatalf("got lang=%q code=%q", lang, code)
|
|
}
|
|
lang, code = ParseFencedCode("plain text")
|
|
if lang != "" || code != "plain text" {
|
|
t.Fatalf("unfenced text mangled: lang=%q code=%q", lang, code)
|
|
}
|
|
}
|
|
|
|
func TestMarkdownTableIsAligned(t *testing.T) {
|
|
out := renderAssistantMarkdown(strings.Join([]string{
|
|
"| Name | Severity |",
|
|
"| --- | --- |",
|
|
"| SQLi | **high** |",
|
|
"| XSS | low |",
|
|
}, "\n"))
|
|
plain := ansi.Strip(out)
|
|
lines := strings.Split(plain, "\n")
|
|
if len(lines) != 4 {
|
|
t.Fatalf("expected 4 table rows, got %d: %q", len(lines), plain)
|
|
}
|
|
if !strings.Contains(lines[0], "Name") || !strings.Contains(lines[0], "│") {
|
|
t.Fatalf("header row not formatted: %q", lines[0])
|
|
}
|
|
if !strings.Contains(lines[1], "─┼─") {
|
|
t.Fatalf("separator rule missing: %q", lines[1])
|
|
}
|
|
if !strings.Contains(lines[2], "high") || strings.Contains(lines[2], "**") {
|
|
t.Fatalf("body cell not inline-formatted: %q", lines[2])
|
|
}
|
|
if strings.Index(lines[2], "│") != strings.Index(lines[3], "│") {
|
|
t.Fatalf("columns misaligned:\n%q\n%q", lines[2], lines[3])
|
|
}
|
|
}
|
|
|
|
func TestNonTablePipeLinesAreLeftAlone(t *testing.T) {
|
|
out := renderAssistantMarkdown("a | b\nplain line")
|
|
if !strings.Contains(ansi.Strip(out), "a | b") {
|
|
t.Fatalf("pipe text mangled: %q", ansi.Strip(out))
|
|
}
|
|
}
|
|
|
|
func TestInlineFormatKeepsNonEmphasisMarkers(t *testing.T) {
|
|
literal := []string{
|
|
"ls *.py *.go",
|
|
"snake_case_name and other_var_here",
|
|
"a * b * c",
|
|
"call obj.__init__ now",
|
|
"rm -rf /tmp/* /var/*",
|
|
"5 * 3 = 15",
|
|
}
|
|
for _, line := range literal {
|
|
if got := ansi.Strip(inlineFormat(line)); got != line {
|
|
t.Fatalf("%q was treated as emphasis: %q", line, got)
|
|
}
|
|
}
|
|
}
|
|
|
|
func TestInlineFormatStillStylesRealEmphasis(t *testing.T) {
|
|
cases := map[string]string{
|
|
"this is *italic* text": "this is italic text",
|
|
"this is **bold** text": "this is bold text",
|
|
"gone ~~away~~ now": "gone away now",
|
|
"use `code` here": "use code here",
|
|
}
|
|
for line, want := range cases {
|
|
if got := ansi.Strip(inlineFormat(line)); got != want {
|
|
t.Fatalf("%q: got %q want %q", line, got, want)
|
|
}
|
|
}
|
|
}
|