Fixed tests

Author: liderman

cached-text-generator_test.go

@@ -36,7 +36,7 @@ func TestCustomCachedGenerate(t *testing.T) {
 func TestCachedTextGenerator_CustomGenerateConfig(t *testing.T) {
 	tg := New()
 	NewCached(tg, 10).Configure('[', ']', '!')
-	assert.Equal(t, tg.startTag, "[")
-	assert.Equal(t, tg.endTag, "]")
-	assert.Equal(t, tg.separator, "!")
+	assert.Equal(t, tg.startTag, rune('['))
+	assert.Equal(t, tg.endTag, rune(']'))
+	assert.Equal(t, tg.separator, rune('!'))
 }

text-generator_test.go

@@ -30,6 +30,15 @@ func TestTextGenerator_Generate(t *testing.T) {
 
 	t8 := tg.Generate("{|}")
 	assert.Equal(t, t8, "")
+
+	t9 := tg.Generate("{")
+	assert.Equal(t, t9, "{")
+
+	t10 := tg.Generate("}")
+	assert.Equal(t, t10, "}")
+
+	t11 := tg.Generate("|")
+	assert.Equal(t, t11, "|")
 }
 
 func TestTextGenerator_CustomGenerateConfig(t *testing.T) {