mirror of
https://github.com/prometheus/prometheus
synced 2024-12-27 17:13:22 +00:00
20dbb128d9
Per Julien's feedback on #10369, we're choosing to be consistent with data types inside the stats structure (ints) rather than with the points format that is part of the normal query responses (strings). We have this option because this data cannot be NaN/Inf. Signed-off-by: Andrew Bloomgarden <blmgrdn@amazon.com>
101 lines
3.9 KiB
Go
101 lines
3.9 KiB
Go
// Copyright 2017 The Prometheus Authors
|
|
// Licensed under the Apache License, Version 2.0 (the "License");
|
|
// you may not use this file except in compliance with the License.
|
|
// You may obtain a copy of the License at
|
|
//
|
|
// http://www.apache.org/licenses/LICENSE-2.0
|
|
//
|
|
// Unless required by applicable law or agreed to in writing, software
|
|
// distributed under the License is distributed on an "AS IS" BASIS,
|
|
// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
|
// See the License for the specific language governing permissions and
|
|
// limitations under the License.
|
|
|
|
package stats
|
|
|
|
import (
|
|
"encoding/json"
|
|
"testing"
|
|
"time"
|
|
|
|
"github.com/grafana/regexp"
|
|
"github.com/prometheus/client_golang/prometheus"
|
|
"github.com/stretchr/testify/require"
|
|
|
|
"github.com/prometheus/prometheus/util/testutil"
|
|
)
|
|
|
|
func TestTimerGroupNewTimer(t *testing.T) {
|
|
tg := NewTimerGroup()
|
|
timer := tg.GetTimer(ExecTotalTime)
|
|
duration := timer.Duration()
|
|
require.Equal(t, 0.0, duration, "Expected duration equal 0")
|
|
minimum := 2 * time.Millisecond
|
|
timer.Start()
|
|
time.Sleep(minimum)
|
|
timer.Stop()
|
|
duration = timer.Duration()
|
|
require.Greater(t, duration, 0.0, "Expected duration greater than 0")
|
|
elapsed := timer.ElapsedTime()
|
|
require.GreaterOrEqual(t, elapsed, minimum,
|
|
"Expected elapsed time to be greater than time slept.")
|
|
}
|
|
|
|
func TestQueryStatsWithTimersAndSamples(t *testing.T) {
|
|
qt := NewQueryTimers()
|
|
qs := NewQuerySamples(true)
|
|
qs.InitStepTracking(20001000, 25001000, 1000000)
|
|
timer := qt.GetTimer(ExecTotalTime)
|
|
timer.Start()
|
|
time.Sleep(2 * time.Millisecond)
|
|
timer.Stop()
|
|
qs.IncrementSamplesAtTimestamp(20001000, 5)
|
|
qs.IncrementSamplesAtTimestamp(25001000, 5)
|
|
|
|
qstats := NewQueryStats(&Statistics{Timers: qt, Samples: qs})
|
|
actual, err := json.Marshal(qstats)
|
|
require.NoError(t, err, "unexpected error during serialization")
|
|
// Timing value is one of multiple fields, unit is seconds (float).
|
|
match, err := regexp.MatchString(`[,{]"execTotalTime":\d+\.\d+[,}]`, string(actual))
|
|
require.NoError(t, err, "unexpected error while matching string")
|
|
require.True(t, match, "Expected timings with one non-zero entry.")
|
|
|
|
require.Regexpf(t, `[,{]"totalQueryableSamples":10[,}]`, string(actual), "expected totalQueryableSamples")
|
|
require.Regexpf(t, `[,{]"totalQueryableSamplesPerStep":\[\[20001,5\],\[21001,0\],\[22001,0\],\[23001,0\],\[24001,0\],\[25001,5\]\]`, string(actual), "expected totalQueryableSamplesPerStep")
|
|
}
|
|
|
|
func TestQueryStatsWithSpanTimers(t *testing.T) {
|
|
qt := NewQueryTimers()
|
|
qs := NewQuerySamples(false)
|
|
ctx := &testutil.MockContext{DoneCh: make(chan struct{})}
|
|
qst, _ := qt.GetSpanTimer(ctx, ExecQueueTime, prometheus.NewSummary(prometheus.SummaryOpts{}))
|
|
time.Sleep(5 * time.Millisecond)
|
|
qst.Finish()
|
|
qstats := NewQueryStats(&Statistics{Timers: qt, Samples: qs})
|
|
actual, err := json.Marshal(qstats)
|
|
require.NoError(t, err, "unexpected error during serialization")
|
|
// Timing value is one of multiple fields, unit is seconds (float).
|
|
match, err := regexp.MatchString(`[,{]"execQueueTime":\d+\.\d+[,}]`, string(actual))
|
|
require.NoError(t, err, "unexpected error while matching string")
|
|
require.True(t, match, "Expected timings with one non-zero entry.")
|
|
}
|
|
|
|
func TestTimerGroup(t *testing.T) {
|
|
tg := NewTimerGroup()
|
|
require.Equal(t, "Exec total time: 0s", tg.GetTimer(ExecTotalTime).String())
|
|
|
|
require.Equal(t, "Exec queue wait time: 0s", tg.GetTimer(ExecQueueTime).String())
|
|
|
|
require.Equal(t, "Inner eval time: 0s", tg.GetTimer(InnerEvalTime).String())
|
|
|
|
require.Equal(t, "Query preparation time: 0s", tg.GetTimer(QueryPreparationTime).String())
|
|
|
|
require.Equal(t, "Result sorting time: 0s", tg.GetTimer(ResultSortTime).String())
|
|
|
|
require.Equal(t, "Eval total time: 0s", tg.GetTimer(EvalTotalTime).String())
|
|
|
|
actual := tg.String()
|
|
expected := "Exec total time: 0s\nExec queue wait time: 0s\nInner eval time: 0s\nQuery preparation time: 0s\nResult sorting time: 0s\nEval total time: 0s\n"
|
|
require.Equal(t, expected, actual)
|
|
}
|