mirror of
https://github.com/alecthomas/chroma.git
synced 2025-03-19 21:10:15 +02:00
116 lines
3.3 KiB
Go
116 lines
3.3 KiB
Go
package html
|
|
|
|
import (
|
|
"bytes"
|
|
"errors"
|
|
"io/ioutil"
|
|
"strings"
|
|
"testing"
|
|
|
|
"github.com/alecthomas/assert"
|
|
"github.com/alecthomas/chroma"
|
|
"github.com/alecthomas/chroma/lexers"
|
|
"github.com/alecthomas/chroma/styles"
|
|
)
|
|
|
|
func TestCompressStyle(t *testing.T) {
|
|
style := "color: #888888; background-color: #faffff"
|
|
actual := compressStyle(style)
|
|
expected := "color:#888;background-color:#faffff"
|
|
assert.Equal(t, expected, actual)
|
|
}
|
|
|
|
func BenchmarkHTMLFormatter(b *testing.B) {
|
|
formatter := New()
|
|
b.ResetTimer()
|
|
for i := 0; i < b.N; i++ {
|
|
it, err := lexers.Get("go").Tokenise(nil, "package main\nfunc main()\n{\nprintln(`hello world`)\n}\n")
|
|
assert.NoError(b, err)
|
|
err = formatter.Format(ioutil.Discard, styles.Fallback, it)
|
|
assert.NoError(b, err)
|
|
}
|
|
}
|
|
|
|
func TestSplitTokensIntoLines(t *testing.T) {
|
|
in := []chroma.Token{
|
|
{Value: "hello", Type: chroma.NameKeyword},
|
|
{Value: " world\nwhat?\n", Type: chroma.NameKeyword},
|
|
}
|
|
expected := [][]chroma.Token{
|
|
{
|
|
{Type: chroma.NameKeyword, Value: "hello"},
|
|
{Type: chroma.NameKeyword, Value: " world\n"},
|
|
},
|
|
{
|
|
{Type: chroma.NameKeyword, Value: "what?\n"},
|
|
},
|
|
}
|
|
actual := chroma.SplitTokensIntoLines(in)
|
|
assert.Equal(t, expected, actual)
|
|
}
|
|
|
|
func TestIteratorPanicRecovery(t *testing.T) {
|
|
it := func() chroma.Token {
|
|
panic(errors.New("bad"))
|
|
}
|
|
err := New().Format(ioutil.Discard, styles.Fallback, it)
|
|
assert.Error(t, err)
|
|
}
|
|
|
|
func TestFormatterStyleToCSS(t *testing.T) {
|
|
builder := styles.Get("github").Builder()
|
|
builder.Add(chroma.LineHighlight, "bg:#ffffcc")
|
|
builder.Add(chroma.LineNumbers, "bold")
|
|
style, err := builder.Build()
|
|
if err != nil {
|
|
t.Error(err)
|
|
}
|
|
formatter := New(WithClasses())
|
|
css := formatter.styleToCSS(style)
|
|
for _, s := range css {
|
|
if strings.HasPrefix(strings.TrimSpace(s), ";") {
|
|
t.Errorf("rule starts with semicolon - expected valid css rule without semicolon: %v", s)
|
|
}
|
|
}
|
|
}
|
|
|
|
func TestClassPrefix(t *testing.T) {
|
|
wantPrefix := "some-prefix-"
|
|
withPrefix := New(WithClasses(), ClassPrefix(wantPrefix))
|
|
noPrefix := New(WithClasses())
|
|
for st := range chroma.StandardTypes {
|
|
if noPrefix.class(st) == "" {
|
|
if got := withPrefix.class(st); got != "" {
|
|
t.Errorf("Formatter.class(%v): prefix shouldn't be added to empty classes", st)
|
|
}
|
|
} else if got := withPrefix.class(st); !strings.HasPrefix(got, wantPrefix) {
|
|
t.Errorf("Formatter.class(%v): %q should have a class prefix", st, got)
|
|
}
|
|
}
|
|
|
|
var styleBuf bytes.Buffer
|
|
withPrefix.WriteCSS(&styleBuf, styles.Fallback)
|
|
if !strings.Contains(styleBuf.String(), ".some-prefix-chroma ") {
|
|
t.Error("Stylesheets should have a class prefix")
|
|
}
|
|
}
|
|
|
|
func TestTableLineNumberNewlines(t *testing.T) {
|
|
f := New(WithClasses(), WithLineNumbers(), LineNumbersInTable())
|
|
it, err := lexers.Get("go").Tokenise(nil, "package main\nfunc main()\n{\nprintln(`hello world`)\n}\n")
|
|
assert.NoError(t, err)
|
|
|
|
var buf bytes.Buffer
|
|
err = f.Format(&buf, styles.Fallback, it)
|
|
assert.NoError(t, err)
|
|
|
|
// Don't bother testing the whole output, just verify it's got line numbers
|
|
// in a <pre>-friendly format.
|
|
// Note: placing the newlines inside the <span> lets browser selections look
|
|
// better, instead of "skipping" over the span margin.
|
|
assert.Contains(t, buf.String(), `<span class="lnt">2
|
|
</span><span class="lnt">3
|
|
</span><span class="lnt">4
|
|
</span>`)
|
|
}
|