options_test.go

package zl

import (
	"testing"

	"github.com/stretchr/testify/assert"
	"go.uber.org/zap/zapcore"
)

func TestSetOutput(t *testing.T) {
	tests := []Output{PrettyOutput, ConsoleAndFileOutput, ConsoleOutput, FileOutput}
	for _, tt := range tests {
		t.Run(tt.String(), func(t *testing.T) {
			SetOutput(tt)
			assert.Equal(t, tt, outputType)
		})
	}
}

func TestSetOutputByString(t *testing.T) {
	tests := []struct {
		in  string
		out Output
	}{
		{"Pretty", PrettyOutput},
		{"ConsoleAndFile", ConsoleAndFileOutput},
		{"Console", ConsoleOutput},
		{"File", FileOutput},
		{"", PrettyOutput},
	}
	for _, tt := range tests {
		t.Run(tt.in, func(t *testing.T) {
			SetOutputByString(tt.in)
			assert.Equal(t, tt.out, outputType)
		})
	}
}

func TestSetLevel(t *testing.T) {
	tests := []zapcore.Level{DebugLevel, InfoLevel, WarnLevel, ErrorLevel, FatalLevel}
	for _, tt := range tests {
		t.Run(tt.String(), func(t *testing.T) {
			SetLevel(tt)
			assert.Equal(t, tt, severityLevel)
			ResetGlobalLoggerSettings()
		})
	}
}

func TestSetLevelByString(t *testing.T) {
	tests := []struct {
		in  string
		out zapcore.Level
	}{
		{"debug", DebugLevel},
		{"info", InfoLevel},
		{"warn", WarnLevel},
		{"error", ErrorLevel},
		{"fatal", FatalLevel},
	}
	for _, tt := range tests {
		t.Run(tt.in, func(t *testing.T) {
			SetLevelByString(tt.in)
			assert.Equal(t, tt.out, severityLevel)
			ResetGlobalLoggerSettings()
		})
	}
}

func TestSetFieldKey(t *testing.T) {
	SetFieldKey(FunctionKey, "func")
	SetFieldKey(StacktraceKey, "stack_trace")
	SetFieldKey(CallerKey, "")
	SetFieldKey("", "abc")
	assert.Equal(t,
		map[Key]string{
			FunctionKey:   "func",
			StacktraceKey: "stack_trace",
		},
		fieldKeys,
	)
	ResetGlobalLoggerSettings()
}

func TestSetSeparator(t *testing.T) {
	SetSeparator(":")
	assert.Equal(t, ":", separator)
	ResetGlobalLoggerSettings()
}