78 lines
1.9 KiB
Go
78 lines
1.9 KiB
Go
package logging
|
|
|
|
import (
|
|
"errors"
|
|
"os"
|
|
"reflect"
|
|
"testing"
|
|
)
|
|
|
|
func Test_ParseLogFormat(t *testing.T) {
|
|
|
|
type testData struct {
|
|
format string
|
|
expected LogFormat
|
|
expectedErr error
|
|
}
|
|
|
|
tests := []testData{
|
|
{format: "", expected: UnspecifiedFormat, expectedErr: nil},
|
|
{format: " ", expected: UnspecifiedFormat, expectedErr: nil},
|
|
{format: "standard", expected: StandardFormat, expectedErr: nil},
|
|
{format: "STANDARD", expected: StandardFormat, expectedErr: nil},
|
|
{format: "json", expected: JSONFormat, expectedErr: nil},
|
|
{format: " json ", expected: JSONFormat, expectedErr: nil},
|
|
{format: "bogus", expected: UnspecifiedFormat, expectedErr: errors.New("Unknown log format: bogus")},
|
|
}
|
|
|
|
for _, test := range tests {
|
|
result, err := ParseLogFormat(test.format)
|
|
if test.expected != result {
|
|
t.Errorf("expected %s, got %s", test.expected, result)
|
|
}
|
|
if !reflect.DeepEqual(test.expectedErr, err) {
|
|
t.Errorf("expected error %v, got %v", test.expectedErr, err)
|
|
}
|
|
}
|
|
}
|
|
|
|
func Test_ParseEnv_VAULT_LOG_FORMAT(t *testing.T) {
|
|
|
|
oldVLF := os.Getenv("VAULT_LOG_FORMAT")
|
|
defer os.Setenv("VAULT_LOG_FORMAT", oldVLF)
|
|
|
|
testParseEnvLogFormat(t, "VAULT_LOG_FORMAT")
|
|
}
|
|
|
|
func Test_ParseEnv_LOGXI_FORMAT(t *testing.T) {
|
|
|
|
oldVLF := os.Getenv("VAULT_LOG_FORMAT")
|
|
defer os.Setenv("VAULT_LOG_FORMAT", oldVLF)
|
|
|
|
oldLogxi := os.Getenv("LOGXI_FORMAT")
|
|
defer os.Setenv("LOGXI_FORMAT", oldLogxi)
|
|
|
|
os.Setenv("VAULT_LOG_FORMAT", "")
|
|
testParseEnvLogFormat(t, "LOGXI_FORMAT")
|
|
}
|
|
|
|
func testParseEnvLogFormat(t *testing.T, name string) {
|
|
|
|
env := []string{
|
|
"json", "vauLT_Json", "VAULT-JSON", "vaulTJSon",
|
|
"standard", "STANDARD",
|
|
"bogus"}
|
|
|
|
formats := []LogFormat{
|
|
JSONFormat, JSONFormat, JSONFormat, JSONFormat,
|
|
StandardFormat, StandardFormat,
|
|
UnspecifiedFormat}
|
|
|
|
for i, e := range env {
|
|
os.Setenv(name, e)
|
|
if lf := ParseEnvLogFormat(); formats[i] != lf {
|
|
t.Errorf("expected %s, got %s", formats[i], lf)
|
|
}
|
|
}
|
|
}
|