127 lines
4.1 KiB
Go
127 lines
4.1 KiB
Go
package request_aggregates
|
|
|
|
import (
|
|
"github.com/stretchr/testify/require"
|
|
"testing"
|
|
"time"
|
|
)
|
|
|
|
func TestTimeWindow_Add(t *testing.T) {
|
|
tw := &TimeWindow{}
|
|
|
|
tw.Add(&Request{Time: 123.45})
|
|
require.Equal(t, 1, len(tw.TimesTotal))
|
|
require.Equal(t, 1, len(tw.TimesSuccess))
|
|
require.Equal(t, 0, len(tw.TimesFail))
|
|
require.Equal(t, float64(123.45), tw.TimesTotal[0])
|
|
require.Equal(t, float64(123.45), tw.TimesSuccess[0])
|
|
|
|
tw.Add(&Request{Time: 100, Failure: false})
|
|
require.Equal(t, 2, len(tw.TimesTotal))
|
|
require.Equal(t, 2, len(tw.TimesSuccess))
|
|
require.Equal(t, 0, len(tw.TimesFail))
|
|
require.Equal(t, float64(100), tw.TimesTotal[1])
|
|
require.Equal(t, float64(100), tw.TimesSuccess[1])
|
|
|
|
tw.Add(&Request{Time: 200, Failure: true})
|
|
require.Equal(t, 3, len(tw.TimesTotal))
|
|
require.Equal(t, 2, len(tw.TimesSuccess))
|
|
require.Equal(t, 1, len(tw.TimesFail))
|
|
require.Equal(t, float64(200), tw.TimesTotal[2])
|
|
require.Equal(t, float64(200), tw.TimesFail[0])
|
|
}
|
|
|
|
func TestTimeWindow_Start(t *testing.T) {
|
|
now := time.Now()
|
|
tw := &TimeWindow{StartTime: now}
|
|
require.Equal(t, now, tw.Start())
|
|
}
|
|
|
|
func TestTimeWindow_End(t *testing.T) {
|
|
now := time.Now()
|
|
tw := &TimeWindow{EndTime: now}
|
|
require.Equal(t, now, tw.End())
|
|
}
|
|
|
|
func TestTimeWindow_Aggregate_All(t *testing.T) {
|
|
start := time.Now()
|
|
end := start.Add(time.Duration(60))
|
|
tw := &TimeWindow{StartTime: start, EndTime: end, OnlyTotal: false}
|
|
metrics, err := tw.Aggregate()
|
|
require.NoError(t, err)
|
|
require.Equal(t, 3, len(metrics))
|
|
require.Equal(t, end, metrics[0].Time())
|
|
require.Equal(t, MeasurementTime, metrics[0].Name())
|
|
require.Equal(t, end, metrics[1].Time())
|
|
require.Equal(t, MeasurementTimeFail, metrics[1].Name())
|
|
require.Equal(t, end, metrics[0].Time())
|
|
require.Equal(t, MeasurementTimeSuccess, metrics[2].Name())
|
|
}
|
|
|
|
func TestTimeWindow_Aggregate_OnlyTotal(t *testing.T) {
|
|
start := time.Now()
|
|
end := start.Add(time.Duration(60))
|
|
tw := &TimeWindow{StartTime: start, EndTime: end, OnlyTotal: true}
|
|
metrics, err := tw.Aggregate()
|
|
require.NoError(t, err)
|
|
require.Equal(t, 1, len(metrics))
|
|
require.Equal(t, end, metrics[0].Time())
|
|
require.Equal(t, MeasurementTime, metrics[0].Name())
|
|
}
|
|
|
|
func TestTimeWindow_aggregateTimes(t *testing.T) {
|
|
end := time.Now()
|
|
metric, err := aggregateTimes(MeasurementTime, []float64{500, 900, 300, 1000, 100, 600, 700, 800, 200, 400},
|
|
[]float32{60, 80, 99.9}, end)
|
|
require.NoError(t, err)
|
|
require.Equal(t, MeasurementTime, metric.Name())
|
|
require.Equal(t, int64(10), metric.Fields()[FieldTimeRequests])
|
|
require.Equal(t, float64(1000), metric.Fields()[FieldTimeMax])
|
|
require.Equal(t, float64(100), metric.Fields()[FieldTimeMin])
|
|
require.Equal(t, float64(550), metric.Fields()[FieldTimeMean])
|
|
require.Equal(t, float64(700), metric.Fields()[FieldTimePerc+"60"])
|
|
require.Equal(t, float64(900), metric.Fields()[FieldTimePerc+"80"])
|
|
require.Equal(t, float64(1000), metric.Fields()[FieldTimePerc+"99_9"])
|
|
}
|
|
|
|
func TestThroughputWindow_Add(t *testing.T) {
|
|
tw := &ThroughputWindow{}
|
|
|
|
tw.Add(&Request{})
|
|
require.Equal(t, int64(1), tw.RequestsTotal)
|
|
require.Equal(t, int64(0), tw.RequestsFail)
|
|
|
|
tw.Add(&Request{Failure: false})
|
|
require.Equal(t, int64(2), tw.RequestsTotal)
|
|
require.Equal(t, int64(0), tw.RequestsFail)
|
|
|
|
tw.Add(&Request{Failure: true})
|
|
require.Equal(t, int64(3), tw.RequestsTotal)
|
|
require.Equal(t, int64(1), tw.RequestsFail)
|
|
}
|
|
|
|
func TestThroughputWindow_Start(t *testing.T) {
|
|
now := time.Now()
|
|
tw := &ThroughputWindow{StartTime: now}
|
|
require.Equal(t, now, tw.Start())
|
|
}
|
|
|
|
func TestThroughputWindow_End(t *testing.T) {
|
|
now := time.Now()
|
|
tw := &ThroughputWindow{EndTime: now}
|
|
require.Equal(t, now, tw.End())
|
|
}
|
|
|
|
func TestThroughputWindow_Aggregate(t *testing.T) {
|
|
start := time.Now()
|
|
end := start.Add(time.Duration(60))
|
|
tw := &ThroughputWindow{StartTime: start, EndTime: end, RequestsTotal: 33, RequestsFail: 11}
|
|
metrics, err := tw.Aggregate()
|
|
require.NoError(t, err)
|
|
require.Equal(t, 1, len(metrics))
|
|
require.Equal(t, end, metrics[0].Time())
|
|
require.Equal(t, MeasurementThroughput, metrics[0].Name())
|
|
require.Equal(t, int64(33), metrics[0].Fields()[FieldThroughputTotal])
|
|
require.Equal(t, int64(11), metrics[0].Fields()[FieldThroughputFailed])
|
|
}
|