enhanced core/stat module unit test

Author: binbin0325

core/stat/base/bucket_leap_array_test.go

@@ -125,3 +125,37 @@ func TestBucketLeapArray_resetBucketTo(t *testing.T) {
 		t.Errorf("BucketLeapArray.ResetBucketTo() execute fail.")
 	}
 }
+
+func TestAddCount(t *testing.T) {
+	bla := NewBucketLeapArray(SampleCount, IntervalInMs)
+	bla.AddCount(base.MetricEventPass, 1)
+	passCount := bla.Count(base.MetricEventPass)
+	assert.True(t, passCount == 1)
+}
+
+func TestMinRt(t *testing.T) {
+	t.Run("TestMinRt_Default", func(t *testing.T) {
+		bla := NewBucketLeapArray(SampleCount, IntervalInMs)
+		minRt := bla.MinRt()
+		assert.True(t, minRt == base.DefaultStatisticMaxRt)
+	})
+
+	t.Run("TestMinRt", func(t *testing.T) {
+		bla := NewBucketLeapArray(SampleCount, IntervalInMs)
+		bla.AddCount(base.MetricEventRt, 100)
+		minRt := bla.MinRt()
+		assert.True(t, minRt == 100)
+	})
+}
+
+func TestGetIntervalInSecond(t *testing.T) {
+	bla := NewBucketLeapArray(SampleCount, IntervalInMs)
+	second := bla.GetIntervalInSecond()
+	assert.True(t, float64(IntervalInMs)/1000.0 == second)
+}
+
+func TestDataType(t *testing.T) {
+	bla := NewBucketLeapArray(SampleCount, IntervalInMs)
+	dataType := bla.DataType()
+	assert.True(t, dataType == "MetricBucket")
+}

core/stat/base/leap_array_test.go

@@ -6,6 +6,8 @@ import (
 	"testing"
 	"unsafe"
 
+	"github.com/stretchr/testify/assert"
+
 	"github.com/alibaba/sentinel-golang/util"
 	"github.com/stretchr/testify/mock"
 )
@@ -264,3 +266,22 @@ func Test_leapArray_isBucketDeprecated_normal(t *testing.T) {
 		})
 	}
 }
+
+func TestNewLeapArray(t *testing.T) {
+	t.Run("TestNewLeapArray_Normal", func(t *testing.T) {
+		_, err := NewLeapArray(SampleCount, IntervalInMs, &leapArrayMock{})
+		assert.Nil(t, err)
+	})
+
+	t.Run("TestNewLeapArray_Generator_Nil", func(t *testing.T) {
+		leapArray, err := NewLeapArray(SampleCount, IntervalInMs, nil)
+		assert.Nil(t, leapArray)
+		assert.Error(t, err, "Invalid parameters, BucketGenerator is nil")
+	})
+
+	t.Run("TestNewLeapArray_Invalid_Parameters", func(t *testing.T) {
+		leapArray, err := NewLeapArray(30, IntervalInMs, nil)
+		assert.Nil(t, leapArray)
+		assert.Error(t, err, "Invalid parameters, intervalInMs is 10000, sampleCount is 30")
+	})
+}

core/stat/base/metric_bucket_test.go

@@ -6,6 +6,7 @@ import (
 	"unsafe"
 
 	"github.com/alibaba/sentinel-golang/core/base"
+	"github.com/stretchr/testify/assert"
 )
 
 func Test_metricBucket_MemSize(t *testing.T) {
@@ -112,3 +113,11 @@ func Test_metricBucket_Concurrent(t *testing.T) {
 		t.Error("unexpect count MetricEventRt")
 	}
 }
+
+func Test_Reset(t *testing.T) {
+	mb := NewMetricBucket()
+	mb.AddRt(100)
+	mb.reset()
+	rt := mb.MinRt()
+	assert.True(t, rt == base.DefaultStatisticMaxRt)
+}

core/stat/base/sliding_window_metric_test.go

@@ -3,6 +3,8 @@ package base
 import (
 	"testing"
 
+	"github.com/alibaba/sentinel-golang/util"
+
 	"github.com/alibaba/sentinel-golang/core/base"
 	"github.com/stretchr/testify/assert"
 )
@@ -166,3 +168,83 @@ func TestSlidingWindowMetric_GetIntervalSumWithTime(t *testing.T) {
 		})
 	}
 }
+
+func TestGetSum(t *testing.T) {
+	got, err := NewSlidingWindowMetric(4, 2000, NewBucketLeapArray(SampleCount, IntervalInMs))
+	assert.True(t, err == nil && got != nil)
+	passSum := got.GetSum(base.MetricEventPass)
+	assert.True(t, passSum == 0)
+}
+
+func TestGetQPS(t *testing.T) {
+	got, err := NewSlidingWindowMetric(4, 2000, NewBucketLeapArray(SampleCount, IntervalInMs))
+	assert.True(t, err == nil && got != nil)
+	qps := got.GetQPS(base.MetricEventPass)
+	assert.True(t, qps == 0)
+}
+
+func TestGetPreviousQPS(t *testing.T) {
+	got, err := NewSlidingWindowMetric(4, 2000, NewBucketLeapArray(SampleCount, IntervalInMs))
+	assert.True(t, err == nil && got != nil)
+	previousQPS := got.GetPreviousQPS(base.MetricEventPass)
+	assert.True(t, previousQPS == 0)
+}
+
+func TestGetQPSWithTime(t *testing.T) {
+	got, err := NewSlidingWindowMetric(4, 2000, NewBucketLeapArray(SampleCount, IntervalInMs))
+	assert.True(t, err == nil && got != nil)
+	qps := got.getQPSWithTime(util.CurrentTimeMillis(), base.MetricEventPass)
+	assert.True(t, qps == 0)
+}
+
+func TestGetMaxOfSingleBucket(t *testing.T) {
+	got, err := NewSlidingWindowMetric(4, 2000, NewBucketLeapArray(SampleCount, IntervalInMs))
+	assert.True(t, err == nil && got != nil)
+	got.real.AddCount(base.MetricEventPass, 100)
+	max := got.GetMaxOfSingleBucket(base.MetricEventPass)
+	assert.True(t, max == 100)
+}
+
+func TestMinRT(t *testing.T) {
+	got, err := NewSlidingWindowMetric(4, 2000, NewBucketLeapArray(SampleCount, IntervalInMs))
+	assert.True(t, err == nil && got != nil)
+	minRt := got.MinRT()
+	assert.True(t, minRt == float64(base.DefaultStatisticMaxRt))
+}
+
+func TestAvgRT(t *testing.T) {
+	got, err := NewSlidingWindowMetric(4, 2000, NewBucketLeapArray(SampleCount, IntervalInMs))
+	assert.True(t, err == nil && got != nil)
+	got.real.AddCount(base.MetricEventRt, 100)
+	got.real.AddCount(base.MetricEventComplete, 100)
+	avgRT := got.AvgRT()
+	assert.True(t, avgRT == 1)
+}
+
+func TestMetricItemFromBuckets(t *testing.T) {
+	got, err := NewSlidingWindowMetric(4, 2000, NewBucketLeapArray(SampleCount, IntervalInMs))
+	assert.True(t, err == nil && got != nil)
+	got.real.AddCount(base.MetricEventPass, 100)
+	item := got.metricItemFromBuckets(util.CurrentTimeMillis(), got.real.data.array.data)
+	assert.True(t, item.PassQps == 100)
+}
+
+func TestMetricItemFromBucket(t *testing.T) {
+	got, err := NewSlidingWindowMetric(4, 2000, NewBucketLeapArray(SampleCount, IntervalInMs))
+	assert.True(t, err == nil && got != nil)
+	got.real.AddCount(base.MetricEventPass, 100)
+	currentBucket, err := got.real.data.CurrentBucket(&leapArrayMock{})
+	assert.Nil(t, err)
+	item := got.metricItemFromBucket(currentBucket)
+	assert.True(t, item.PassQps == 100)
+}
+
+func TestSecondMetricsOnCondition(t *testing.T) {
+	got, err := NewSlidingWindowMetric(4, 2000, NewBucketLeapArray(SampleCount, IntervalInMs))
+	assert.True(t, err == nil && got != nil)
+	start, end := got.getBucketStartRange(util.CurrentTimeMillis())
+	items := got.SecondMetricsOnCondition(func(ws uint64) bool {
+		return ws >= start && ws <= end
+	})
+	assert.True(t, len(items) == 1)
+}