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.
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"))
|
|
}
|