mirror of
https://github.com/zeromicro/go-zero.git
synced 2025-01-24 01:30:25 +08:00
ac321fc146
* backup * simplify * chore: remove unused pool * chore: fix lint errors * chore: use strings.Builder instead of bytes.Buffer * test: add more tests * chore: fix reviewdog * test: fix data race * feat: make logger customizable * chore: fix reviewdog * test: fix fails * chore: fix set writer twice * chore: use context instead of golang.org context * chore: specify uint32 for level types
30 lines
513 B
Go
30 lines
513 B
Go
package proc
|
|
|
|
import (
|
|
"strings"
|
|
"testing"
|
|
|
|
"github.com/stretchr/testify/assert"
|
|
"github.com/zeromicro/go-zero/core/logx"
|
|
)
|
|
|
|
func TestProfile(t *testing.T) {
|
|
var buf strings.Builder
|
|
w := logx.NewWriter(&buf)
|
|
o := logx.Reset()
|
|
logx.SetWriter(w)
|
|
|
|
defer func() {
|
|
logx.Reset()
|
|
logx.SetWriter(o)
|
|
}()
|
|
|
|
profiler := StartProfile()
|
|
// start again should not work
|
|
assert.NotNil(t, StartProfile())
|
|
profiler.Stop()
|
|
// stop twice
|
|
profiler.Stop()
|
|
assert.True(t, strings.Contains(buf.String(), ".pprof"))
|
|
}
|