mirror of
https://github.com/zeromicro/go-zero.git
synced 2025-01-24 01:30:25 +08:00
41 lines
714 B
Go
41 lines
714 B
Go
package stat
|
|
|
|
import (
|
|
"strconv"
|
|
"testing"
|
|
"time"
|
|
|
|
"github.com/stretchr/testify/assert"
|
|
)
|
|
|
|
func TestMetrics(t *testing.T) {
|
|
DisableLog()
|
|
defer logEnabled.Set(true)
|
|
|
|
counts := []int{1, 5, 10, 100, 1000, 1000}
|
|
for _, count := range counts {
|
|
m := NewMetrics("foo")
|
|
m.SetName("bar")
|
|
for i := 0; i < count; i++ {
|
|
m.Add(Task{
|
|
Duration: time.Millisecond * time.Duration(i),
|
|
Description: strconv.Itoa(i),
|
|
})
|
|
}
|
|
m.AddDrop()
|
|
var writer mockedWriter
|
|
SetReportWriter(&writer)
|
|
m.executor.Flush()
|
|
assert.Equal(t, "bar", writer.report.Name)
|
|
}
|
|
}
|
|
|
|
type mockedWriter struct {
|
|
report *StatReport
|
|
}
|
|
|
|
func (m *mockedWriter) Write(report *StatReport) error {
|
|
m.report = report
|
|
return nil
|
|
}
|