add unit test for logger package

pkg/logger/logger.go

@@ -28,15 +28,26 @@ func getLoggerLevel(lvl string) zapcore.Level {
 // Init initializes global custom zap logger
 func Init(encoding, level string) {
 
+	// select encoding level
 	encodingLevel := zapcore.LowercaseColorLevelEncoder
 	if encoding == "json" {
 		encodingLevel = zapcore.LowercaseLevelEncoder
 	}
 
+	// get logger
+	logger := GetLogger(level, encoding, encodingLevel)
+
+	// initialize global logger
+	zap.ReplaceGlobals(logger)
+}
+
+// GetLogger creates a customer zap logger
+func GetLogger(logLevel, encoding string, encodingLevel func(zapcore.Level, zapcore.PrimitiveArrayEncoder)) *zap.Logger {
+
 	// build zap config
 	zapConfig := zap.Config{
 		Encoding:    encoding,
-		Level:       zap.NewAtomicLevelAt(getLoggerLevel(level)),
+		Level:       zap.NewAtomicLevelAt(getLoggerLevel(logLevel)),
 		OutputPaths: []string{"stdout"},
 		EncoderConfig: zapcore.EncoderConfig{
 			LevelKey:     "level",
@@ -48,8 +59,9 @@ func Init(encoding, level string) {
 			EncodeCaller: zapcore.ShortCallerEncoder,
 		},
 	}
+
+	// create zap logger
 	logger, _ := zapConfig.Build()
 
-	// initialize global logger
-	zap.ReplaceGlobals(logger)
+	return logger
 }

pkg/logger/logger_test.go

@@ -0,0 +1,73 @@
+package logger
+
+import (
+	"testing"
+
+	"go.uber.org/zap/zapcore"
+)
+
+func TestGetLoggerLevel(t *testing.T) {
+	table := []struct {
+		name  string
+		input string
+		want  zapcore.Level
+	}{
+		{
+			name:  "empty log level",
+			input: "",
+			want:  zapcore.InfoLevel,
+		},
+		{
+			name:  "invalid log level",
+			input: "some log level",
+			want:  zapcore.InfoLevel,
+		},
+		{
+			name:  "debug log level",
+			input: "debug",
+			want:  zapcore.DebugLevel,
+		},
+		{
+			name:  "panic log level",
+			input: "panic",
+			want:  zapcore.PanicLevel,
+		},
+	}
+
+	for _, tt := range table {
+		got := getLoggerLevel(tt.input)
+		if got != tt.want {
+			t.Errorf("got: '%v', want: '%v'", got, tt.want)
+		}
+	}
+}
+
+func TestGetLogger(t *testing.T) {
+
+	table := []struct {
+		name          string
+		logLevel      string
+		encoding      string
+		encodingLevel func(zapcore.Level, zapcore.PrimitiveArrayEncoder)
+	}{
+		{
+			name:          "check debug log level",
+			logLevel:      "debug",
+			encoding:      "json",
+			encodingLevel: zapcore.LowercaseLevelEncoder,
+		},
+		{
+			name:          "check log level",
+			logLevel:      "panic",
+			encoding:      "console",
+			encodingLevel: zapcore.LowercaseLevelEncoder,
+		},
+	}
+
+	for _, tt := range table {
+		got := GetLogger(tt.logLevel, tt.encoding, tt.encodingLevel)
+		if ce := got.Check(getLoggerLevel(tt.logLevel), "testing"); ce == nil {
+			t.Errorf("unexpected error")
+		}
+	}
+}