mirror of
https://github.com/zeromicro/go-zero.git
synced 2025-01-24 01:30:25 +08:00
22 lines
376 B
Go
22 lines
376 B
Go
package proc
|
|
|
|
import (
|
|
"log"
|
|
"strings"
|
|
"testing"
|
|
|
|
"github.com/stretchr/testify/assert"
|
|
)
|
|
|
|
func TestProfile(t *testing.T) {
|
|
var buf strings.Builder
|
|
log.SetOutput(&buf)
|
|
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"))
|
|
}
|