Add more tests to helper

1 files changed, 11 insertions, 0 deletions

diff --git a/helpers/content_test.go b/helpers/content_test.go
index 18cdfcc5d..63ef82f2f 100644
--- a/helpers/content_test.go
+++ b/helpers/content_test.go
@@ -1,6 +1,8 @@
 package helpers
 
 import (
+	"github.com/stretchr/testify/assert"
+	"html/template"
 	"testing"
 )
 
@@ -20,3 +22,12 @@ func TestStripHTML(t *testing.T) {
 		}
 	}
 }
+
+func TestStripEmptyNav(t *testing.T) {
+	cleaned := StripEmptyNav([]byte("do<nav>\n</nav>\n\nbedobedo"))
+	assert.Equal(t, []byte("dobedobedo"), cleaned)
+}
+
+func TestBytesToHTML(t *testing.T) {
+	assert.Equal(t, template.HTML("dobedobedo"), BytesToHTML([]byte("dobedobedo")))
+}