entry_test.go

package log

import (
	"fmt"
	"testing"

	"github.com/stretchr/testify/assert"
)

func TestEntry_WithFields(t *testing.T) {
	a := NewEntry(nil)
	assert.Nil(t, a.Fields)

	b := a.WithFields(Fields{"foo": "bar"})
	assert.Equal(t, Fields{}, a.mergedFields())
	assert.Equal(t, Fields{"foo": "bar"}, b.mergedFields())

	c := a.WithFields(Fields{"foo": "hello", "bar": "world"})

	e := c.finalize(InfoLevel, "upload")
	assert.Equal(t, e.Message, "upload")
	assert.Equal(t, e.Fields, Fields{"foo": "hello", "bar": "world"})
	assert.Equal(t, e.Level, InfoLevel)
	assert.NotEmpty(t, e.Timestamp)
}

func TestEntry_WithField(t *testing.T) {
	a := NewEntry(nil)
	b := a.WithField("foo", "bar")
	assert.Equal(t, Fields{}, a.mergedFields())
	assert.Equal(t, Fields{"foo": "bar"}, b.mergedFields())
}

func TestEntry_WithError(t *testing.T) {
	a := NewEntry(nil)
	b := a.WithError(fmt.Errorf("boom"))
	assert.Equal(t, Fields{}, a.mergedFields())
	assert.Equal(t, Fields{"error": "boom"}, b.mergedFields())

	c := a.WithError(nil)
	assert.Equal(t, Fields{"error": nil}, c.mergedFields())
}

func TestEntry_WithMem(t *testing.T) {
	a := NewEntry(nil).WithError(fmt.Errorf("boom")).WithMemStats()
	x := a.mergedFields()
	assert.NotEqual(t, x["error"], nil)
	assert.NotEqual(t, x["memstats"], nil)
	assert.NotEqual(t, x["memstats"].(logMemStats).Alloc, nil)
	assert.NotEqual(t, x["memstats"].(logMemStats).HeapAlloc, nil)
	assert.NotEqual(t, x["memstats"].(logMemStats).StackSys, nil)
	assert.Equal(t, x["notset"], nil)
}