html/escape_test.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())
		}
	})
}