From 8c17c13a19df425f7cebb849ecc32b495d89b038 Mon Sep 17 00:00:00 2001
From: Osipov Konstantin <k.osipov.msk@gmail.com>
Date: Sun, 14 Nov 2021 23:17:07 +0300
Subject: [PATCH] Fixed tests

---
 cached-text-generator_test.go | 6 +++---
 text-generator_test.go        | 9 +++++++++
 2 files changed, 12 insertions(+), 3 deletions(-)

diff --git a/cached-text-generator_test.go b/cached-text-generator_test.go
index 3f3dcc6..ae65d2c 100644
--- 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
index a11b093..9ec38b8 100644
--- 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) {