Test coverage

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

Hide file tree

Showing 3 changed files with 10 additions and 1 deletion.
diff --git a/README.md b/README.md
@@ -2,6 +2,7 @@
 Fast text generator on a mask.
 
 [![Go Report Card](https://goreportcard.com/badge/github.com/liderman/text-generator)](https://goreportcard.com/report/github.com/liderman/text-generator)
+[![codecov](https://codecov.io/gh/liderman/text-generator/branch/master/graph/badge.svg?token=53NH949TQY)](https://codecov.io/gh/liderman/text-generator)
 [![Go Reference](https://pkg.go.dev/badge/github.com/liderman/text-generator.svg)](https://pkg.go.dev/github.com/liderman/text-generator)
 [![Release](https://img.shields.io/github/release/liderman/text-generator.svg?style=flat-square)](https://github.com/liderman/text-generator/releases/latest)
 

diff --git a/cached-text-generator_test.go b/cached-text-generator_test.go
@@ -32,3 +32,11 @@ func TestCustomCachedGenerate(t *testing.T) {
 		assert.Equal(t, result, prevResult)
 	}
 }
+
+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, "!")
+}
diff --git a/text-generator.go b/text-generator.go
@@ -19,7 +19,7 @@ type TextGenerator struct {
 }
 
 // New returns a new instance a text generator.
-func New() GeneratorInterface {
+func New() *TextGenerator {
 	rand.Seed(time.Now().UnixNano())
 	return &TextGenerator{
 		startTag:  '{',