mirror of https://github.com/prometheus/prometheus
You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
100 lines
3.9 KiB
100 lines
3.9 KiB
// 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) |
|
}
|
|
|