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.
go-zero/core/load/sheddingstat_test.go

25 lines
440 B
Go

package load
import (
"testing"
"github.com/stretchr/testify/assert"
)
func TestSheddingStat(t *testing.T) {
st := NewSheddingStat("any")
for i := 0; i < 3; i++ {
st.IncrementTotal()
}
for i := 0; i < 5; i++ {
st.IncrementPass()
}
for i := 0; i < 7; i++ {
st.IncrementDrop()
}
result := st.reset()
assert.Equal(t, int64(3), result.Total)
assert.Equal(t, int64(5), result.Pass)
assert.Equal(t, int64(7), result.Drop)
}