package nbtohtml

import (
	"html/template"
	"testing"

	"github.com/stretchr/testify/assert"
)

func TestEscapeHTML(t *testing.T) {
	expected := template.HTML(
		"<p>Hello world</p>\n<script>window.alert('I'm evil!');</script>",
	)
	actual := escapeHTML(`<p>Hello world</p>
<script>window.alert('I'm evil!');</script>`)
	assert.Equal(t, expected, actual)
}

func TestSanitizeHTML(t *testing.T) {
	expected := template.HTML("<p>Hello world</p>\n")
	actual := sanitizeHTML(`<p>Hello world</p>
<script>window.alert('I'm evil!');</script>`)
	assert.Equal(t, expected, actual)
}