wlog_test.go

package wlog

import (
	"bytes"
	"encoding/json"
	"testing"
	"time"
)

func TestHooks(t *testing.T) {

	// Install a hook to catch all NFO (Info) messages
	InstallHook(Nfo, func(timestamp time.Time, logLevel LogLevel, message string) {
		if logLevel != Nfo {
			t.Fatalf("Expected %s but got %s", Nfo, logLevel)
		}
	})

	// Install a hook to catch all WRN (Warning) messages
	InstallHook(Wrn, func(timestamp time.Time, logLevel LogLevel, message string) {
		if logLevel != Wrn {
			t.Fatalf("Expected %s but got %s", Wrn, logLevel)
		}
	})

	Info("This is a NFO log entry")
	Warning("This is a WRN log entry")
	Error("This is a ERR log entry. No hooks installed for this level")
}

func TestFieldMapping(t *testing.T) {

	SetFormatter(JSONFormatter{
		Compact: true,
	})

	SetStdOut(false)

	w := &bytes.Buffer{}

	SetWriter(w)

	SetFieldMapping(FieldMapping{"name": "n", "address": "addr", "tenantId": "tid"})

	logger := WithScope(Fields{"tenantId": "1223456", "name": "user", "address": "my street"})

	logger.Info("This is a test")

	reader := bytes.NewReader(w.Bytes())
	d := json.NewDecoder(reader)

	var data map[string]string
	d.Decode(&data)

	expectedKeys := []string{"@t", "@m", "@l", "tid", "n", "addr"}

	for _, k := range expectedKeys {
		if _, ok := data[k]; !ok {
			t.Fatalf("should contain a key: %s", k)
		}
	}
}

func TestLogLevel(t *testing.T) {

	txt := "data"

	SetStdOut(false)

	w := &bytes.Buffer{}

	SetWriter(w)

	SetLogLevel(Dbg)
	Debug(txt)
	if w.Len() == 0 {
		t.Fatalf("expected log output")
	}

	w.Reset()
	Info(txt)
	if w.Len() == 0 {
		t.Fatalf("expected log output")
	}

	SetLogLevel(Nfo)
	w.Reset()
	Debug(txt)
	if w.Len() > 0 {
		t.Fatalf("unexpected log output")
	}

	w.Reset()
	Info(txt)
	if w.Len() == 0 {
		t.Fatalf("expected log output")
	}

	w.Reset()
	Warningf(txt)
	if w.Len() == 0 {
		t.Fatalf("expected log output")
	}

	SetLogLevel(Ftl)
	w.Reset()
	Debug(txt)
	if w.Len() > 0 {
		t.Fatalf("unexpected log output")
	}

	w.Reset()
	Debugf(txt)
	if w.Len() > 0 {
		t.Fatalf("unexpected log output")
	}

	w.Reset()
	Info(txt)
	if w.Len() > 0 {
		t.Fatalf("unexpected log output")
	}

	w.Reset()
	Infof(txt)
	if w.Len() > 0 {
		t.Fatalf("unexpected log output")
	}

	w.Reset()
	Warning(txt)
	if w.Len() > 0 {
		t.Fatalf("unexpected log output")
	}

	w.Reset()
	Warningf(txt)
	if w.Len() > 0 {
		t.Fatalf("unexpected log output")
	}

	w.Reset()
	Error(txt)
	if w.Len() > 0 {
		t.Fatalf("unexpected log output")
	}

	w.Reset()
	Errorf(txt)
	if w.Len() > 0 {
		t.Fatalf("unexpected log output")
	}
}