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.
45 lines
831 B
Go
45 lines
831 B
Go
package logtest
|
|
|
|
import (
|
|
"errors"
|
|
"testing"
|
|
|
|
"github.com/stretchr/testify/assert"
|
|
"github.com/zeromicro/go-zero/core/logx"
|
|
)
|
|
|
|
func TestCollector(t *testing.T) {
|
|
const input = "hello"
|
|
c := NewCollector(t)
|
|
logx.Info(input)
|
|
assert.Equal(t, input, c.Content())
|
|
assert.Contains(t, c.String(), input)
|
|
c.Reset()
|
|
assert.Empty(t, c.Bytes())
|
|
}
|
|
|
|
func TestPanicOnFatal(t *testing.T) {
|
|
const input = "hello"
|
|
Discard(t)
|
|
logx.Info(input)
|
|
|
|
PanicOnFatal(t)
|
|
PanicOnFatal(t)
|
|
assert.Panics(t, func() {
|
|
logx.Must(errors.New("foo"))
|
|
})
|
|
}
|
|
|
|
func TestCollectorContent(t *testing.T) {
|
|
const input = "hello"
|
|
c := NewCollector(t)
|
|
c.buf.WriteString(input)
|
|
assert.Empty(t, c.Content())
|
|
c.Reset()
|
|
c.buf.WriteString(`{}`)
|
|
assert.Empty(t, c.Content())
|
|
c.Reset()
|
|
c.buf.WriteString(`{"content":1}`)
|
|
assert.Equal(t, "1", c.Content())
|
|
}
|