From 7c417bbc9861ddba735cc4134d6ef1ea1e8a27fd Mon Sep 17 00:00:00 2001 From: Alysson Ribeiro <15274059+sonalys@users.noreply.github.com> Date: Thu, 14 Aug 2025 14:55:08 +0200 Subject: [PATCH] increase policies test coverage --- policies_test.go | 35 +++++++++++++++++++++++++++++++++++ 1 file changed, 35 insertions(+) diff --git a/policies_test.go b/policies_test.go index 8aaa92d..66fa066 100644 --- a/policies_test.go +++ b/policies_test.go @@ -6,6 +6,7 @@ import ( "github.com/sonalys/sanitize" "github.com/stretchr/testify/require" + "golang.org/x/net/html/atom" ) func Test_SanitizeEmail(t *testing.T) { @@ -20,3 +21,37 @@ func Test_SanitizeEmail(t *testing.T) { require.Equal(t, string(content), out.String()) } + +func Test_BlockAttrs(t *testing.T) { + content := []byte(``) + out := bytes.NewBuffer(make([]byte, 0, 1024)) + err := sanitize.HTML(bytes.NewReader(content), out, + sanitize.BlockAttrs("style"), + ) + require.NoError(t, err) + + require.Equal(t, "", out.String()) +} + +func Test_BlockTags(t *testing.T) { + content := []byte(``) + out := bytes.NewBuffer(make([]byte, 0, 1024)) + err := sanitize.HTML(bytes.NewReader(content), out, + sanitize.BlockTags(atom.A), + ) + require.NoError(t, err) + + require.Equal(t, "", out.String()) +} + +func Test_AllowTags(t *testing.T) { + content := []byte(``) + out := bytes.NewBuffer(make([]byte, 0, 1024)) + err := sanitize.HTML(bytes.NewReader(content), out, + sanitize.Blacklist(), + sanitize.AllowTags(atom.Html, atom.Body, atom.A), + ) + require.NoError(t, err) + + require.Equal(t, "", out.String()) +}