test: added unit tests for new colors and Log method

colors/color_test.go

@@ -0,0 +1,19 @@
+package colors
+
+import (
+	"github.com/stretchr/testify/assert"
+	"testing"
+)
+
+func TestIsColorValid(t *testing.T) {
+	assert.True(t, IsColorValid(Red))
+	assert.True(t, IsColorValid(Magenta))
+	assert.True(t, IsColorValid(Yellow))
+	assert.True(t, IsColorValid(Cyan))
+	assert.True(t, IsColorValid(Gray))
+	assert.True(t, IsColorValid(White))
+	assert.True(t, IsColorValid(Black))
+	assert.False(t, IsColorValid("  "))
+	assert.False(t, IsColorValid("lfdjlfkasd"))
+	assert.False(t, IsColorValid("\\testno"))
+}

logs/logger_test.go

@@ -3,6 +3,7 @@ package logs
 import (
 	"bytes"
 	"github.com/stretchr/testify/assert"
+	"gitlab.com/docebo/libraries/go/tiny-logger/colors"
 	"io"
 	"os"
 	"testing"
@@ -135,6 +136,26 @@ func TestLogger_Error(t *testing.T) {
 	assert.Contains(t, buf.String(), testLog)
 }
 
+func TestLogger_Log(t *testing.T) {
+	var buf bytes.Buffer
+	testLog := "my GENERIC log test"
+	originalStdOut := os.Stdout
+	r, w, _ := os.Pipe()
+
+	os.Stdout = w
+	logger := NewLogger()
+	logger.Log("jldfald", testLog)
+	logger.Log(colors.Black, testLog)
+	logger.Log(colors.Cyan, testLog)
+
+	_ = w.Close()
+	_, _ = io.Copy(&buf, r)
+	os.Stdout = originalStdOut
+	assert.Contains(t, buf.String(), colors.White)
+	assert.Contains(t, buf.String(), colors.Black)
+	assert.Contains(t, buf.String(), colors.Cyan)
+}
+
 func TestLogger_BuildingMethods(t *testing.T) {
 	logger := NewLogger()
 	assert.IsType(t, &Logger{}, logger)

logs/package_logger_test.go

@@ -3,6 +3,7 @@ package logs
 import (
 	"bytes"
 	"github.com/stretchr/testify/assert"
+	"gitlab.com/docebo/libraries/go/tiny-logger/colors"
 	"io"
 	"os"
 	"testing"
@@ -123,3 +124,22 @@ func TestPackageLogger_Error(t *testing.T) {
 	os.Stderr = originalStdErr
 	assert.Contains(t, buf.String(), testLog)
 }
+
+func TestPackageLogger_Log(t *testing.T) {
+	var buf bytes.Buffer
+	testLog := "my GENERIC log test"
+	originalStdOut := os.Stdout
+	r, w, _ := os.Pipe()
+
+	os.Stdout = w
+	Log("jldfald", testLog)
+	Log(colors.Black, testLog)
+	Log(colors.Cyan, testLog)
+
+	_ = w.Close()
+	_, _ = io.Copy(&buf, r)
+	os.Stdout = originalStdOut
+	assert.Contains(t, buf.String(), colors.White)
+	assert.Contains(t, buf.String(), colors.Black)
+	assert.Contains(t, buf.String(), colors.Cyan)
+}