Fixed tests

liderman · Nov 14, 2021 · 8c17c13 · 8c17c13
1 parent c672d25
commit 8c17c13
Show file tree

Hide file tree

Showing 2 changed files with 12 additions and 3 deletions.
diff --git a/cached-text-generator_test.go b/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('!'))
 }
diff --git a/text-generator_test.go b/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) {