Skip to content

Commit 42c9390

Browse files
committed
fix: Update test
1 parent a5eb586 commit 42c9390

File tree

1 file changed

+12
-9
lines changed

1 file changed

+12
-9
lines changed

monitor/service_test.go

Lines changed: 12 additions & 9 deletions
Original file line numberDiff line numberDiff line change
@@ -416,21 +416,24 @@ func TestMonitor_QuickClose(t *testing.T) {
416416
}
417417
}
418418

419-
func TestMonitor_CQDiagnostics(t *testing.T) {
419+
func TestMonitor_CQStatistics(t *testing.T) {
420420
s := monitor.New(nil, monitor.Config{}, &tsdb.Config{})
421421
err := s.Open()
422422
require.NoError(t, err, "monitor open")
423423
defer s.Close()
424424

425425
s.RegisterDiagnosticsClient("cq", continuous_querier.NewService(continuous_querier.NewConfig()))
426-
d, err := s.Diagnostics()
427-
require.NoError(t, err, "cq diagnostics")
428-
429-
diags, ok := d["cq"]
430-
require.True(t, ok, "no diagnostics found for 'cq'")
431-
432-
require.Equal(t, diags.Columns, []string{"queryFail", "queryOk"}, "diagnostics columns")
433-
require.Equal(t, diags.Rows, [][]interface{}{{int64(0), int64(0)}}, "diagnostics rows")
426+
stats, err := s.Statistics(nil)
427+
require.NoError(t, err, "cq statistics")
428+
429+
for _, stat := range stats {
430+
if stat.Name == "cq" {
431+
require.Equal(t, stat.Values, map[string]interface{}{
432+
"queryOk": 0,
433+
"queryFail": 0,
434+
}, "statistics")
435+
}
436+
}
434437
}
435438

436439
func TestStatistic_ValueNames(t *testing.T) {

0 commit comments

Comments
 (0)