You cannot select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
88 lines
2.5 KiB
Go
88 lines
2.5 KiB
Go
package logx
|
|
|
|
import (
|
|
"log"
|
|
"strings"
|
|
"sync/atomic"
|
|
"testing"
|
|
"time"
|
|
|
|
"github.com/stretchr/testify/assert"
|
|
)
|
|
|
|
func TestWithDurationError(t *testing.T) {
|
|
var builder strings.Builder
|
|
log.SetOutput(&builder)
|
|
WithDuration(time.Second).Error("foo")
|
|
assert.True(t, strings.Contains(builder.String(), "duration"), builder.String())
|
|
}
|
|
|
|
func TestWithDurationErrorf(t *testing.T) {
|
|
var builder strings.Builder
|
|
log.SetOutput(&builder)
|
|
WithDuration(time.Second).Errorf("foo")
|
|
assert.True(t, strings.Contains(builder.String(), "duration"), builder.String())
|
|
}
|
|
|
|
func TestWithDurationErrorv(t *testing.T) {
|
|
var builder strings.Builder
|
|
log.SetOutput(&builder)
|
|
WithDuration(time.Second).Errorv("foo")
|
|
assert.True(t, strings.Contains(builder.String(), "duration"), builder.String())
|
|
}
|
|
|
|
func TestWithDurationInfo(t *testing.T) {
|
|
var builder strings.Builder
|
|
log.SetOutput(&builder)
|
|
WithDuration(time.Second).Info("foo")
|
|
assert.True(t, strings.Contains(builder.String(), "duration"), builder.String())
|
|
}
|
|
|
|
func TestWithDurationInfoConsole(t *testing.T) {
|
|
old := atomic.LoadUint32(&encoding)
|
|
atomic.StoreUint32(&encoding, plainEncodingType)
|
|
defer func() {
|
|
atomic.StoreUint32(&encoding, old)
|
|
}()
|
|
|
|
var builder strings.Builder
|
|
log.SetOutput(&builder)
|
|
WithDuration(time.Second).Info("foo")
|
|
assert.True(t, strings.Contains(builder.String(), "ms"), builder.String())
|
|
}
|
|
|
|
func TestWithDurationInfof(t *testing.T) {
|
|
var builder strings.Builder
|
|
log.SetOutput(&builder)
|
|
WithDuration(time.Second).Infof("foo")
|
|
assert.True(t, strings.Contains(builder.String(), "duration"), builder.String())
|
|
}
|
|
|
|
func TestWithDurationInfov(t *testing.T) {
|
|
var builder strings.Builder
|
|
log.SetOutput(&builder)
|
|
WithDuration(time.Second).Infov("foo")
|
|
assert.True(t, strings.Contains(builder.String(), "duration"), builder.String())
|
|
}
|
|
|
|
func TestWithDurationSlow(t *testing.T) {
|
|
var builder strings.Builder
|
|
log.SetOutput(&builder)
|
|
WithDuration(time.Second).Slow("foo")
|
|
assert.True(t, strings.Contains(builder.String(), "duration"), builder.String())
|
|
}
|
|
|
|
func TestWithDurationSlowf(t *testing.T) {
|
|
var builder strings.Builder
|
|
log.SetOutput(&builder)
|
|
WithDuration(time.Second).WithDuration(time.Hour).Slowf("foo")
|
|
assert.True(t, strings.Contains(builder.String(), "duration"), builder.String())
|
|
}
|
|
|
|
func TestWithDurationSlowv(t *testing.T) {
|
|
var builder strings.Builder
|
|
log.SetOutput(&builder)
|
|
WithDuration(time.Second).WithDuration(time.Hour).Slowv("foo")
|
|
assert.True(t, strings.Contains(builder.String(), "duration"), builder.String())
|
|
}
|