forked from harness/gitness
-
Notifications
You must be signed in to change notification settings - Fork 0
/
stages_test.go
97 lines (80 loc) · 2.5 KB
/
stages_test.go
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
// Copyright 2019 Drone.IO Inc. All rights reserved.
// Use of this source code is governed by the Drone Non-Commercial License
// that can be found in the LICENSE file.
// +build !oss
package metric
import (
"testing"
"github.com/drone/drone/core"
"github.com/drone/drone/mock"
"github.com/golang/mock/gomock"
"github.com/prometheus/client_golang/prometheus"
)
func TestStagePendingCount(t *testing.T) {
controller := gomock.NewController(t)
// restore the default prometheus registerer
// when the unit test is complete.
snapshot := prometheus.DefaultRegisterer
defer func() {
prometheus.DefaultRegisterer = snapshot
controller.Finish()
}()
// creates a blank registry
registry := prometheus.NewRegistry()
prometheus.DefaultRegisterer = registry
// x5 stage count
data := []*core.Stage{{}, {}, {}, {}, {}}
stages := mock.NewMockStageStore(controller)
stages.EXPECT().ListState(gomock.Any(), core.StatusPending).Return(data, nil)
PendingJobCount(stages)
metrics, err := registry.Gather()
if err != nil {
t.Error(err)
return
}
if want, got := len(metrics), 1; want != got {
t.Errorf("Expect registered metric")
return
}
metric := metrics[0]
if want, got := metric.GetName(), "drone_pending_jobs"; want != got {
t.Errorf("Expect metric name %s, got %s", want, got)
}
if want, got := metric.Metric[0].Gauge.GetValue(), float64(len(data)); want != got {
t.Errorf("Expect metric value %f, got %f", want, got)
}
}
func TestStageRunningCount(t *testing.T) {
controller := gomock.NewController(t)
// restore the default prometheus registerer
// when the unit test is complete.
snapshot := prometheus.DefaultRegisterer
defer func() {
prometheus.DefaultRegisterer = snapshot
controller.Finish()
}()
// creates a blank registry
registry := prometheus.NewRegistry()
prometheus.DefaultRegisterer = registry
// x5 stage count
data := []*core.Stage{{}, {}, {}, {}, {}}
stages := mock.NewMockStageStore(controller)
stages.EXPECT().ListState(gomock.Any(), core.StatusRunning).Return(data, nil)
RunningJobCount(stages)
metrics, err := registry.Gather()
if err != nil {
t.Error(err)
return
}
if want, got := len(metrics), 1; want != got {
t.Errorf("Expect registered metric")
return
}
metric := metrics[0]
if want, got := metric.GetName(), "drone_running_jobs"; want != got {
t.Errorf("Expect metric name %s, got %s", want, got)
}
if want, got := metric.Metric[0].Gauge.GetValue(), float64(len(data)); want != got {
t.Errorf("Expect metric value %f, got %f", want, got)
}
}