1
0
html/escape_test.go

97 lines
2.1 KiB
Go

package html_test
import (
"bytes"
"code.squareroundforest.org/arpio/html"
. "code.squareroundforest.org/arpio/html/tag"
"testing"
)
func TestEscape(t *testing.T) {
t.Run("attribute escape", func(t *testing.T) {
var buf bytes.Buffer
if err := html.WriteRaw(&buf, Div(Attr("foo", "\"bar\" & \"baz\""))); err != nil {
t.Fatal(err)
}
if buf.String() != "<div foo=\"&quot;bar&quot; &amp; &quot;baz&quot;\"></div>" {
t.Fatal(buf.String())
}
})
t.Run("failing writer", func(t *testing.T) {
ew := &errorWriter{}
if err := html.WriteRaw(ew, Span("foo bar baz")); err == nil {
t.Fatal("failed to fail")
}
})
t.Run("broken unicode", func(t *testing.T) {
b := []byte{'f', 0xc2, 'o', 'o'}
var buf bytes.Buffer
if err := html.WriteRaw(&buf, Span(string(b))); err != nil {
t.Fatal(err)
}
if buf.String() != "<span>foo</span>" {
t.Fatal(buf.String())
}
})
t.Run("multiple spaces", func(t *testing.T) {
var buf bytes.Buffer
if err := html.WriteRaw(&buf, Span("foo bar")); err != nil {
t.Fatal(err)
}
if buf.String() != "<span>foo&nbsp;&nbsp;&nbsp;bar</span>" {
t.Fatal(buf.String())
}
})
t.Run("single space", func(t *testing.T) {
var buf bytes.Buffer
if err := html.WriteRaw(&buf, Span("foo bar")); err != nil {
t.Fatal(err)
}
if buf.String() != "<span>foo bar</span>" {
t.Fatal(buf.String())
}
})
t.Run("tailing space", func(t *testing.T) {
var buf bytes.Buffer
if err := html.WriteRaw(&buf, Span("foo ")); err != nil {
t.Fatal(err)
}
if buf.String() != "<span>foo </span>" {
t.Fatal(buf.String())
}
})
t.Run("unicode non-break space", func(t *testing.T) {
var buf bytes.Buffer
if err := html.WriteRaw(&buf, Span(string([]rune{0xa0}))); err != nil {
t.Fatal(err)
}
if buf.String() != "<span>&nbsp;</span>" {
t.Fatal(buf.String())
}
})
t.Run("html characters", func(t *testing.T) {
var buf bytes.Buffer
if err := html.WriteRaw(&buf, Span("<foo&bar>")); err != nil {
t.Fatal(err)
}
if buf.String() != "<span>&lt;foo&amp;bar&gt;</span>" {
t.Fatal(buf.String())
}
})
}